comcigan.ts/src/client.ts

119 lines
3.4 KiB
TypeScript

import axios from 'axios'
import { BASE_URL, USER_AGENT } from './constants'
import DataManager from './data'
import School from './models/School'
import type Timetable from './models/Timetable'
import { encodeBase64, encodeEUCKR } from './utils/encode'
import { parseResponse } from './utils/parse'
export default class Comcigan {
private readonly rest = axios.create({
baseURL: BASE_URL,
headers: {
'User-Agent': USER_AGENT,
},
})
private readonly dataManager = new DataManager(this.rest)
/** 학교를 검색합니다. */
async searchSchools(schoolName: string): Promise<School[]> {
const { mainRoute, searchRoute } = await this.dataManager.getData()
const res = await this.rest.get(
`${mainRoute}?${searchRoute}l${encodeEUCKR(schoolName)}`,
)
const { 학교검색: data } = parseResponse<{
: [number, string, string, number][]
}>(res.data)
return data.map(
([regionCode, regionName, schoolName, schoolCode]) =>
new School(this, {
code: schoolCode,
name: schoolName,
region: { code: regionCode, name: regionName },
}),
)
}
private async getRawTimetable(
schoolCode: number,
): Promise<Timetable[][][][]> {
const {
mainRoute,
timetableRoute,
teacherCode,
originalCode,
dayCode,
subjectCode,
} = await this.dataManager.getData()
const res = await this.rest.get(
`${mainRoute}_T?${encodeBase64(`${timetableRoute}_${schoolCode}_0_1`)}`,
)
const data = parseResponse(res.data)
const teachers = data[`자료${teacherCode}`] as string[]
const teachersLen = Math.floor(Math.log10(teachers.length - 1)) + 1
const subjects = data[`자료${subjectCode}`] as string[]
const original = data[`자료${originalCode}`] as number[][][][]
const day = data[`자료${dayCode}`] as number[][][][]
return day.slice(1).map((grade, gIdx) =>
grade.slice(1).map((cls, cIdx) =>
cls.slice(1).map((day, dIdx) =>
day.slice(1).map((period, pIdx) => {
const p = period.toString()
return {
subject: subjects[Number(p.slice(0, p.length - teachersLen - 1))],
teacher: teachers[Number(p.slice(-teachersLen))],
changed:
period !== original[gIdx + 1][cIdx + 1][dIdx + 1][pIdx + 1],
}
}),
),
),
)
}
async getTimetable(schoolCode: number): Promise<Timetable[][][][]>
async getTimetable(
schoolCode: number,
grade: number,
): Promise<Timetable[][][]>
async getTimetable(
schoolCode: number,
grade: number,
cls: number,
): Promise<Timetable[][]>
async getTimetable(
schoolCode: number,
grade: number,
cls: number,
day: number,
): Promise<Timetable[]>
async getTimetable(
schoolCode: number,
grade: number,
cls: number,
day: number,
period: number,
): Promise<Timetable>
/** 학교 코드를 이용해 학교 시간표를 불러옵니다. */
async getTimetable(
schoolCode: number,
grade?: number,
cls?: number,
day?: number,
period?: number,
) {
const raw = await this.getRawTimetable(schoolCode)
if (grade === undefined) return raw
if (cls === undefined) return raw[grade - 1]
if (day === undefined) return raw[grade - 1][cls - 1]
if (period === undefined) return raw[grade - 1][cls - 1][day - 1]
return raw[grade - 1][cls - 1][day - 1][period - 1]
}
}