diff --git a/cmd/huashijie_work_go/huashijie_work_go.go b/cmd/huashijie_work_go/huashijie_work_go.go index bf6fd86..05e8db2 100644 --- a/cmd/huashijie_work_go/huashijie_work_go.go +++ b/cmd/huashijie_work_go/huashijie_work_go.go @@ -175,6 +175,8 @@ func main() { go InterruptHandler() go ShowStatus(tracker) + huashijie_api.EnsureConnection(*tracker.HTTP_client) + Logger.Println("-- Start --") for i := 0; i < BASE_CONCURRENCY; i++ { diff --git a/pkg/huashijie_api.go b/pkg/huashijie_api.go index f958399..1767bb0 100644 --- a/pkg/huashijie_api.go +++ b/pkg/huashijie_api.go @@ -5,6 +5,7 @@ import ( "io" "math/rand" "net/http" + "strings" ) var XIAOMI_MODELS = []string{ @@ -21,6 +22,32 @@ var XIAOMI_MODELS = []string{ "MDE5", } +func EnsureConnection(client http.Client) { + // https://app.huashijie.art/api/update/checkUpdate + req, err := http.NewRequest("GET", "https://app.huashijie.art/api/update/checkUpdate", nil) + if err != nil { + panic(err) + } + req.Header.Add("Referer", "https://app.huashijie.art/") + req.Header.Add("User-Agent", "Dalvik/2.1.0 (Linux; U; Android 9; MI 8 SE MIUI/V12.0.2.0.PEBCNXM)") + + resp, err := client.Do(req) + + if err != nil { + panic(err) + } + defer resp.Body.Close() + body, err := io.ReadAll(resp.Body) + if err != nil { + panic(err) + } + text := string(body) + fmt.Println(text) + if !strings.Contains(text, "update_ver_code") { + panic("NotImplementedError: " + text) + } +} + func GetWorkDetailResponse(client http.Client, work_id string) ([]byte, int) { req, err := http.NewRequest("GET", "https://app.huashijie.art/api/work/detailV2", nil) if err != nil {