HIMBOT RETURNS

This commit is contained in:
2024-11-22 17:04:33 -06:00
parent ba9bf88a1c
commit 4cf9c3295f
5 changed files with 132 additions and 174 deletions

View File

@ -17,35 +17,16 @@ var DBClient *sql.DB
var DBConnector *libsql.Connector
func InitDB() error {
dbUrl := os.Getenv("DATABASE_URL")
dbToken := os.Getenv("DATABASE_AUTH_TOKEN")
if dbUrl == "" || dbToken == "" {
return fmt.Errorf("database configuration missing")
}
// Determine DB path based on /data directory existence
dbPath := "himbot.db" // default to local
if _, err := os.Stat("/data"); !os.IsNotExist(err) {
dbPath = "/data/himbot.db"
}
dbName := "file:./himbot.db"
connector, connectorError := libsql.NewEmbeddedReplicaConnector(
dbPath,
dbUrl,
libsql.WithAuthToken(dbToken),
)
if connectorError != nil {
fmt.Fprintf(os.Stderr, "failed to open db %s: %s", dbUrl, connectorError)
db, err := sql.Open("libsql", dbName)
if err != nil {
fmt.Fprintf(os.Stderr, "failed to open db %s", err)
os.Exit(1)
}
// finalDBUrl := fmt.Sprintf("%s?authToken=%s", dbUrl, dbToken)
client := sql.OpenDB(connector)
DBClient = client
DBConnector = connector
DBClient = db
return runMigrations()
}