feat(table): add load-predefined tables function

internal/entity/diffTableHeader.go

@@ -1,10 +1,12 @@
 package entity
 
 import (
-	"strings"
+    "encoding/json"
+    "strings"
+    "embed"
 
-	"github.com/Catizard/bmstable"
-	"gorm.io/gorm"
+    "github.com/Catizard/bmstable"
+    "gorm.io/gorm"
 )
 
 type DiffTableHeader struct {
@@ -45,259 +47,68 @@ func NewDiffTableHeaderFromImport(importHeader *bmstable.DifficultTable, inherit
 
 // Cropped struct definition of DiffTableHeader
 type PredefineTableHeader struct {
-	HeaderUrl    string
-	Name         string
-	Symbol       string
-	TagColor     string
-	TagTextColor string
-	Category     string
-
-	headerUrls map[string]string // Internal use, to simplify the init step
+    HeaderUrl    string
+    Name         string
+    Symbol       string
+    TagColor     string
+    TagTextColor string
+    Category     string
 }
 
 // One predefined header scheme is basically an array of headers
 // and some additional meta info
 type PredefineTableScheme struct {
-	Headers []PredefineTableHeader
-	Name    string
-	//SelectedCategory []string
-}
-
-var PredefineTableSchemeNames []string = []string{
-	"Raw",
-	"Zris",
-	"616sb",
+    Headers []PredefineTableHeader
+    Name    string
+    //SelectedCategory []string
 }
 
 var PredefineTableSchemes map[string]PredefineTableScheme = make(map[string]PredefineTableScheme)
 
-func init() {
-	urlJoinedHeaders := make([]PredefineTableHeader, 0)
-	// NOTE: The element in headerUrls' order must follow the order of PredefineTableSchemeNames
-	// Starter category
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "Stardust",
-		Symbol:       "ξ",
-		Category:     "Starter",
-		TagColor:     "#E5F4D3",
-		TagTextColor: "#2B410D",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://mqppppp.neocities.org/StardustTable.html",
-			/* No zris mirror */
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "Starlight",
-		Symbol:       "sr",
-		Category:     "Starter",
-		TagColor:     "#E8EDFF",
-		TagTextColor: "#121D52",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://djkuroakari.github.io/starlighttable.html",
-			/* No zris mirror */
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "通常難易度表",
-		Symbol:       "☆",
-		Category:     "Starter",
-		TagColor:     "#EDF7FD",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://www.ribbit.xyz/bms/tables/normal.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/normal/normal_header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/ff4cb9d4c6f14f68a895798143a2a7b9/header.json",
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "NEW GENERATION 通常難易度表",
-		Symbol:       "▽",
-		Category:     "Starter",
-		TagColor:     "#FFFAEB",
-		TagTextColor: "#946D18",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://rattoto10.jounin.jp/table.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/normal2/header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/1c7c8c5d80338a5835f8221ca8d1bc47/header.json",
-		},
-	})
-	// Insane category
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "Satellite",
-		Symbol:       "sl",
-		Category:     "Insane",
-		TagColor:     "#B6EAD2",
-		TagTextColor: "#0A4D2F",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://stellabms.xyz/sl/table.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/satellite/header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/adceda62af0cac6fdba049c0ced12479/header.json",
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "発狂BMS難易度表",
-		Symbol:       "★",
-		Category:     "Insane",
-		TagColor:     "#A3D8F5",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://www.ribbit.xyz/bms/tables/insane.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/insane/insane_header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/ea159b8f57a768682957735baf332e04/header.json",
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "NEW GENERATION 発狂難易度表",
-		Symbol:       "▼",
-		Category:     "Insane",
-		TagColor:     "#FFECB9",
-		TagTextColor: "#946D18",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://rattoto10.jounin.jp/table_insane.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/insane2/insane_header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/5f5d17d84528e331316304128063a515/header.json",
-		},
-	})
-	// Overjoy category
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "第三期Overjoy",
-		Symbol:       "★★",
-		Category:     "Overjoy",
-		TagColor:     "#DDBDF2",
-		TagTextColor: "#5C2989",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://rattoto10.jounin.jp/table_overjoy.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/overjoy/header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/6096431ac74339abf61389cd60a5586e/header.json",
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "Stella",
-		Symbol:       "st",
-		Category:     "Overjoy",
-		TagColor:     "#FFB5A8",
-		TagTextColor: "#331510",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://stellabms.xyz/st/table.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/stella/header.json",
-			PredefineTableSchemeNames[2]: "https://bms.alvorna.com/tables/be86d1e944f4003491b970f4797d6c9a/header.json",
-		},
-	})
-	// DP category
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "δ難易度表",
-		Symbol:       "δ",
-		Category:     "DP",
-		TagColor:     "#EDF7FD",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://deltabms.yaruki0.net/table/dpdelta",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/dp_normal/dpn_header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "発狂DP難易度表",
-		Symbol:       "★",
-		Category:     "DP",
-		TagColor:     "#A3D8F5",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://deltabms.yaruki0.net/table/dpinsane",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/dp_insane/dpi_header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "DP Overjoy",
-		Symbol:       "★★",
-		Category:     "DP",
-		TagColor:     "#DDBDF2",
-		TagTextColor: "#5C2989",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://ereter.net/dpoverjoy",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/dp_overjoy/header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "DP Satellite",
-		Symbol:       "DPsl",
-		Category:     "DP",
-		TagColor:     "#B6EAD2",
-		TagTextColor: "#0A4D2F",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://stellabms.xyz/dp/table.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/dp_satellite/header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "DP Stella",
-		Symbol:       "DPst",
-		Category:     "DP",
-		TagColor:     "#FFB5A8",
-		TagTextColor: "#331510",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://stellabms.xyz/dpst/table.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/dp_stella/header.json",
-			/* No 616sb mirror */
-		},
-	})
-	// PMS category
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "PMSデータベース(Lv1~45)",
-		Symbol:       "PLv",
-		Category:     "PMS",
-		TagColor:     "#EDF7FD",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "http://pmsdifficulty.xxxxxxxx.jp/PMSdifficulty.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/pms_normal/pmsdatabase_header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "発狂PMSデータベース(lv46～)",
-		Symbol:       "P●",
-		Category:     "PMS",
-		TagColor:     "#A3D8F5",
-		TagTextColor: "#0F5F8A",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://pmsdifficulty.xxxxxxxx.jp/insane_PMSdifficulty.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/pms_insane/insane_pmsdatabase_header.json",
-			/* No 616sb mirror */
-		},
-	})
-	urlJoinedHeaders = append(urlJoinedHeaders, PredefineTableHeader{
-		Name:         "発狂PMS難易度表",
-		Symbol:       "●",
-		Category:     "PMS",
-		TagColor:     "#FFECB9",
-		TagTextColor: "#946D18",
-		headerUrls: map[string]string{
-			PredefineTableSchemeNames[0]: "https://pmsdifficulty.xxxxxxxx.jp/_pastoral_home.html",
-			PredefineTableSchemeNames[1]: "http://zris.work/bmstable/pms_upper/header.json",
-			/* No 616sb mirror */
-		},
-	})
+//go:embed predefine_table_schemes.json
+var predefineTableSchemesFS embed.FS
 
