Add `users` table
This commit is contained in:
parent
83a6194b7c
commit
bf6ef1d07c
|
@ -8,23 +8,6 @@ import (
|
||||||
"modernc.org/ql"
|
"modernc.org/ql"
|
||||||
)
|
)
|
||||||
|
|
||||||
type migration struct {
|
|
||||||
up string
|
|
||||||
down string
|
|
||||||
}
|
|
||||||
|
|
||||||
var migrations = []migration{
|
|
||||||
migration{
|
|
||||||
up: `CREATE TABLE IF NOT EXISTS configs (
|
|
||||||
key string NOT NULL,
|
|
||||||
value string NOT NULL
|
|
||||||
);
|
|
||||||
CREATE UNIQUE INDEX IF NOT EXISTS unique_configs_key ON configs (key);
|
|
||||||
`,
|
|
||||||
down: `DROP TABLE IF EXISTS configs`,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
|
|
||||||
func openDB() *sql.DB {
|
func openDB() *sql.DB {
|
||||||
ql.RegisterDriver()
|
ql.RegisterDriver()
|
||||||
pigeonPath := maybeSetupPigeonDir()
|
pigeonPath := maybeSetupPigeonDir()
|
||||||
|
@ -41,22 +24,7 @@ func openDB() *sql.DB {
|
||||||
log.Fatalf("failed to ping db: %s", err1)
|
log.Fatalf("failed to ping db: %s", err1)
|
||||||
}
|
}
|
||||||
|
|
||||||
tx, err := db.Begin()
|
migrateUp(db)
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Failed to start transaction: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, migration := range migrations {
|
|
||||||
_, err := tx.Exec(migration.up)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Migration failure: %s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if tx.Commit() != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return db
|
return db
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,51 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type migration struct {
|
||||||
|
up string
|
||||||
|
down string
|
||||||
|
}
|
||||||
|
|
||||||
|
var migrations = []migration{
|
||||||
|
migration{
|
||||||
|
up: `CREATE TABLE IF NOT EXISTS configs (
|
||||||
|
key string NOT NULL,
|
||||||
|
value string NOT NULL
|
||||||
|
);
|
||||||
|
CREATE UNIQUE INDEX IF NOT EXISTS unique_configs_key ON configs (key);
|
||||||
|
`,
|
||||||
|
down: `DROP TABLE IF EXISTS configs`,
|
||||||
|
},
|
||||||
|
migration{
|
||||||
|
up: `CREATE TABLE IF NOT EXISTS users (
|
||||||
|
mhash string NOT NULL,
|
||||||
|
status string NOT NULL
|
||||||
|
);
|
||||||
|
CREATE UNIQUE INDEX IF NOT EXISTS unique_users_mhash ON users (mhash);
|
||||||
|
`,
|
||||||
|
down: `DROP TABLE IF EXISTS users`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func migrateUp(db *sql.DB) {
|
||||||
|
tx, err := db.Begin()
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Failed to start transaction: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, migration := range migrations {
|
||||||
|
_, err := tx.Exec(migration.up)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Migration failure(%d): %s", i, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if tx.Commit() != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
|
@ -6,7 +6,7 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
func showIdentity() string {
|
func showIdentity() string {
|
||||||
existingKey := GetConfig("private_key")
|
existingKey := GetConfig("public_key")
|
||||||
if len(existingKey) == 0 {
|
if len(existingKey) == 0 {
|
||||||
return "NONE"
|
return "NONE"
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue