ergo/irc/database.go

package irc

import (
	"database/sql"
	"fmt"
	_ "github.com/mattn/go-sqlite3"
	"log"
	"os"
)

func InitDB(path string) {
	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
}