mirror of
https://github.com/d0zingcat/gocryptotrader.git
synced 2026-05-13 23:16:45 +00:00
d64d56f77c
* build/ci: Update Go to v1.24, golangci-lint to v1.64.5 and fix issues * Address shazbert's nitters * linter/config: Fix new linter issue and use versionSize const * Address gk's nitters and fix additional linter issue after rebase * Address glorious nits * staticcheck: Fix additional linter issues after upgrading to Go 1.24.1 and golangci-lint v1.64.6 Also addresses nits * Improve testing, assertify usage and use common.ErrParsingWSField * TestCreateNewStrategy: Replace must > should wording
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/thrasher-corp/gocryptotrader/common/file"
|
|
"github.com/thrasher-corp/gocryptotrader/config"
|
|
"github.com/thrasher-corp/gocryptotrader/core"
|
|
"github.com/thrasher-corp/gocryptotrader/database"
|
|
"github.com/thrasher-corp/gocryptotrader/database/repository"
|
|
"github.com/thrasher-corp/gocryptotrader/encoding/json"
|
|
)
|
|
|
|
var (
|
|
configFile string
|
|
outputFolder string
|
|
)
|
|
|
|
var sqlboilerConfig map[string]driverConfig
|
|
|
|
type driverConfig struct {
|
|
DBName string `json:"dbname,omitempty"`
|
|
Host string `json:"host,omitempty"`
|
|
Port uint32 `json:"port,omitempty"`
|
|
User string `json:"user,omitempty"`
|
|
Pass string `json:"pass,omitempty"`
|
|
Schema string `json:"schema,omitempty"`
|
|
SSLMode string `json:"sslmode,omitempty"`
|
|
Blacklist []string `json:"blacklist,omitempty"`
|
|
}
|
|
|
|
func main() {
|
|
fmt.Println("GoCryptoTrader SQLBoiler config generation tool")
|
|
fmt.Println(core.Copyright)
|
|
fmt.Println()
|
|
|
|
flag.StringVar(&configFile, "config", config.DefaultFilePath(), "config file to load")
|
|
flag.StringVar(&outputFolder, "outdir", "", "overwrite default output folder")
|
|
flag.Parse()
|
|
|
|
var cfg config.Config
|
|
err := cfg.LoadConfig(configFile, true)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
convertGCTtoSQLBoilerConfig(&cfg.Database)
|
|
|
|
jsonOutput, err := json.MarshalIndent(sqlboilerConfig, "", " ")
|
|
if err != nil {
|
|
fmt.Printf("Marshal failed: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
path := filepath.Join(outputFolder, "sqlboiler.json")
|
|
err = os.WriteFile(path, jsonOutput, file.DefaultPermissionOctal)
|
|
if err != nil {
|
|
fmt.Printf("Write failed: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
fmt.Println("sqlboiler.json file created")
|
|
}
|
|
|
|
func convertGCTtoSQLBoilerConfig(c *database.Config) {
|
|
tempConfig := driverConfig{
|
|
Blacklist: []string{"goose_db_version"},
|
|
}
|
|
|
|
sqlboilerConfig = make(map[string]driverConfig)
|
|
|
|
dbType := repository.GetSQLDialect()
|
|
|
|
if dbType == database.DBPostgreSQL {
|
|
dbType = "psql"
|
|
}
|
|
if dbType == database.DBSQLite || dbType == database.DBSQLite3 {
|
|
tempConfig.DBName = getLoadedDBPath()
|
|
} else {
|
|
tempConfig.User = c.Username
|
|
tempConfig.Pass = c.Password
|
|
tempConfig.Port = c.Port
|
|
tempConfig.Host = c.Host
|
|
tempConfig.DBName = c.Database
|
|
tempConfig.SSLMode = c.SSLMode
|
|
}
|
|
|
|
sqlboilerConfig[dbType] = tempConfig
|
|
}
|
|
|
|
// getLoadedDBPath gets the path loaded by 'database/drivers/sqlite3'
|
|
func getLoadedDBPath() string {
|
|
cfg := database.DB.GetConfig()
|
|
return filepath.Join(database.DB.DataPath, cfg.Database)
|
|
}
|