mirror of https://github.com/ergochat/ergo.git
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
// Copyright (c) 2012-2014 Jeremy Latt
|
|
// released under the MIT license
|
|
|
|
package irc
|
|
|
|
import (
|
|
"database/sql"
|
|
"encoding/base64"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/tidwall/buntdb"
|
|
)
|
|
|
|
const (
|
|
// key for the primary salt used by the ircd
|
|
keySalt = "crypto.salt"
|
|
)
|
|
|
|
func InitDB(buntpath string, path string) {
|
|
// prepare kvstore db
|
|
os.Remove(buntpath)
|
|
store, err := buntdb.Open(buntpath)
|
|
if err != nil {
|
|
log.Fatal(fmt.Sprintf("Failed to open datastore: %s", err.Error()))
|
|
}
|
|
defer store.Close()
|
|
|
|
err = store.Update(func(tx *buntdb.Tx) error {
|
|
salt, err := NewSalt()
|
|
encodedSalt := base64.StdEncoding.EncodeToString(salt)
|
|
if err != nil {
|
|
log.Fatal("Could not generate cryptographically-secure salt for the user:", err.Error())
|
|
}
|
|
tx.Set(keySalt, encodedSalt, nil)
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
log.Fatal("Could not save bunt store:", err.Error())
|
|
}
|
|
|
|
// prepare SQLite db
|
|
os.Remove(path)
|
|
db := OpenDB(path)
|
|
defer db.Close()
|
|
_, err = db.Exec(`
|
|
CREATE TABLE channel (
|
|
name TEXT NOT NULL UNIQUE,
|
|
flags TEXT DEFAULT '',
|
|
key TEXT DEFAULT '',
|
|
topic TEXT DEFAULT '',
|
|
user_limit INTEGER DEFAULT 0,
|
|
ban_list TEXT DEFAULT '',
|
|
except_list TEXT DEFAULT '',
|
|
invite_list TEXT DEFAULT '')`)
|
|
if err != nil {
|
|
log.Fatal("initdb error: ", err)
|
|
}
|
|
}
|
|
|
|
func UpgradeDB(path string) {
|
|
db := OpenDB(path)
|
|
alter := `ALTER TABLE channel ADD COLUMN %s TEXT DEFAULT ''`
|
|
cols := []string{"ban_list", "except_list", "invite_list"}
|
|
for _, col := range cols {
|
|
_, err := db.Exec(fmt.Sprintf(alter, col))
|
|
if err != nil {
|
|
log.Fatal("updatedb error: ", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func OpenDB(path string) *sql.DB {
|
|
db, err := sql.Open("sqlite3", path)
|
|
if err != nil {
|
|
log.Fatal("open db error: ", err)
|
|
}
|
|
return db
|
|
}
|