-	for _, schemeName := range PredefineTableSchemeNames {
-		headers := make([]PredefineTableHeader, 0)
-		for _, header := range urlJoinedHeaders {
-			shallow := header
-			// push down
-			shallow.HeaderUrl = header.headerUrls["Raw"]
-			if url, ok := header.headerUrls[schemeName]; ok {
-				shallow.HeaderUrl = url
-			}
-			headers = append(headers, shallow)
-		}
-		PredefineTableSchemes[schemeName] = PredefineTableScheme{
-			Name:    schemeName,
-			Headers: headers,
-		}
-	}
+type predefineTableSchemesFile struct {
+    SchemeNames []string `json:"schemeNames"`
+    Headers     []struct {
+        Name         string            `json:"name"`
+        Symbol       string            `json:"symbol"`
+        Category     string            `json:"category"`
+        TagColor     string            `json:"tagColor"`
+        TagTextColor string            `json:"tagTextColor"`
+        HeaderUrls   map[string]string `json:"headerUrls"`
+    } `json:"headers"`
+}
+
+func init() {
+    var file predefineTableSchemesFile
+    data, err := predefineTableSchemesFS.ReadFile("predefine_table_schemes.json")
+    if err != nil {
+        panic(err)
+    }
+    if err := json.Unmarshal(data, &file); err != nil {
+        panic(err)
+    }
+    for _, schemeName := range file.SchemeNames {
+        headers := make([]PredefineTableHeader, 0, len(file.Headers))
+        for _, h := range file.Headers {
+            shallow := PredefineTableHeader{
+                Name:         h.Name,
+                Symbol:       h.Symbol,
+                Category:     h.Category,
+                TagColor:     h.TagColor,
+                TagTextColor: h.TagTextColor,
+            }
+            url := h.HeaderUrls["Raw"]
+            if u, ok := h.HeaderUrls[schemeName]; ok && u != "" {
+                url = u
+            }
+            shallow.HeaderUrl = url
+            headers = append(headers, shallow)
+        }
+        PredefineTableSchemes[schemeName] = PredefineTableScheme{
+            Name:    schemeName,
+            Headers: headers,
+        }
+    }
 }