package grpc

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

	pb "schedule_converter/proto/runner"

	"schedule_converter/config"
)

func (h *TaskHandler) getSchedule(ctx context.Context, payload []byte) ([]byte, error) {
	var req pb.GetSchedulePayload
	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, "")

	courses, err := executeLoginAndFetchSchedule(ctx, httpClient, req.Username, req.Password, req.Semester)
	if err != nil {
		return nil, err
	}

	result := &pb.ScheduleResultData{
		StudentId: req.Username,
		Semester:  req.Semester,
		Courses:   convertCourses(courses),
	}

	log.Info("课表获取成功", "course_count", len(courses))
	return json.Marshal(result)
}