64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
// Copyright (C) 2023 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
files := []string{}
|
|
globPaths, err := filepath.Glob("*.dbx")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
sort.Strings(globPaths) // sort so files are always in a specific order regardless of os
|
|
for _, p := range globPaths {
|
|
if p != "satellitedb.dbx" {
|
|
files = append(files, fmt.Sprintf("-i=%s", p))
|
|
}
|
|
}
|
|
files = append(files, "satellitedb.dbx", ".")
|
|
|
|
// final commands look like `dbx schema -d pgx -d pgxcockroach -i accounting.dbx ... -i user.dbx satellitedb.dbx .`
|
|
// and `dbx golang -d pgx -d pgxcockroach -t templates -i accounting.dbx ... -i user.dbx satellitedb.dbx .`
|
|
schemaArgs := append([]string{"schema", "-d=pgx", "-d=pgxcockroach"}, files...)
|
|
schemaOut, err := exec.Command("dbx", schemaArgs...).CombinedOutput()
|
|
if err != nil {
|
|
fmt.Println("schema out", string(schemaOut))
|
|
log.Fatal(err)
|
|
}
|
|
gogenArgs := append([]string{"golang", "-d=pgx", "-d=pgxcockroach", "-p=dbx", "-t=templates"}, files...)
|
|
gogenOut, err := exec.Command("dbx", gogenArgs...).CombinedOutput()
|
|
if err != nil {
|
|
fmt.Println("gogen out", string(gogenOut))
|
|
log.Fatal(err)
|
|
}
|
|
|
|
originalDBXBytes, err := os.ReadFile("satellitedb.dbx.go")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
replacer := strings.NewReplacer(
|
|
"*sql.Tx", "tagsql.Tx",
|
|
"*sql.Rows", "tagsql.Rows",
|
|
`_ "github.com/jackc/pgx/v5/stdlib"`, `"storj.io/private/tagsql"`,
|
|
"type DB struct {\n\t*sql.DB", "type DB struct {\n\ttagsql.DB",
|
|
"db = &DB{\n\t\tDB: sql_db", "db = &DB{\n\t\tDB: tagsql.Wrap(sql_db)",
|
|
)
|
|
newDBX := replacer.Replace(string(originalDBXBytes))
|
|
fileString := "//lint:file-ignore U1000,ST1012 generated file\n" + newDBX
|
|
|
|
err = os.WriteFile("satellitedb.dbx.go", []byte(fileString), 0755)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|