正方课表到 ics

课表.js

const DEFAULT_VERSION = "2.0";
const DEFAULT_PRODID = "-//TypeScript/Node.js//SCAU Course Exporter v1.0//CN";
const CRLF = "\r\n";
const SPACE = " ";
const iframe = document.querySelector("#iframeautoheight");
const PAGE = {
    "main": /xs_main.aspx/,
    "course_table": /xskbcx.aspx/
};
class Value {
    constructor(_init) {
        this.content = _init;
    }
}
class VEvent {
    constructor(evt) {
        if (evt) {
            this.uid = evt.uid;
            this.dtstamp = evt.dtstamp;
            this.organizer = evt.organizer;
            this.dtstart = evt.dtstart;
            this.dtend = evt.dtend;
            this.summary = evt.summary;
            this.categories = evt.categories;
            this._class = evt._class;
        }
    }
}
class VCalendar {
    constructor() {
        this.version = DEFAULT_VERSION;
        this.prodid = DEFAULT_PRODID;
        this.events = new Array();
    }
}
 
function isDigit(_dat) {
    let _zero = "0".charCodeAt(0);
    let _ni = "9".charCodeAt(0);
    let _code = _dat.charCodeAt(0);
    if (_code >= _zero && _code <= _ni)
        return true;
    else
        return false;
};
var Day;
(function(Day) {
    Day[Day["Monday"] = 1] = "Monday";
    Day[Day["TuesDay"] = 2] = "TuesDay";
    Day[Day["Wednesday"] = 3] = "Wednesday";
    Day[Day["ThursDay"] = 4] = "ThursDay";
    Day[Day["Friday"] = 5] = "Friday";
    Day[Day["Saturday"] = 6] = "Saturday";
    Day[Day["Sunday"] = 7] = "Sunday";
})(Day || (Day = {}));;
class Teacher {
    constructor(_name) {
        this.name = _name;
    }
};
class Course {};
 
function createCalendar(courses, startDate, dtstamp) {
    console.assert(startDate && startDate.getDay() === 1);
    if (!dtstamp)
        dtstamp = new Date();
 
    function eventFactory(course) {
        let result = new Array();
        const course_begin_table = {
            1: {
                hour: 8,
                minute: 0
            },
            2: {
                hour: 8,
                minute: 50
            },
            3: {
                hour: 9,
                minute: 55
            },
            4: {
                hour: 10,
                minute: 45
            },
            5: {
                hour: 13,
                minute: 30
            },
            6: {
                hour: 14,
                minute: 20
            },
            7: {
                hour: 15,
                minute: 25
            },
            8: {
                hour: 16,
                minute: 15
            },
            9: {
                hour: 18,
                minute: 0
            },
            10: {
                hour: 18,
                minute: 50
            },
            11: {
                hour: 19,
                minute: 55
            },
            12: {
                hour: 20,
                minute: 45
            },
        };
        const course_end_table = {
            1: {
                hour: 8,
                minute: 45
            },
            2: {
                hour: 9,
                minute: 35
            },
            3: {
                hour: 10,
                minute: 40
            },
            4: {
                hour: 11,
                minute: 30
            },
            5: {
                hour: 14,
                minute: 15
            },
            6: {
                hour: 15,
                minute: 05
            },
            7: {
                hour: 16,
                minute: 10
            },
            8: {
                hour: 17,
                minute: 00
            },
            9: {
                hour: 18,
                minute: 45
            },
            10: {
                hour: 19,
                minute: 35
            },
            11: {
                hour: 20,
                minute: 40
            },
            12: {
                hour: 21,
                minute: 30
            },
        };
        for (var week = course.schoolWeeks["from"]; week <= course.schoolWeeks["to"]; ++week) {
            var _flag = course.schoolWeeks.flag;
            if ((_flag === 1 && week % 2 === 0) ||
                (_flag === 2 && week % 2 === 1))
                continue; 
            let _evt = new VEvent();
            let _start = new Date(startDate.toString());
            let _end = new Date(startDate.toString());
            _start.setDate(_start.getDate() + (week - 1) * 7 + (course.day - 1));
            _start.setHours(course_begin_table[course.courseNumbers[0]].hour);
            _start.setMinutes(course_begin_table[course.courseNumbers[0]].minute);
            _start.setSeconds(0);
            _start.setMilliseconds(0);
            _end.setDate(_end.getDate() + (week - 1) * 7 + (course.day - 1));
            _end.setHours(course_end_table[course.courseNumbers[course.courseNumbers.length - 1]].hour);
            _end.setMinutes(course_end_table[course.courseNumbers[course.courseNumbers.length - 1]].minute);
            _end.setSeconds(0);
            _end.setMilliseconds(0);
            if (dtstamp)
                _evt.dtstamp = new Value(dtstamp);
            _evt.uid = new Value("course#" +
                course.schoolYear.from.toString() + "#" +
                course.uid.toString() + "#" +
                week.toString());
            _evt.summary = new Value(course.name +
                " (" + course.teacher.name + ")");
            _evt.location = new Value(course.location);
            _evt.dtstart = new Value(_start);
            _evt.dtend = new Value(_end);
            result.push(_evt);
        }
        return result;
    }
    var result = new VCalendar();
    for (var i = 0; i < courses.length; ++i) {
        result.events = result.events.concat(eventFactory(courses[i]));
    }
    return result;
}
 
