persist and load channel mask lists

irc/database.go

@@ -2,6 +2,7 @@ package irc
 
 import (
 	"database/sql"
+	"fmt"
 	_ "github.com/mattn/go-sqlite3"
 	"log"
 	"os"
@@ -14,15 +15,30 @@ func InitDB(path string) {
 	_, err := db.Exec(`
         CREATE TABLE channel (
           name TEXT NOT NULL UNIQUE,
-          flags TEXT NOT NULL,
-          key TEXT NOT NULL,
-          topic TEXT NOT NULL,
-          user_limit INTEGER DEFAULT 0)`)
+          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 {