schedule_converter/grpc/convert.go

package grpc

import (
	pb "schedule_converter/proto/runner"

	"schedule_converter/models"
	"schedule_converter/parser"
)

func convertCourses(courses []models.Course) []*pb.Course {
	result := make([]*pb.Course, len(courses))
	for i, c := range courses {
		schedule := make([]*pb.ScheduleTime, len(c.Schedule))
		for j, s := range c.Schedule {
			schedule[j] = &pb.ScheduleTime{
				Day:      s.Day,
				Sections: convertToInt32(s.Section),
				Weeks:    convertToInt32(s.Weeks),
			}
		}
		result[i] = &pb.Course{
			Name:     c.Name,
			Teacher:  c.Teacher,
			Room:     c.Room,
			Schedule: schedule,
		}
	}
	return result
}

func convertToInt32(arr []int) []int32 {
	result := make([]int32, len(arr))
	for i, v := range arr {
		result[i] = int32(v)
	}
	return result
}

func convertGrades(grades []parser.GradeEntry) []*pb.Grade {
	result := make([]*pb.Grade, 0, len(grades))
	for _, g := range grades {
		result = append(result, &pb.Grade{
			Term:       g.Term,
			CourseCode: g.Code,
			CourseName: g.Name,
			Nature:     g.Nature,
			Category:   g.Category,
			Credit:     g.Credit,
			Score:      g.Score,
			GradePoint: g.GradePoint,
			IsExam:     g.IsExam,
		})
	}
	return result
}

func convertExams(exams []parser.ExamEntry) []*pb.Exam {
	result := make([]*pb.Exam, 0, len(exams))
	for _, e := range exams {
		examTypeStr := "期末"
		switch e.Type {
		case parser.ExamTypeMidterm:
			examTypeStr = "期中"
		case parser.ExamTypeMakeup:
			examTypeStr = "补考"
		}
		result = append(result, &pb.Exam{
			CourseName: e.Course,
			ExamType:   examTypeStr,
			Date:       e.Date,
			Time:       e.Time,
			Week:       e.Week,
			Weekday:    e.Weekday,
		})
	}
	return result
}

func convertEmptyClassrooms(entries []parser.EmptyClassroomEntry) []*pb.EmptyClassroomEntry {
	result := make([]*pb.EmptyClassroomEntry, 0, len(entries))
	for _, e := range entries {
		result = append(result, &pb.EmptyClassroomEntry{
			Classroom: e.Classroom,
			Weekday:   e.Weekday,
			Periods:   e.Periods,
			Term:      e.Term,
			Weeks:     e.Weeks,
		})
	}
	return result
}