function cookCourse(raw) {
    function parseSchoolYear(_str) {
        let number_test = /[0-9]+/g;
        let _data = _str.match(number_test);
        if (_data.length < 2)
            return null;
        else
            return {
                from: parseInt(_data[0]),
                to: parseInt(_data[1])
            };
    }
 
    function parseSchoolWeek(_timeinfo) {
        let school_week_tester = /{[^}]+}/;
        let _fuck = _timeinfo.match(school_week_tester);
        if (_fuck === null || _fuck.length < 1)
            return null;
        let info = _fuck[0].slice(1, _fuck[0].length - 1); // get the info in the brackets
        let _data = {
            from: 0,
            to: 0,
            flag: 0
        };
        let _single = info.match("单周");
        let _double = info.match("双周");
        if (_single && _single.length > 0)
            _data.flag = 1;
        else if (_double && _double.length > 0)
            _data.flag = 2;
        else
            _data.flag = 0;
        let _number_test = /[0-9]+/g;
        let _dat = info.match(_number_test);
        if (_dat && _dat.length >= 2) {
            _data["from"] = parseInt(_dat[0]);
            _data["to"] = parseInt(_dat[1]);
        }
        return _data;
    }
 
    function parseCourseNumber(timeinfo) {
        let _bracket_index = timeinfo.indexOf("{");
        let _slice = timeinfo.slice(0, _bracket_index);
        let _number_test = /[0-9]+/g;
        let _dat = _slice.match(_number_test);
        let result = new Array();
        _dat.forEach((value) => {
            result.push(parseInt(value));
        });
        return result;
    }
 
    function parseDay(_timeinfo) {
        var word = _timeinfo[1];
        switch (word) {
            case "一":
                return Day.Monday;
            case "二":
                return Day.TuesDay;
            case "三":
                return Day.Wednesday;
            case "四":
                return Day.ThursDay;
            case "五":
                return Day.Friday;
            case "六":
                return Day.Saturday;
            case "日":
                return Day.Sunday;
            default:
                return null;
        }
    }
    let raw_co = raw.courses;
    let result = new Array();
    let _term = parseInt(raw.term);
    for (var i = 0; i < raw_co.length; ++i) {
        var course = new Course();
        course.uid = i;
        course.name = raw_co[i].name;
        course.term = parseInt(raw.term);
        course.schoolYear = parseSchoolYear(raw.schoolYear);
        course.schoolWeeks = parseSchoolWeek(raw_co[i].timeinfo);
        course.location = raw_co[i].location;
        course.teacher = new Teacher(raw_co[i].teacher_name);
        course.courseNumbers = parseCourseNumber(raw_co[i].timeinfo);
        course.day = parseDay(raw_co[i].timeinfo);
        result.push(course);
    }
    return result;
}
 
function exportCourse(doc) {
    function tryParse(td) {
        var str = td.innerHTML;
        const rowspan = td.rowSpan;
        let result = [];
        if (str && str.length > 0) {
            str = str.replace(/<br>/g, "\n");
            var _data = str.split("\n");
            if (_data.length >= 3) {
                while (_data.length > 0) {
                    if (_data[0] == "") {
                        _data.shift();
                        continue;
                    }
                    var data = {
                        name: _data.shift(),
                        timeinfo: _data.shift(),
                        teacher_name: _data.shift(),
                        quantity: 1
                    };
                    if (_data.length > 0 || _data[0] != "") {
                        data.location = _data.shift();;
                    }
                    result.push(data);
                }
            }
            return result;
        }
        return null;
    }
    var result = new Array();
    const table = doc.querySelector('#Table1');
    const tds = table.querySelectorAll('td');
    for (var i = 0; i < tds.length; ++i) {
        var _td = tds[i];
        var re = tryParse(_td);
        if (re)
            result = result.concat(re);
    }
    console.log(result);
    return result;
}
var ENDL = "\n";
 
