package grpc

import (
	"context"
	"encoding/json"
	"log/slog"

	pb "schedule_converter/proto/runner"

	"schedule_converter/config"
	"schedule_converter/parser"
	apperr "schedule_converter/pkg/errors"
)

func (h *TaskHandler) getExams(ctx context.Context, payload []byte) ([]byte, error) {
	var req pb.GetExamsPayload
	if err := unmarshalPayload(payload, &req); err != nil {
		return nil, err
	}

	log := slog.With("username", req.Username, "semester", req.Semester)
	log.Info("获取考试安排任务")

	httpClient, err := createHTTPClient()
	if err != nil {
		return nil, err
	}

	cfg := config.Get()
	setTWFIDCookie(httpClient, cfg.BaseURL, "")

	if err := executeLogin(ctx, httpClient, req.Username, req.Password); err != nil {
		return nil, err
	}

	exams, err := parser.FetchAllExams(httpClient)
	if err != nil {
		return nil, apperr.New("fetch_exams", err, "获取考试安排失败")
	}

	pbExams := convertExams(exams)

	result := &pb.ExamsResultData{
		StudentId: req.Username,
		Semester:  req.Semester,
		Exams:     pbExams,
	}

	log.Info("考试安排获取成功", "exam_count", len(pbExams))
	return json.Marshal(result)
}