init

Author: BaiMeow

go.mod

@@ -0,0 +1,15 @@
+module github.com/hduLib/CourseICS
+
+go 1.19
+
+require (
+	github.com/emersion/go-ical v0.0.0-20220601085725-0864dccc089f
+	github.com/hduLib/hdu v0.0.0-20220831153942-4b1d5a2b88f1
+	github.com/teambition/rrule-go v1.7.2
+)
+
+require (
+	github.com/tidwall/gjson v1.14.2 // indirect
+	github.com/tidwall/match v1.1.1 // indirect
+	github.com/tidwall/pretty v1.2.0 // indirect
+)

go.sum

@@ -0,0 +1,12 @@
+github.com/emersion/go-ical v0.0.0-20220601085725-0864dccc089f h1:feGUUxxvOtWVOhTko8Cbmp33a+tU0IMZxMEmnkoAISQ=
+github.com/emersion/go-ical v0.0.0-20220601085725-0864dccc089f/go.mod h1:2MKFUgfNMULRxqZkadG1Vh44we3y5gJAtTBlVsx1BKQ=
+github.com/hduLib/hdu v0.0.0-20220831153942-4b1d5a2b88f1 h1:yCZLPCjxuM3iOniBXNbt9aXxd5gqbrh7tOW8b9ZTvLI=
+github.com/hduLib/hdu v0.0.0-20220831153942-4b1d5a2b88f1/go.mod h1:rONuk9BD/HoKL6gJ8UITyl8ZEKnCU2daApH9WY6YGQc=
+github.com/teambition/rrule-go v1.7.2 h1:goEajFWYydfCgavn2m/3w5U+1b3PGqPUHx/fFSVfTy0=
+github.com/teambition/rrule-go v1.7.2/go.mod h1:mBJ1Ht5uboJ6jexKdNUJg2NcwP8uUMNvStWXlJD3MvU=
+github.com/tidwall/gjson v1.14.2 h1:6BBkirS0rAHjumnjHF6qgy5d2YAJ1TLIaFE2lzfOLqo=
+github.com/tidwall/gjson v1.14.2/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
+github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
+github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
+github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
+github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=

main.go

@@ -0,0 +1,110 @@
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/emersion/go-ical"
+	"github.com/hduLib/hdu/skl"
+	"github.com/teambition/rrule-go"
+	"log"
+	"os"
+	"time"
+)
+
+var hduTimeList = []time.Duration{
+	time.Hour*8 + time.Minute*5,
+	time.Hour*8 + time.Minute*55,
+	time.Hour*10 + time.Minute*0,
+	time.Hour*10 + time.Minute*50,
+	time.Hour*11 + time.Minute*40,
+	time.Hour*13 + time.Minute*30,
+	time.Hour*14 + time.Minute*20,
+	time.Hour*15 + time.Minute*15,
+	time.Hour*16 + time.Minute*5,
+	time.Hour*18 + time.Minute*30,
+	time.Hour*19 + time.Minute*20,
+	time.Hour*20 + time.Minute*10,
+}
+
+var weekday = []rrule.Weekday{rrule.SU, rrule.MO, rrule.TU, rrule.WE, rrule.TH, rrule.FR, rrule.SA, rrule.SU}
+
+func main() {
+	var account, passwd string
+	fmt.Println("智慧杭电账号:")
+	if _, err := fmt.Scan(&account); err != nil {
+		log.Fatalln(err)
+		return
+	}
+	fmt.Println("智慧杭电密码:")
+	if _, err := fmt.Scan(&passwd); err != nil {
+		log.Fatalln(err)
+		return
+	}
+
+	user, err := skl.Login(account, passwd)
+	if err != nil {
+		log.Fatalln(err)
+		return
+	}
+	startDay := time.Now()
+	startDay = startDay.Add(-time.Duration(int(time.Hour)*startDay.Hour() + int(time.Minute)*startDay.Minute() + int(time.Second)*startDay.Second() + startDay.Nanosecond()))
+	resp, err := user.Course(startDay)
+	if err != nil {
+		log.Fatalln(err)
+		return
+	}
+	startDay = startDay.AddDate(0, 0, -((resp.Week-1)*7 + (int(startDay.Weekday())+6)%7))
+	ics := ical.NewCalendar()
+	ics.Props.SetText(ical.PropVersion, "2.0")
+	ics.Props.SetText(ical.PropProductID, "-//student//hduLib//CN")
+	ics.Props.SetText(ical.PropCalendarScale, "GREGORIAN")
+
+	alarm := ical.NewComponent(ical.CompAlarm)
+	alarm.Props.SetText(ical.PropAction, ical.ParamDisplay)
+	alarm.Props.SetText(ical.PropDescription, "距离上课还有30min")
+	alarm.Props.SetText(ical.PropTrigger, "-P0DT0H30M0S")
+
+	for _, v := range resp.List {
+		event := ical.NewEvent()
+		event.Props.SetDateTime(ical.PropDateTimeStart, startDay.AddDate(0, 0, (v.StartWeek-1)*7+v.WeekDay-1).Add(hduTimeList[v.StartSection-1]))
+		event.Props.SetDateTime(ical.PropDateTimeEnd, startDay.AddDate(0, 0, (v.StartWeek-1)*7+v.WeekDay-1).Add(hduTimeList[v.EndSection-1]).Add(time.Minute*45))
+		event.Props.SetText(ical.PropSummary, fmt.Sprintf("%s %s %s %s", v.CourseName, v.ClassRoom, v.TeacherName, v.CourseSchema))
+		event.Props.SetDateTime(ical.PropDateTimeStamp, startDay)
+		// 我不理解，但是不这么做这个go-ical库不让我过编译
+		event.Props.SetText(ical.PropUID, "")
+		count := v.EndWeek - v.StartWeek + 1
+		interval := 1
+		if v.Period != "" {
+			count = (count + 1) / 2
+			interval = 2
+		}
+		if err != nil {
+			log.Fatalln(err)
+			return
+		}
+
+		event.Props.SetRecurrenceRule(&rrule.ROption{
+			Freq:      rrule.WEEKLY,
+			Interval:  interval,
+			Wkst:      rrule.SU,
+			Count:     count,
+			Byweekday: []rrule.Weekday{weekday[v.WeekDay]},
+		})
+
+		event.Component.Children = append(event.Component.Children, alarm)
+
+		ics.Children = append(ics.Children, event.Component)
+	}
+
+	buf := new(bytes.Buffer)
+	encoder := ical.NewEncoder(buf)
+	if err := encoder.Encode(ics); err != nil {
+		log.Fatalln(err)
+		return
+	}
+	if err := os.WriteFile("./course.ics", buf.Bytes(), os.ModePerm); err != nil {
+		log.Fatalln(err)
+		return
+	}
+
+}