function genDate(_d) {
    function getFixedLen(s, len) {
        if (s.length < len) {
            return getFixedLen("0" + s, len);
        } else if (s.length > len) {
            return s.slice(0, len);
        } else
            return s;
    }
    var result = "";
    result += getFixedLen(_d.getUTCFullYear().toString(), 4);
    result += getFixedLen((_d.getUTCMonth() + 1).toString(), 2);
    result += getFixedLen(_d.getUTCDate().toString(), 2);
    result += "T";
    result += getFixedLen(_d.getUTCHours().toString(), 2);
    result += getFixedLen(_d.getUTCMinutes().toString(), 2);
    result += getFixedLen(_d.getUTCSeconds().toString(), 2);
    result += "Z";
    return result;
}
 
function iCalGen(obj) {
    var result = new Array();
    if (obj instanceof VEvent) {
        var _evt = obj;
        result.push("BEGIN:VEVENT");
        result.push("UID:" + _evt.uid.content);
        result.push("DTSTAMP:" + genDate(_evt.dtstamp.content));
        if (_evt.organizer)
            result.push("ORGANIZER:" + _evt.organizer.content);
        if (_evt.location)
            result.push("LOCATION:" + _evt.location.content);
        result.push("DTSTART:" + genDate(_evt.dtstart.content));
        result.push("DTEND:" + genDate(_evt.dtend.content));
        result.push("SUMMARY:" + _evt.summary.content);
        result.push("END:VEVENT");
    } else if (obj instanceof VCalendar) {
        var _cal = obj;
        result.push("BEGIN:VCALENDAR");
        result.push("VERSION:" + _cal.version);
        result.push("PRODID:" + _cal.prodid);
        for (var i = 0; i < _cal.events.length; ++i)
            result = result.concat(iCalGen(_cal.events[i]));
        result.push("END:VCALENDAR");
    }
    return result;
};
 
function genFinal(init) {
    var result = "";
    for (var i = 0; i < init.length; ++i) {
        var finalLine = "";
        var line = init[i];
        var len = line.length;
        if (len > 60) {
            var remain_len = len;
            var index = 0;
            while (remain_len > 0) {
                for (var i = 0; i < index; ++i)
                    result += SPACE;
                result += line.slice(0, 60);
                line = line.slice(61);
                remain_len -= 60;
                index++;
            }
            line.slice(0, 60);
        } else
            finalLine = line;
        result += line + CRLF;
    }
    return result;
}
 
function calendarGen(cal) {
    return genFinal(iCalGen(cal));
}
if (iframe) {
    iframe.addEventListener('load', function() {
        const content = iframe.contentDocument || iframe.contentWindow.document;
        const URL = content.location.href;
        if (PAGE['course_table'].test(URL)) {
            const schoolyear_select = content.querySelector('#xnd');
            const term_select = content.querySelector('#xqd');
            const inject_td = content
                .querySelector('#Table2 > tbody > tr:nth-child(1) > td:nth-child(1)');
            const inject_p = content.createElement("p");
            const export_span = content.createElement("span");
            const export_first_day = content.createElement('input');
            const export_btn = content.createElement('button');
            const dwn_anchor = content.createElement("a");
            dwn_anchor.innerHTML = "点击这儿下载ics日历文件";
            dwn_anchor.style.visibility = "hidden";
            inject_p.appendChild(export_span);
            inject_p.appendChild(export_first_day);
            inject_p.appendChild(export_btn);
            inject_p.appendChild(dwn_anchor);
            inject_td.appendChild(content.createElement("br"));
            inject_td.appendChild(inject_p);
            export_span.innerHTML = "   这个学期的第一个星期一是：";
            export_first_day.type = "date";
            export_first_day.value = "2018-09-10";
            export_btn.innerText = "导出课表";
            export_btn.addEventListener('click', (evt) => {
                evt.preventDefault();
                document.querySelector('iframe').contentDocument.querySelector('form').querySelectorAll('font').forEach(node => node.remove())
                var raw_courses = exportCourse(content);
                var data = {
                    courses: raw_courses,
                    schoolYear: schoolyear_select.value,
                    term: term_select.value
                };
                var courses = cookCourse(data);
                var first_day = new Date(export_first_day.value);
                var cal = createCalendar(courses, first_day);
                var result = calendarGen(cal);
                var link = window.URL.createObjectURL(new Blob([result], {
                    type: "text/x-vCalendar"
                }));
                dwn_anchor.style.visibility = "visible";
                dwn_anchor.setAttribute("href", link);
                dwn_anchor.setAttribute("download", "我的课程表.ics");
            });
        }
    });
}