diff --git a/assets/main.db b/assets/main.db index ef65588..6a2457e 100644 Binary files a/assets/main.db and b/assets/main.db differ diff --git a/handler/album.go b/handler/album.go index cf9204c..f225d44 100644 --- a/handler/album.go +++ b/handler/album.go @@ -19,9 +19,7 @@ import ( func AlbumSeriesAllHandler(ctx *gin.Context) { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() reqTime := time.Now().Unix() @@ -58,34 +56,24 @@ func AlbumSeriesAllHandler(ctx *gin.Context) { // sql := `SELECT album_series_id FROM album_series_m` seriesRows, err := db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) albumSeriesAllResp := []model.AlbumSeriesResp{} for seriesRows.Next() { var series int err = seriesRows.Scan(&series) - if err != nil { - panic(err) - } + CheckErr(err) albumSeriesAll := []model.AlbumResult{} stmt, err := db.Prepare("SELECT unit_id,rarity FROM unit_m WHERE album_series_id = ?") - if err != nil { - panic(err) - } + CheckErr(err) unitRows, err := stmt.Query(series) - if err != nil { - panic(err) - } + CheckErr(err) for unitRows.Next() { var unitId, rarity int err = unitRows.Scan(&unitId, &rarity) - if err != nil { - panic(err) - } + CheckErr(err) albumSeries := model.AlbumResult{ UnitID: unitId, @@ -118,20 +106,14 @@ func AlbumSeriesAllHandler(ctx *gin.Context) { // IsSigned stmt, err = db.Prepare("SELECT COUNT(*) AS ct FROM unit_sign_asset_m WHERE unit_id = ?") - if err != nil { - panic(err) - } + CheckErr(err) signRows, err := stmt.Query(unitId) - if err != nil { - panic(err) - } + CheckErr(err) ct := 0 for signRows.Next() { err = signRows.Scan(&ct) - if err != nil { - panic(err) - } + CheckErr(err) } if ct > 0 { albumSeries.SignFlag = true @@ -148,18 +130,14 @@ func AlbumSeriesAllHandler(ctx *gin.Context) { }) } rb, err := json.Marshal(albumSeriesAllResp) - if err != nil { - panic(err) - } + CheckErr(err) resp := model.Response{ ResponseData: rb, ReleaseInfo: []interface{}{}, StatusCode: 200, } respb, err := json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(respb)) xms := encrypt.RSA_Sign_SHA1(respb, "privatekey.pem") diff --git a/handler/announce.go b/handler/announce.go index 63b7297..aef189e 100644 --- a/handler/announce.go +++ b/handler/announce.go @@ -73,9 +73,7 @@ func AnnounceCheckStateHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(announceResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/api.go b/handler/api.go index 202dffb..15d246b 100644 --- a/handler/api.go +++ b/handler/api.go @@ -148,36 +148,26 @@ func ApiHandler(ctx *gin.Context) { err = errors.New("invalid option") } - if err != nil { - panic(err) - } + CheckErr(err) res, err = database.LevelDb.Get([]byte(key)) - if err != nil { - panic(err) - } + CheckErr(err) var result interface{} err = json.Unmarshal([]byte(res), &result) - if err != nil { - panic(err) - } + CheckErr(err) results = append(results, result) } // fmt.Println(results) b, err := json.Marshal(results) - if err != nil { - panic(err) - } + CheckErr(err) rp := model.Response{ ResponseData: b, ReleaseInfo: []interface{}{}, StatusCode: 200, } b, err = json.Marshal(rp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(b)) authorizeStr := ctx.Request.Header["Authorize"] diff --git a/handler/download.go b/handler/download.go index 05f3ed3..b827059 100644 --- a/handler/download.go +++ b/handler/download.go @@ -19,9 +19,7 @@ import ( func DownloadAdditionalHandler(ctx *gin.Context) { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() reqTime := time.Now().Unix() @@ -63,19 +61,13 @@ func DownloadAdditionalHandler(ctx *gin.Context) { if CdnUrl != "" { pkgType, pkgId := downloadReq.PackageType, downloadReq.PackageID stmt, err := db.Prepare("SELECT pkg_order,pkg_size FROM download_db WHERE pkg_type = ? AND pkg_id = ? ORDER BY pkg_order ASC") - if err != nil { - panic(err) - } + CheckErr(err) rows, err := stmt.Query(pkgType, pkgId) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { var pkgOrder, pkgSize int err = rows.Scan(&pkgOrder, &pkgSize) - if err != nil { - panic(err) - } + CheckErr(err) pkgList = append(pkgList, model.AdditionalResult{ Size: pkgSize, URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), @@ -89,9 +81,7 @@ func DownloadAdditionalHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(addResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -104,9 +94,7 @@ func DownloadAdditionalHandler(ctx *gin.Context) { func DownloadBatchHandler(ctx *gin.Context) { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() reqTime := time.Now().Unix() @@ -156,19 +144,13 @@ func DownloadBatchHandler(ctx *gin.Context) { } // fmt.Println(exPkgId) stmt, err := db.Prepare("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = ? " + exPkgId + " ORDER BY pkg_id ASC, pkg_order ASC") - if err != nil { - panic(err) - } + CheckErr(err) rows, err := stmt.Query(pkgType) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { var pkgId, pkgOrder, pkgSize int err = rows.Scan(&pkgId, &pkgOrder, &pkgSize) - if err != nil { - panic(err) - } + CheckErr(err) pkgList = append(pkgList, model.BatchResult{ Size: pkgSize, URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), @@ -182,9 +164,7 @@ func DownloadBatchHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(batchResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -197,9 +177,7 @@ func DownloadBatchHandler(ctx *gin.Context) { func DownloadUpdateHandler(ctx *gin.Context) { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() reqTime := time.Now().Unix() @@ -241,15 +219,11 @@ func DownloadUpdateHandler(ctx *gin.Context) { if downloadReq.ExternalVersion != PackageVersion && CdnUrl != "" { pkgType := 99 rows, err := db.Query("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = 99 ORDER BY pkg_id ASC, pkg_order ASC") - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { var pkgId, pkgOrder, pkgSize int err = rows.Scan(&pkgId, &pkgOrder, &pkgSize) - if err != nil { - panic(err) - } + CheckErr(err) pkgList = append(pkgList, model.UpdateResult{ Size: pkgSize, URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), @@ -264,9 +238,7 @@ func DownloadUpdateHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(updateResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -327,9 +299,7 @@ func DownloadUrlHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(urlResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -378,9 +348,7 @@ func DownloadEventHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(eventResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/event.go b/handler/event.go index 1bb109c..f676e74 100644 --- a/handler/event.go +++ b/handler/event.go @@ -78,9 +78,7 @@ func EventListHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(eventsResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/gdpr.go b/handler/gdpr.go index 8d54475..c412e8d 100644 --- a/handler/gdpr.go +++ b/handler/gdpr.go @@ -68,9 +68,7 @@ func GdprHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(gdprResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/global.go b/handler/global.go index 3dc7100..78fae7d 100644 --- a/handler/global.go +++ b/handler/global.go @@ -14,3 +14,9 @@ func init() { CdnUrl = config.Conf.Cdn.CdnUrl } } + +func CheckErr(err error) { + if err != nil { + panic(err) + } +} diff --git a/handler/lbonus.go b/handler/lbonus.go index 04220d1..d7173a4 100644 --- a/handler/lbonus.go +++ b/handler/lbonus.go @@ -185,9 +185,7 @@ func LBonusExecuteHandler(ctx *gin.Context) { } resp, err := json.Marshal(res) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/live.go b/handler/live.go index 8e0205e..daa6da2 100644 --- a/handler/live.go +++ b/handler/live.go @@ -72,9 +72,7 @@ func PlayLiveHandler(ctx *gin.Context) { playReq := model.PlayReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -106,34 +104,26 @@ func PlayLiveHandler(ctx *gin.Context) { // fmt.Println(newAuthorizeStr) db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() db.SetMaxOpenConns(1) tDifficultyId := playReq.LiveDifficultyID difficultyId, err := strconv.Atoi(tDifficultyId) - if err != nil { - panic(err) - } + CheckErr(err) deckId := playReq.UnitDeckID // Song type: normal / special // sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here. sql := `SELECT notes_list,c_rank_score,b_rank_score,a_rank_score,s_rank_score,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)` rows, err := db.Query(sql, difficultyId, difficultyId) - if err != nil { - panic(err) - } + CheckErr(err) var notes_list string var c_rank_score, b_rank_score, a_rank_score, s_rank_score, ac_flag, swing_flag int for rows.Next() { err = rows.Scan(¬es_list, &c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &ac_flag, &swing_flag) - if err != nil { - panic(err) - } + CheckErr(err) } // fmt.Println(len(notes_list)) @@ -141,9 +131,7 @@ func PlayLiveHandler(ctx *gin.Context) { notes := []model.NotesList{} err = json.Unmarshal([]byte(notes_list), ¬es) - if err != nil { - panic(err) - } + CheckErr(err) ranks := []model.RankInfo{} ranks = append(ranks, model.RankInfo{ @@ -211,9 +199,7 @@ func PlayLiveHandler(ctx *gin.Context) { } m, err := json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) res := model.Response{ ResponseData: m, @@ -222,9 +208,7 @@ func PlayLiveHandler(ctx *gin.Context) { } mm, err := json.Marshal(res) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -274,9 +258,7 @@ func GameOverHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(overResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -292,9 +274,7 @@ func PlayScoreHandler(ctx *gin.Context) { playScoreReq := model.PlayScoreReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playScoreReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -326,33 +306,25 @@ func PlayScoreHandler(ctx *gin.Context) { // fmt.Println(newAuthorizeStr) db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() db.SetMaxOpenConns(1) tDifficultyId := playScoreReq.LiveDifficultyID difficultyId, err := strconv.Atoi(tDifficultyId) - if err != nil { - panic(err) - } + CheckErr(err) // Song type: normal / special // sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here. sql := `SELECT notes_list,c_rank_score,b_rank_score,a_rank_score,s_rank_score,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)` rows, err := db.Query(sql, difficultyId, difficultyId) - if err != nil { - panic(err) - } + CheckErr(err) var notes_list string var c_rank_score, b_rank_score, a_rank_score, s_rank_score, ac_flag, swing_flag int for rows.Next() { err = rows.Scan(¬es_list, &c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &ac_flag, &swing_flag) - if err != nil { - panic(err) - } + CheckErr(err) } // fmt.Println(len(notes_list)) @@ -360,9 +332,7 @@ func PlayScoreHandler(ctx *gin.Context) { notes := []model.NotesList{} err = json.Unmarshal([]byte(notes_list), ¬es) - if err != nil { - panic(err) - } + CheckErr(err) ranks := []model.RankInfo{} ranks = append(ranks, model.RankInfo{ @@ -414,9 +384,7 @@ func PlayScoreHandler(ctx *gin.Context) { } m, err := json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) res := model.Response{ ResponseData: m, @@ -425,9 +393,7 @@ func PlayScoreHandler(ctx *gin.Context) { } mm, err := json.Marshal(res) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(mm)) xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem") @@ -445,9 +411,7 @@ func PlayRewardHandler(ctx *gin.Context) { playRewardReq := model.PlayRewardReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &playRewardReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -479,9 +443,7 @@ func PlayRewardHandler(ctx *gin.Context) { // fmt.Println(newAuthorizeStr) db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() db.SetMaxOpenConns(1) @@ -491,27 +453,19 @@ func PlayRewardHandler(ctx *gin.Context) { // sqlite3 doesn't support FULL OUTER JOIN so use UNION ALL here. sql := `SELECT c_rank_score,b_rank_score,a_rank_score,s_rank_score,c_rank_combo,b_rank_combo,a_rank_combo,s_rank_combo,ac_flag,swing_flag FROM live_setting_m WHERE live_setting_id IN (SELECT live_setting_id FROM normal_live_m WHERE live_difficulty_id = ? UNION ALL SELECT live_setting_id FROM special_live_m WHERE live_difficulty_id = ?)` rows, err := db.Query(sql, difficultyId, difficultyId) - if err != nil { - panic(err) - } + CheckErr(err) var c_rank_score, b_rank_score, a_rank_score, s_rank_score, c_rank_combo, b_rank_combo, a_rank_combo, s_rank_combo, ac_flag, swing_flag int for rows.Next() { err = rows.Scan(&c_rank_score, &b_rank_score, &a_rank_score, &s_rank_score, &c_rank_combo, &b_rank_combo, &a_rank_combo, &s_rank_combo, &ac_flag, &swing_flag) - if err != nil { - panic(err) - } + CheckErr(err) } unitsList := []model.PlayRewardUnitList{} deck, err := database.LevelDb.Get([]byte("deck_info")) - if err != nil { - panic(err) - } + CheckErr(err) err = json.Unmarshal(deck, &unitsList) - if err != nil { - panic(err) - } + CheckErr(err) totalScore := playRewardReq.ScoreSmile + playRewardReq.ScoreCool + playRewardReq.ScoreCute resp := model.RewardResponseData{ @@ -662,9 +616,7 @@ func PlayRewardHandler(ctx *gin.Context) { } m, err := json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) res := model.Response{ ResponseData: m, @@ -673,9 +625,7 @@ func PlayRewardHandler(ctx *gin.Context) { } mm, err := json.Marshal(res) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(mm)) xms := encrypt.RSA_Sign_SHA1(mm, "privatekey.pem") diff --git a/handler/login.go b/handler/login.go index f39a061..63d0625 100644 --- a/handler/login.go +++ b/handler/login.go @@ -20,21 +20,15 @@ func AuthKeyHandler(ctx *gin.Context) { authReq := model.AuthKeyReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &authReq) - if err != nil { - panic(err) - } + CheckErr(err) dummyToken64, err := base64.StdEncoding.DecodeString(authReq.DummyToken) - if err != nil { - panic(err) - } + CheckErr(err) dummyTokenDecrypted := encrypt.RSA_Decrypt(dummyToken64, "privatekey.pem") // aesKey := dummyTokenDecrypted[0:16] // data64, err := base64.StdEncoding.DecodeString(authReq.AuthData) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // dataDecrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(data64, aesKey)) // fmt.Println(string(dataDecrypted)) @@ -51,22 +45,16 @@ func AuthKeyHandler(ctx *gin.Context) { "server_token": serverToken, } authJson, err := json.Marshal(authData) - if err != nil { - panic(err) - } + CheckErr(err) database.LevelDb.Put([]byte(authorizeToken), authJson) - if err != nil { - panic(err) - } + CheckErr(err) authResp := model.AuthKeyResp{} authResp.ResponseData.DummyToken = serverToken authResp.ResponseData.AuthorizeToken = authorizeToken authResp.StatusCode = 200 resp, err := json.Marshal(authResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) respTime := time.Now().Unix() @@ -94,39 +82,27 @@ func LoginHandler(ctx *gin.Context) { // authData, err := database.RedisCli.HGetAll(database.RedisCtx, authToken).Result() // authData, err := database.LevelDb.Get([]byte(authToken)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(authData) // clientToken, serverToken := authData["client_token"], authData["server_token"] // clientToken := gjson.Get(string(authData), "client_token").String() // serverToken := gjson.Get(string(authData), "server_token").String() // clientToken64, err := base64.RawStdEncoding.DecodeString(clientToken) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // serverToken64, err := base64.RawStdEncoding.DecodeString(serverToken) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // xmcKey := utils.SliceXor([]byte(clientToken64), []byte(serverToken64)) // aesKey := xmcKey[0:16] loginReq := model.LoginReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &loginReq) - if err != nil { - panic(err) - } + CheckErr(err) // key64, err := base64.StdEncoding.DecodeString(loginReq.LoginKey) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // pass64, err := base64.StdEncoding.DecodeString(loginReq.LoginPasswd) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // keyDescrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(key64, aesKey)) // fmt.Println(string(keyDescrypted)) // passDescrypted := utils.Sub16(encrypt.AES_CBC_Decrypt(pass64, aesKey)) @@ -151,9 +127,7 @@ func LoginHandler(ctx *gin.Context) { // _, err = database.RedisCli.HSet(database.RedisCtx, "token_uid", authorizeToken, userId).Result() err = database.LevelDb.Put([]byte(authorizeToken), []byte(sUserId)) - if err != nil { - panic(err) - } + CheckErr(err) loginResp := model.LoginResp{} loginResp.ResponseData.AuthorizeToken = authorizeToken @@ -162,9 +136,7 @@ func LoginHandler(ctx *gin.Context) { loginResp.ResponseData.AdultFlag = 2 loginResp.StatusCode = 200 resp, err := json.Marshal(loginResp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(resp)) respTime := time.Now().Unix() diff --git a/handler/multiunit.go b/handler/multiunit.go index e46a0a3..5078baa 100644 --- a/handler/multiunit.go +++ b/handler/multiunit.go @@ -40,9 +40,7 @@ func MultiUnitStartUpHandler(ctx *gin.Context) { startReq := MultiUnitStartUpReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -83,9 +81,7 @@ func MultiUnitStartUpHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(startResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/notice.go b/handler/notice.go index 420e9b5..d6963d3 100644 --- a/handler/notice.go +++ b/handler/notice.go @@ -93,9 +93,7 @@ func NoticeFriendVarietyHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(noticeResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -148,9 +146,7 @@ func NoticeFriendGreetingHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(noticeResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -204,9 +200,7 @@ func NoticeUserGreetingHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(noticeResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/payment.go b/handler/payment.go index 625e189..8cb1d1b 100644 --- a/handler/payment.go +++ b/handler/payment.go @@ -91,9 +91,7 @@ func ProductListHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(prodReesp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/personalnotice.go b/handler/personalnotice.go index 14a7d96..5403e2b 100644 --- a/handler/personalnotice.go +++ b/handler/personalnotice.go @@ -74,9 +74,7 @@ func PersonalNoticeHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(noticeResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/private.go b/handler/private.go index 5df2745..72275b3 100644 --- a/handler/private.go +++ b/handler/private.go @@ -37,33 +37,25 @@ func PublicKeyHandler(ctx *gin.Context) { func HandshakeHandler(ctx *gin.Context) { body, err := io.ReadAll(ctx.Request.Body) - if err != nil { - panic(err) - } + CheckErr(err) defer ctx.Request.Body.Close() // fmt.Println(string(body)) body64, err := base64.StdEncoding.DecodeString(string(body)) - if err != nil { - panic(err) - } + CheckErr(err) decryptedBody := encrypt.RSA_Decrypt(body64, "privatekey.pem") // fmt.Println(decryptedBody) // fmt.Println(string(decryptedBody)) params, err := url.ParseQuery(string(decryptedBody)) - if err != nil { - panic(err) - } + CheckErr(err) randKey = params.Get("randkey") // fmt.Println(randKey) token := strings.ToUpper(utils.RandomStr(33)) token = fmt.Sprintf(`{"message":"ok","result":0,"token":"%s"}`, token) encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING) - if err != nil { - panic(err) - } + CheckErr(err) encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken) // fmt.Println(encryptedToken64) @@ -75,22 +67,16 @@ func HandshakeHandler(ctx *gin.Context) { func InitializeHandler(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) // body64, err := base64.StdEncoding.DecodeString(string(body)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(body64)) // decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(decryptedBody)) data := `{ @@ -129,9 +115,7 @@ func InitializeHandler(ctx *gin.Context) { "weixin_key":"" }` encryptedToken, err := openssl.Des3ECBEncrypt([]byte(data), []byte(randKey)[0:24], openssl.PKCS7_PADDING) - if err != nil { - panic(err) - } + CheckErr(err) encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken) // fmt.Println(encryptedToken64) @@ -149,30 +133,22 @@ func GetCodeHandler(ctx *gin.Context) { func LoginAutoHandler(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) // body64, err := base64.StdEncoding.DecodeString(string(body)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(body64)) // decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(decryptedBody)) // Unable to decrypt server data token := `{"message":"Hello, world!"}` encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING) - if err != nil { - panic(err) - } + CheckErr(err) encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken) // fmt.Println(encryptedToken64) @@ -194,22 +170,16 @@ func LoginAreaHandler(ctx *gin.Context) { func AccountLoginHandler(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) // body64, err := base64.StdEncoding.DecodeString(string(body)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(body64)) // decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(decryptedBody)) // userid: 实际登录用的账号 @@ -241,9 +211,7 @@ func AccountLoginHandler(ctx *gin.Context) { "userid":"2592007136" }` encryptedToken, err := openssl.Des3ECBEncrypt([]byte(data), []byte(randKey)[0:24], openssl.PKCS7_PADDING) - if err != nil { - panic(err) - } + CheckErr(err) encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken) // fmt.Println(encryptedToken64) @@ -255,36 +223,26 @@ func AccountLoginHandler(ctx *gin.Context) { func ReportRoleHandler(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) // body64, err := base64.StdEncoding.DecodeString(string(body)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(body64)) // decryptedBody, err := openssl.Des3ECBDecrypt(body64, []byte(randKey)[0:24], openssl.PKCS7_PADDING) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(string(decryptedBody)) // decrypted, err := url.QueryUnescape(string(decryptedBody)) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // fmt.Println(decrypted) // Unable to decrypt server data token := `{"message":"Hello, world!"}` encryptedToken, err := openssl.Des3ECBEncrypt([]byte(token), []byte(randKey)[0:24], openssl.PKCS7_PADDING) - if err != nil { - panic(err) - } + CheckErr(err) encryptedToken64 := base64.StdEncoding.EncodeToString(encryptedToken) // fmt.Println(encryptedToken64) @@ -302,9 +260,7 @@ func GetProductListHandler(ctx *gin.Context) { func ReportLog(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) ctx.Header("Content-Type", "text/html;charset=utf-8") @@ -313,9 +269,7 @@ func ReportLog(ctx *gin.Context) { func ReportApp(ctx *gin.Context) { // body, err := io.ReadAll(ctx.Request.Body) - // if err != nil { - // panic(err) - // } + // CheckErr(err) // defer ctx.Request.Body.Close() // fmt.Println(string(body)) resp := `{ "code": 0, "msg": "", "data": { "needReport": 0 } }` diff --git a/handler/scenario.go b/handler/scenario.go index cde7299..99cd41d 100644 --- a/handler/scenario.go +++ b/handler/scenario.go @@ -40,9 +40,7 @@ func ScenarioStartupHandler(ctx *gin.Context) { startReq := ScenarioReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -83,9 +81,7 @@ func ScenarioStartupHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(startResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/subscenario.go b/handler/subscenario.go index d0dbe2e..32e4649 100644 --- a/handler/subscenario.go +++ b/handler/subscenario.go @@ -40,9 +40,7 @@ func SubScenarioStartupHandler(ctx *gin.Context) { startReq := SubScenarioReq{} err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq) - if err != nil { - panic(err) - } + CheckErr(err) authorizeStr := ctx.Request.Header["Authorize"] authToken, err := utils.GetAuthorizeToken(authorizeStr) @@ -83,9 +81,7 @@ func SubScenarioStartupHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(startResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/tos.go b/handler/tos.go index b62917d..c90fd45 100644 --- a/handler/tos.go +++ b/handler/tos.go @@ -69,9 +69,7 @@ func TosCheckHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(tosResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/unit.go b/handler/unit.go index 20b500e..56268eb 100644 --- a/handler/unit.go +++ b/handler/unit.go @@ -58,9 +58,7 @@ func SetDisplayRankHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(dispResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/user.go b/handler/user.go index 3a47c99..67eeee2 100644 --- a/handler/user.go +++ b/handler/user.go @@ -58,9 +58,7 @@ func SetNotificationTokenHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(notifResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) @@ -109,9 +107,7 @@ func ChangeNaviHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(notifResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/handler/userinfo.go b/handler/userinfo.go index e1d9a0e..3a111af 100644 --- a/handler/userinfo.go +++ b/handler/userinfo.go @@ -141,9 +141,7 @@ func UserInfoHandler(ctx *gin.Context) { StatusCode: 200, } resp, err := json.Marshal(userResp) - if err != nil { - panic(err) - } + CheckErr(err) xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem") xms64 := base64.RawStdEncoding.EncodeToString(xms) diff --git a/tools/api.go b/tools/api.go index 817e474..ea17415 100644 --- a/tools/api.go +++ b/tools/api.go @@ -12,13 +12,10 @@ func LoadApi1Data(path string) { if apiData != "" { var obj model.Response err := json.Unmarshal([]byte(apiData), &obj) - if err != nil { - panic(err) - } + CheckErr(err) var data interface{} - err = json.Unmarshal(obj.ResponseData, &data) - if err != nil { + if err = json.Unmarshal(obj.ResponseData, &data); err != nil { panic(err) } // resultType := reflect.TypeOf(data) @@ -28,9 +25,7 @@ func LoadApi1Data(path string) { result := data.([]interface{}) for k, v := range result { m, err := json.Marshal(v) - if err != nil { - panic(err) - } + CheckErr(err) key := "" switch k { @@ -91,9 +86,7 @@ func LoadApi1Data(path string) { if key != "" { // database.RedisCli.HSet(database.RedisCtx, "temp_dataset", key, string(m)) err = database.LevelDb.Put([]byte(key), m) - if err != nil { - panic(err) - } + CheckErr(err) } } } @@ -104,15 +97,11 @@ func LoadApi2Data(path string) { if apiData != "" { var obj model.Response err := json.Unmarshal([]byte(apiData), &obj) - if err != nil { - panic(err) - } + CheckErr(err) var data interface{} err = json.Unmarshal(obj.ResponseData, &data) - if err != nil { - panic(err) - } + CheckErr(err) // resultType := reflect.TypeOf(data) // if resultType.Kind() == reflect.Map { // data = data.(map[string]interface{}) @@ -120,9 +109,7 @@ func LoadApi2Data(path string) { result := data.([]interface{}) for k, v := range result { m, err := json.Marshal(v) - if err != nil { - panic(err) - } + CheckErr(err) key := "" switch k { @@ -139,9 +126,7 @@ func LoadApi2Data(path string) { if key != "" { // database.RedisCli.HSet(database.RedisCtx, "temp_dataset", key, string(m)) err = database.LevelDb.Put([]byte(key), m) - if err != nil { - panic(err) - } + CheckErr(err) } } } @@ -152,15 +137,11 @@ func LoadApi3Data(path string) { if apiData != "" { var obj model.Response err := json.Unmarshal([]byte(apiData), &obj) - if err != nil { - panic(err) - } + CheckErr(err) var data interface{} err = json.Unmarshal(obj.ResponseData, &data) - if err != nil { - panic(err) - } + CheckErr(err) // resultType := reflect.TypeOf(data) // if resultType.Kind() == reflect.Map { // data = data.(map[string]interface{}) @@ -168,9 +149,7 @@ func LoadApi3Data(path string) { result := data.([]interface{}) for k, v := range result { m, err := json.Marshal(v) - if err != nil { - panic(err) - } + CheckErr(err) key := "" switch k { @@ -185,9 +164,7 @@ func LoadApi3Data(path string) { if key != "" { // database.RedisCli.HSet(database.RedisCtx, "temp_dataset", key, string(m)) err = database.LevelDb.Put([]byte(key), m) - if err != nil { - panic(err) - } + CheckErr(err) } } } diff --git a/tools/download_db.go b/tools/download_db.go index 863a551..89e6c07 100644 --- a/tools/download_db.go +++ b/tools/download_db.go @@ -20,43 +20,31 @@ func GenDownloadDb() { // PRIMARY KEY ("id") // )` db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() fileLists, err := os.ReadDir("F:/sif_dl/list_CN_Android") - if err != nil { - panic(err) - } + CheckErr(err) for _, v := range fileLists { if v.IsDir() { panic(err) } fileList := "F:/sif_dl/list_CN_Android/" + v.Name() fileStat, err := os.Stat(fileList) - if err != nil { - panic(err) - } + CheckErr(err) pkgSize := fileStat.Size() fileInfo := strings.Split(strings.ReplaceAll(v.Name(), ".zip", ""), "_") pkgType, pkgId, pkgOrder := fileInfo[0], fileInfo[1], fileInfo[2] fmt.Printf("%s - %s - %s - %d\n", pkgType, pkgId, pkgOrder, pkgSize) stmt, err := db.Prepare("INSERT INTO download_db(pkg_type,pkg_id,pkg_order,pkg_size) VALUES (?,?,?,?)") - if err != nil { - panic(err) - } + CheckErr(err) res, err := stmt.Exec(pkgType, pkgId, pkgOrder, pkgSize) - if err != nil { - panic(err) - } + CheckErr(err) id, err := res.LastInsertId() - if err != nil { - panic(err) - } + CheckErr(err) fmt.Println("LastInsertId:", id) } } diff --git a/tools/gen.go b/tools/gen.go index f9d0803..08c3b1d 100644 --- a/tools/gen.go +++ b/tools/gen.go @@ -155,9 +155,7 @@ var ( func GenApi1Data() { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() // global @@ -168,14 +166,10 @@ func GenApi1Data() { normalLives := []model.NormalLiveStatusList{} sql := `SELECT live_difficulty_id FROM normal_live_m ORDER BY live_difficulty_id ASC` rows, err := db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { err = rows.Scan(&liveDifficultyId) - if err != nil { - panic(err) - } + CheckErr(err) normalLive := model.NormalLiveStatusList{ LiveDifficultyID: liveDifficultyId, @@ -192,14 +186,10 @@ func GenApi1Data() { specialLives := []model.SpecialLiveStatusList{} sql = `SELECT live_difficulty_id FROM special_live_m ORDER BY live_difficulty_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { err = rows.Scan(&liveDifficultyId) - if err != nil { - panic(err) - } + CheckErr(err) specialLive := model.SpecialLiveStatusList{ LiveDifficultyID: liveDifficultyId, @@ -259,9 +249,7 @@ func GenApi1Data() { // unit_list_result sql = `SELECT unit_id,album_series_id,rarity,rank_min,rank_max,hp_max,default_removable_skill_capacity FROM unit_m WHERE unit_id NOT IN (SELECT unit_id FROM unit_m WHERE unit_type_id IN (SELECT unit_type_id FROM unit_type_m WHERE image_button_asset IS NULL AND (background_color = 'dcdbe3' AND original_attribute_id IS NULL) OR (unit_type_id IN (10,110,127,128,129) OR unit_type_id BETWEEN 131 AND 140))) ORDER BY unit_number ASC;` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) unitsData := []model.Active{} oId := 3000000000 // 起始卡片 ID，随意设置 @@ -353,16 +341,12 @@ func GenApi1Data() { unitData.UnitRemovableSkillCapacity = 8 rs, err := db.Query("SELECT COUNT(*) AS ct FROM unit_sign_asset_m WHERE unit_id = ?", uid) - if err != nil { - panic(err) - } + CheckErr(err) ct := 0 for rs.Next() { err = rs.Scan(&ct) - if err != nil { - panic(err) - } + CheckErr(err) } if ct > 0 { unitData.IsSigned = true @@ -441,13 +425,9 @@ func GenApi1Data() { // HACK 保存卡组信息 deck, err := json.Marshal(defaultDeckUnitList) - if err != nil { - panic(err) - } + CheckErr(err) err = database.LevelDb.Put([]byte("deck_info"), deck) - if err != nil { - panic(err) - } + CheckErr(err) // unit_support_result unitSupportResp := model.UnitSupportResp{ @@ -490,9 +470,7 @@ func GenApi1Data() { albumLists := []model.AlbumResult{} sql = `SELECT unit_id,rarity FROM unit_m ORDER BY unit_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { albumList := model.AlbumResult{ RankMaxFlag: true, @@ -503,9 +481,7 @@ func GenApi1Data() { } var uid, rit int err = rows.Scan(&uid, &rit) - if err != nil { - panic(err) - } + CheckErr(err) albumList.UnitID = uid if rit != 4 { albumList.SignFlag = false @@ -527,16 +503,12 @@ func GenApi1Data() { albumList.TotalLove = 1000 rs, err := db.Query("SELECT COUNT(*) AS ct FROM unit_sign_asset_m WHERE unit_id = ?", uid) - if err != nil { - panic(err) - } + CheckErr(err) ct := 0 for rs.Next() { err = rs.Scan(&ct) - if err != nil { - panic(err) - } + CheckErr(err) } if ct > 0 { albumList.SignFlag = true @@ -560,16 +532,12 @@ func GenApi1Data() { // scenario_status_result sql = `SELECT scenario_id FROM scenario_m ORDER BY scenario_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) scenarioLists := []model.ScenarioStatusList{} for rows.Next() { var sid int err = rows.Scan(&sid) - if err != nil { - panic(err) - } + CheckErr(err) scenarioLists = append(scenarioLists, model.ScenarioStatusList{ ScenarioID: sid, Status: 2, @@ -589,16 +557,12 @@ func GenApi1Data() { // subscenario_status_result sql = `SELECT subscenario_id FROM subscenario_m ORDER BY subscenario_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) subScenarioLists := []model.SubscenarioStatusList{} for rows.Next() { var sid int err = rows.Scan(&sid) - if err != nil { - panic(err) - } + CheckErr(err) subScenarioLists = append(subScenarioLists, model.SubscenarioStatusList{ SubscenarioID: sid, Status: 2, @@ -620,30 +584,22 @@ func GenApi1Data() { eventsList := []model.EventScenarioList{} sql = `SELECT event_id FROM event_scenario_m GROUP BY event_id ORDER BY event_id DESC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) for rows.Next() { var eventId int err = rows.Scan(&eventId) - if err != nil { - panic(err) - } + CheckErr(err) sql = `SELECT event_scenario_id,chapter,chapter_asset,open_date FROM event_scenario_m WHERE event_id = ? ORDER BY chapter DESC` chaps, err := db.Query(sql, eventId) - if err != nil { - panic(err) - } + CheckErr(err) chapsList := []model.EventScenarioChapterList{} var open_date string for chaps.Next() { var event_scenario_id, chapter int var chapter_asset interface{} err = chaps.Scan(&event_scenario_id, &chapter, &chapter_asset, &open_date) - if err != nil { - panic(err) - } + CheckErr(err) chapList := model.EventScenarioChapterList{ EventScenarioID: event_scenario_id, Chapter: chapter, @@ -692,29 +648,21 @@ func GenApi1Data() { // multi_unit_scenario_result sql = `SELECT multi_unit_id FROM multi_unit_scenario_m GROUP BY multi_unit_id ORDER BY multi_unit_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) var mId int multiUnitsList := []model.MultiUnitScenarioStatusList{} for rows.Next() { err = rows.Scan(&mId) - if err != nil { - panic(err) - } + CheckErr(err) sql = `SELECT multi_unit_scenario_btn_asset,open_date,multi_unit_scenario_id,chapter FROM multi_unit_scenario_m a LEFT JOIN multi_unit_scenario_open_m b ON a.multi_unit_id = b.multi_unit_id WHERE a.multi_unit_id = ?` units, err := db.Query(sql, mId) - if err != nil { - panic(err) - } + CheckErr(err) var multi_unit_scenario_id, chapter int var multi_unit_scenario_btn_asset, open_date string for units.Next() { err = units.Scan(&multi_unit_scenario_btn_asset, &open_date, &multi_unit_scenario_id, &chapter) - if err != nil { - panic(err) - } + CheckErr(err) } multiUnitsList = append(multiUnitsList, model.MultiUnitScenarioStatusList{ @@ -846,16 +794,12 @@ func GenApi1Data() { // award_result sql = `SELECT award_id FROM award_m ORDER BY award_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) awardsList := []model.AwardInfo{} for rows.Next() { var aId int err = rows.Scan(&aId) - if err != nil { - panic(err) - } + CheckErr(err) isSet := false if aId == 113 { // 极推穗乃果 isSet = true @@ -880,16 +824,12 @@ func GenApi1Data() { // background_result sql = `SELECT background_id FROM background_m ORDER BY background_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) backgroundsList := []model.BackgroundInfo{} for rows.Next() { var bId int err = rows.Scan(&bId) - if err != nil { - panic(err) - } + CheckErr(err) isSet := false if bId == 143 { // 穗乃果的房间[情人节] isSet = true @@ -915,25 +855,19 @@ func GenApi1Data() { stampResp := utils.ReadAllText("assets/stamp.json") var mStampResp interface{} err = json.Unmarshal([]byte(stampResp), &mStampResp) - if err != nil { - panic(err) - } + CheckErr(err) // _ = utils.ReadAllText("assets/stamp.json") respAll = append(respAll, mStampResp) // exchange_point_result sql = `SELECT exchange_point_id FROM exchange_point_m ORDER BY exchange_point_id ASC` rows, err = db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) exPointsList := []model.ExchangePointList{} for rows.Next() { var eId int err = rows.Scan(&eId) - if err != nil { - panic(err) - } + CheckErr(err) exPointsList = append(exPointsList, model.ExchangePointList{ Rarity: eId, ExchangePoint: 9999, @@ -979,9 +913,7 @@ func GenApi1Data() { // _ = utils.ReadAllText("assets/item.json") var mItemResp interface{} err = json.Unmarshal([]byte(itemResp), &mItemResp) - if err != nil { - panic(err) - } + CheckErr(err) respAll = append(respAll, mItemResp) // marathon_result @@ -1006,18 +938,14 @@ func GenApi1Data() { // Final k, err := json.Marshal(respAll) - if err != nil { - panic(err) - } + CheckErr(err) resp := model.Response{ ResponseData: k, ReleaseInfo: []interface{}{}, StatusCode: 200, } k, err = json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(k)) utils.WriteAllText("assets/api1.json", string(k)) @@ -1025,9 +953,7 @@ func GenApi1Data() { func GenApi2Data() { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() // global @@ -1119,17 +1045,13 @@ func GenApi2Data() { // museum_result sql := `SELECT museum_contents_id,smile_buff,pure_buff,cool_buff FROM museum_contents_m ORDER BY museum_contents_id ASC` rows, err := db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) var smileBuf, pureBuf, coolBuf int var mIds []int for rows.Next() { var museum_contents_id, smile_buff, pure_buff, cool_buff int err = rows.Scan(&museum_contents_id, &smile_buff, &pure_buff, &cool_buff) - if err != nil { - panic(err) - } + CheckErr(err) smileBuf += smile_buff pureBuf += pure_buff coolBuf += cool_buff @@ -1155,18 +1077,14 @@ func GenApi2Data() { // Final k, err := json.Marshal(respAll) - if err != nil { - panic(err) - } + CheckErr(err) resp := model.Response{ ResponseData: k, ReleaseInfo: []interface{}{}, StatusCode: 200, } k, err = json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(k)) utils.WriteAllText("assets/api2.json", string(k)) @@ -1211,9 +1129,7 @@ func GenApi3Data() { var result []interface{} love := utils.ReadAllText("assets/love.json") err := json.Unmarshal([]byte(love), &result) - if err != nil { - panic(err) - } + CheckErr(err) loveResp := LoveResp{ // _ = LoveResp{ Result: result, @@ -1319,18 +1235,14 @@ func GenApi3Data() { // Final k, err := json.Marshal(respAll) - if err != nil { - panic(err) - } + CheckErr(err) resp := model.Response{ ResponseData: k, ReleaseInfo: []interface{}{}, StatusCode: 200, } k, err = json.Marshal(resp) - if err != nil { - panic(err) - } + CheckErr(err) // fmt.Println(string(k)) utils.WriteAllText("assets/api3.json", string(k)) diff --git a/tools/notes.go b/tools/notes.go index 64bb49c..9ba4645 100644 --- a/tools/notes.go +++ b/tools/notes.go @@ -15,9 +15,7 @@ import ( func SyncNotesList() { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer func() { db.Close() fmt.Println("Sync notes list done!") @@ -26,9 +24,7 @@ func SyncNotesList() { sql := `SELECT live_setting_id,notes_setting_asset,notes_list FROM live_setting_m ORDER BY live_setting_id ASC` rows, err := db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) liveList := make(map[int]string) diff --git a/tools/tools.go b/tools/tools.go index 9050da8..50aee22 100644 --- a/tools/tools.go +++ b/tools/tools.go @@ -11,3 +11,9 @@ func init() { // go SyncNotesList() // GenDownloadDb() } + +func CheckErr(err error) { + if err != nil { + panic(err) + } +} diff --git a/tools/unit.go b/tools/unit.go index a8dd6a0..f611d08 100644 --- a/tools/unit.go +++ b/tools/unit.go @@ -12,16 +12,12 @@ import ( func ListUnitData() { db, err := sql.Open("sqlite3", "assets/main.db") - if err != nil { - panic(err) - } + CheckErr(err) defer db.Close() sql := `SELECT unit_id,rarity,rank_min,rank_max,hp_max,default_removable_skill_capacity FROM unit_m WHERE unit_id NOT IN (SELECT unit_id FROM unit_m WHERE unit_type_id IN (SELECT unit_type_id FROM unit_type_m WHERE image_button_asset IS NULL AND (background_color = 'dcdbe3' AND original_attribute_id IS NULL) OR (unit_type_id IN (10,110,127,128,129) OR unit_type_id BETWEEN 131 AND 140))) ORDER BY unit_number ASC;` rows, err := db.Query(sql) - if err != nil { - panic(err) - } + CheckErr(err) unitsData := []model.Active{} oId := 3071290948 @@ -110,16 +106,12 @@ func ListUnitData() { unitData.UnitRemovableSkillCapacity = 8 rs, err := db.Query("SELECT COUNT(*) AS ct FROM unit_sign_asset_m WHERE unit_id = ?", uid) - if err != nil { - panic(err) - } + CheckErr(err) ct := 0 for rs.Next() { err = rs.Scan(&ct) - if err != nil { - panic(err) - } + CheckErr(err) } if ct > 0 { unitData.IsSigned = true @@ -132,8 +124,6 @@ func ListUnitData() { } data, err := json.Marshal(unitsData) - if err != nil { - panic(err) - } + CheckErr(err) fmt.Println(string(data)) }