User persistence to sqlite.

.gitignore

@@ -1,4 +1,5 @@
 pkg
 bin
 src/code.google.com/
+src/github.com/
 ergonomadic.db

sql/drop.sql

@@ -1,10 +1,10 @@
-DROP INDEX user_id_channel_id
-DROP TABLE user_channel
+DROP INDEX IF EXISTS index_user_id_channel_id;
+DROP TABLE IF EXISTS user_channel;
 
-DROP INDEX channel_name
-DROP INDEX channel_id
-DROP TABLE channel
+DROP INDEX IF EXISTS index_channel_name;
+DROP INDEX IF EXISTS index_channel_id;
+DROP TABLE IF EXISTS channel;
 
-DROP INDEX user_nick
-DROP INDEX user_id
-DROP TABLE user
+DROP INDEX IF EXISTS index_user_nick;
+DROP INDEX IF EXISTS index_user_id;
+DROP TABLE IF EXISTS user;

sql/init.sql

@@ -1,10 +1,20 @@
-CREATE TABLE user (id integer not null primary key autoincrement, nick text not null, hash blob not null)
-CREATE UNIQUE INDEX user_id ON user (id)
-CREATE UNIQUE INDEX user_nick ON user (nick)
+CREATE TABLE user (
+  id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
+  nick TEXT NOT NULL UNIQUE,
+  hash BLOB NOT NULL
+);
+CREATE INDEX index_user_id ON user(id);
+CREATE INDEX index_user_nick ON user(nick);
 
-CREATE TABLE channel (id integer not null primary key autoincrement, name text not  null)
-CREATE UNIQUE INDEX channel_id ON channel (id)
-CREATE UNIQUE INDEX channel_name ON channel (name)
+CREATE TABLE channel (
+  id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
+  name TEXT NOT NULL UNIQUE
+);
+CREATE INDEX index_channel_id ON channel(id);
 
-CREATE_TABLE user_channel (id integer not null primary key autoincrement, user_id integer not null, channel_id integer not null)
-CREATE UNIQUE INDEX user_id_channel_id ON user_channel (user_id, channel_id)
+CREATE TABLE user_channel (
+  id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
+  user_id INTEGER NOT NULL,
+  channel_id INTEGER NOT NULL
+);
+CREATE UNIQUE INDEX index_user_id_channel_id ON user_channel (user_id, channel_id);

src/ergonomadicdb/ergonomadicdb.go

@@ -5,18 +5,7 @@ import (
 	"irc"
 )
 
-var (
-	actions = map[string]func(*irc.Database){
-		"init": func(db *irc.Database) {
-			db.InitTables()
-		},
-		"drop": func(db *irc.Database) {
-			db.DropTables()
-		},
-	}
-)
-
 func main() {
 	flag.Parse()
-	actions[flag.Arg(0)](irc.NewDatabase())
+	irc.NewDatabase().ExecSqlFile(flag.Arg(0) + ".sql").Close()
 }

src/irc/channel.go

@@ -49,8 +49,8 @@ type ChannelCommand interface {
 // NewChannel creates a new channel from a `Server` and a `name` string, which
 // must be unique on the server.
 func NewChannel(s *Server, name string) *Channel {
-	replies := make(chan Reply)
-	commands := make(chan ChannelCommand)
+	commands := make(chan ChannelCommand, 1)
+	replies := make(chan Reply, 1)
 	channel := &Channel{
 		name:     name,
 		members:  make(UserSet),
@@ -58,8 +58,8 @@ func NewChannel(s *Server, name string) *Channel {
 		commands: commands,
 		replies:  replies,
 	}
-	go channel.receiveReplies(replies)
 	go channel.receiveCommands(commands)
+	go channel.receiveReplies(replies)
 	return channel
 }
 
@@ -81,29 +81,27 @@ func (channel *Channel) Save(q Queryable) bool {
 	return true
 }
 
-// Forward `Reply`s to all `User`s of the `Channel`.
+func (channel *Channel) receiveCommands(commands <-chan ChannelCommand) {
+	for command := range commands {
+		if DEBUG_CHANNEL {
+			log.Printf("%s → %s : %s", command.Source(), channel, command)
+		}
+		command.HandleChannel(channel)
+	}
+}
+
 func (channel *Channel) receiveReplies(replies <-chan Reply) {
 	for reply := range replies {
 		if DEBUG_CHANNEL {
-			log.Printf("%s → %s", channel, reply)
+			log.Printf("%s ← %s : %s", channel, reply.Source(), reply)
 		}
 		for user := range channel.members {
 			if user != reply.Source() {
-				user.replies <- reply
+				user.Replies() <- reply
 			}
 		}
 	}
 }
-
-func (channel *Channel) receiveCommands(commands <-chan ChannelCommand) {
-	for command := range commands {
-		if DEBUG_CHANNEL {
-			log.Printf("%s ← %s %s", channel, command.Source(), command)
-		}
-		command.HandleChannel(channel)
-	}
-}
-
 func (channel *Channel) Nicks() []string {
 	return channel.members.Nicks()
 }
@@ -121,6 +119,10 @@ func (channel *Channel) GetTopic(replier Replier) {
 	replier.Replies() <- RplTopic(channel)
 }
 
+func (channel *Channel) Replies() chan<- Reply {
+	return channel.replies
+}
+
 func (channel *Channel) Id() string {
 	return channel.name
 }
@@ -133,10 +135,6 @@ func (channel *Channel) Commands() chan<- ChannelCommand {
 	return channel.commands
 }
 
-func (channel *Channel) Replies() chan<- Reply {
-	return channel.replies
-}
-
 func (channel *Channel) String() string {
 	return channel.Id()
 }
@@ -150,17 +148,17 @@ func (m *JoinCommand) HandleChannel(channel *Channel) {
 	user := client.user
 
 	if channel.key != m.channels[channel.name] {
-		client.user.replies <- ErrBadChannelKey(channel)
+		client.user.Replies() <- ErrBadChannelKey(channel)
 		return
 	}
 
 	channel.members.Add(user)
 	user.channels.Add(channel)
 
-	channel.replies <- RplJoin(channel, user)
+	channel.Replies() <- RplJoin(channel, user)
 	channel.GetTopic(user)
-	user.replies <- RplNamReply(channel)
-	user.replies <- RplEndOfNames(channel.server)
+	user.Replies() <- RplNamReply(channel)
+	user.Replies() <- RplEndOfNames(channel.server)
 }
 
 func (m *PartCommand) HandleChannel(channel *Channel) {
@@ -176,7 +174,7 @@ func (m *PartCommand) HandleChannel(channel *Channel) {
 		msg = user.Nick()
 	}
 
-	channel.replies <- RplPart(channel, user, msg)
+	channel.Replies() <- RplPart(channel, user, msg)
 
 	channel.members.Remove(user)
 	user.channels.Remove(channel)
@@ -190,7 +188,7 @@ func (m *TopicCommand) HandleChannel(channel *Channel) {
 	user := m.User()
 
 	if !channel.members[user] {
-		user.replies <- ErrNotOnChannel(channel)
+		user.Replies() <- ErrNotOnChannel(channel)
 		return
 	}
 
@@ -202,11 +200,10 @@ func (m *TopicCommand) HandleChannel(channel *Channel) {
 	channel.topic = m.topic
 
 	if channel.topic == "" {
-		channel.replies <- RplNoTopic(channel)
+		channel.Replies() <- RplNoTopic(channel)
 		return
 	}
-
-	channel.replies <- RplTopic(channel)
+	channel.Replies() <- RplTopic(channel)
 }
 
 func (m *PrivMsgCommand) HandleChannel(channel *Channel) {

src/irc/client.go

@@ -31,7 +31,7 @@ type ClientSet map[*Client]bool
 func NewClient(server *Server, conn net.Conn) *Client {
 	read := StringReadChan(conn)
 	write := StringWriteChan(conn)
-	replies := make(chan Reply)
+	replies := make(chan Reply, 1)
 
 	client := &Client{
 		conn:     conn,
@@ -67,7 +67,7 @@ func (c *Client) readConn(recv <-chan string) {
 func (c *Client) writeConn(write chan<- string, replies <-chan Reply) {
 	for reply := range replies {
 		if DEBUG_CLIENT {
-			log.Printf("%s ← %s", c, reply)
+			log.Printf("%s ← %s : %s", c, reply.Source(), reply)
 		}
 		write <- reply.Format(c)
 	}

src/irc/net.go

@@ -30,7 +30,7 @@ func StringReadChan(conn net.Conn) <-chan string {
 				break
 			}
 			if DEBUG_NET {
-				log.Printf("%s → %s", conn.RemoteAddr(), line)
+				log.Printf("%s → %s : %s", conn.RemoteAddr(), conn.LocalAddr(), line)
 			}
 			ch <- line
 		}
@@ -45,7 +45,7 @@ func StringWriteChan(conn net.Conn) chan<- string {
 	go func() {
 		for str := range ch {
 			if DEBUG_NET {
-				log.Printf("%s ← %s", conn.RemoteAddr(), str)
+				log.Printf("%s ← %s : %s", conn.RemoteAddr(), conn.LocalAddr(), str)
 			}
 			if _, err := writer.WriteString(str + "\r\n"); err != nil {
 				break

src/irc/nickserv.go

@@ -106,7 +106,8 @@ func (m *RegisterCommand) HandleNickServ(ns *NickServ) {
 		return
 	}
 
-	user := NewUser(client.nick, m.password, ns.server)
+	user := NewUser(client.nick, ns.server).SetPassword(m.password)
+	ns.server.db.Save(user)
 	ns.Reply(client, "You have registered.")
 
 	if !user.Login(client, client.nick, m.password) {

src/irc/persistence.go

@@ -27,7 +27,9 @@ type Queryable interface {
 	QueryRow(string, ...interface{}) *sql.Row
 }
 
-type TransactionFunc func(Queryable) bool
+type Savable interface {
+	Save(q Queryable) bool
+}
 
 //
 // general
@@ -36,7 +38,7 @@ type TransactionFunc func(Queryable) bool
 func NewDatabase() *Database {
 	db, err := sql.Open("sqlite3", "ergonomadic.db")
 	if err != nil {
-		panic("cannot open database")
+		log.Fatalln("cannot open database")
 	}
 	return &Database{db}
 }
@@ -48,7 +50,7 @@ func NewTransaction(tx *sql.Tx) *Transaction {
 func readLines(filename string) <-chan string {
 	file, err := os.Open(filename)
 	if err != nil {
-		panic(err)
+		log.Fatalln(err)
 	}
 	reader := bufio.NewReader(file)
 	lines := make(chan string)
@@ -56,7 +58,7 @@ func readLines(filename string) <-chan string {
 		defer file.Close()
 		defer close(lines)
 		for {
-			line, err := reader.ReadString('\n')
+			line, err := reader.ReadString(';')
 			if err != nil {
 				break
 			}
@@ -70,28 +72,24 @@ func readLines(filename string) <-chan string {
 	return lines
 }
 
-func (db *Database) execSqlFile(filename string) {
+func (db *Database) ExecSqlFile(filename string) *Database {
 	db.Transact(func(q Queryable) bool {
 		for line := range readLines(filepath.Join("sql", filename)) {
 			log.Println(line)
-			q.Exec(line)
+			_, err := q.Exec(line)
+			if err != nil {
+				log.Fatalln(err)
+			}
 		}
 		return true
 	})
+	return db
 }
 
-func (db *Database) InitTables() {
-	db.execSqlFile("init.sql")
-}
-
-func (db *Database) DropTables() {
-	db.execSqlFile("drop.sql")
-}
-
-func (db *Database) Transact(txf TransactionFunc) {
+func (db *Database) Transact(txf func(Queryable) bool) {
 	tx, err := db.Begin()
 	if err != nil {
-		panic(err)
+		log.Panicln(err)
 	}
 	if txf(tx) {
 		tx.Commit()
@@ -100,6 +98,28 @@ func (db *Database) Transact(txf TransactionFunc) {
 	}
 }
 
+func (db *Database) Save(s Savable) {
+	db.Transact(func(tx Queryable) bool {
+		return s.Save(tx)
+	})
+}
+
+//
+// general purpose sql
+//
+
+func FindId(q Queryable, sql string, args ...interface{}) (rowId RowId, err error) {
+	row := q.QueryRow(sql, args...)
+	err = row.Scan(&rowId)
+	return
+}
+
+func Count(q Queryable, sql string, args ...interface{}) (count uint, err error) {
+	row := q.QueryRow(sql, args...)
+	err = row.Scan(&count)
+	return
+}
+
 //
 // data
 //
@@ -117,25 +137,39 @@ type ChannelRow struct {
 
 // user
 
-func FindUserByNick(q Queryable, nick string) (ur *UserRow) {
-	ur = new(UserRow)
-	row := q.QueryRow("SELECT * FROM user LIMIT 1 WHERE nick = ?", nick)
-	err := row.Scan(&ur.id, &ur.nick, &ur.hash)
+func FindAllUsers(q Queryable) (urs []UserRow, err error) {
+	var rows *sql.Rows
+	rows, err = q.Query("SELECT id, nick, hash FROM user")
 	if err != nil {
-		ur = nil
+		return
+	}
+	urs = make([]UserRow, 0)
+	for rows.Next() {
+		ur := UserRow{}
+		err = rows.Scan(&(ur.id), &(ur.nick), &(ur.hash))
+		if err != nil {
+			return
+		}
+		urs = append(urs, ur)
 	}
 	return
 }
 
-func FindUserIdByNick(q Queryable, nick string) (rowId RowId, err error) {
-	row := q.QueryRow("SELECT id FROM user WHERE nick = ?", nick)
-	err = row.Scan(&rowId)
+func FindUserByNick(q Queryable, nick string) (ur *UserRow, err error) {
+	ur = &UserRow{}
+	row := q.QueryRow("SELECT id, nick, hash FROM user LIMIT 1 WHERE nick = ?",
+		nick)
+	err = row.Scan(&(ur.id), &(ur.nick), &(ur.hash))
 	return
 }
 
+func FindUserIdByNick(q Queryable, nick string) (RowId, error) {
+	return FindId(q, "SELECT id FROM user WHERE nick = ?", nick)
+}
+
 func FindChannelByName(q Queryable, name string) (cr *ChannelRow) {
 	cr = new(ChannelRow)
-	row := q.QueryRow("SELECT * FROM channel LIMIT 1 WHERE name = ?", name)
+	row := q.QueryRow("SELECT id, name FROM channel LIMIT 1 WHERE name = ?", name)
 	err := row.Scan(&(cr.id), &(cr.name))
 	if err != nil {
 		cr = nil
@@ -185,25 +219,31 @@ func InsertUserChannels(q Queryable, userId RowId, channelIds []RowId) (err erro
 
 // channel
 
-func FindChannelIdByName(q Queryable, name string) (channelId RowId, err error) {
-	row := q.QueryRow("SELECT id FROM channel WHERE name = ?", name)
-	err = row.Scan(&channelId)
-	return
+func FindChannelIdByName(q Queryable, name string) (RowId, error) {
+	return FindId(q, "SELECT id FROM channel WHERE name = ?", name)
 }
 
-func FindChannelsForUser(q Queryable, userId RowId) (crs []ChannelRow) {
-	rows, err := q.Query(`SELECT * FROM channel WHERE id IN
-(SELECT channel_id from user_channel WHERE user_id = ?)`, userId)
+func FindChannelsForUser(q Queryable, userId RowId) (crs []ChannelRow, err error) {
+	query := ` FROM channel WHERE id IN
+(SELECT channel_id from user_channel WHERE user_id = ?)`
+	count, err := Count(q, "SELECT COUNT(id)"+query, userId)
 	if err != nil {
-		panic(err)
+		return
 	}
-	crs = make([]ChannelRow, 0)
+	rows, err := q.Query("SELECT id, name"+query, userId)
+	if err != nil {
+		return
+	}
+	crs = make([]ChannelRow, count)
+	var i = 0
 	for rows.Next() {
 		cr := ChannelRow{}
-		if err := rows.Scan(&(cr.id), &(cr.name)); err != nil {
-			panic(err)
+		err = rows.Scan(&(cr.id), &(cr.name))
+		if err != nil {
+			return
 		}
-		crs = append(crs, cr)
+		crs[i] = cr
+		i++
 	}
 	return
 }

src/irc/server.go

@@ -24,10 +24,11 @@ type Server struct {
 	channels ChannelNameMap
 	services ServiceNameMap
 	commands chan<- Command
+	db       *Database
 }
 
 func NewServer(name string) *Server {
-	commands := make(chan Command)
+	commands := make(chan Command, 1)
 	server := &Server{
 		ctime:    time.Now(),
 		name:     name,
@@ -35,16 +36,27 @@ func NewServer(name string) *Server {
 		users:    make(UserNameMap),
 		channels: make(ChannelNameMap),
 		services: make(ServiceNameMap),
+		db:       NewDatabase(),
 	}
 	go server.receiveCommands(commands)
 	NewNickServ(server)
+	server.db.Transact(func(q Queryable) bool {
+		urs, err := FindAllUsers(server.db)
+		if err != nil {
+			return false
+		}
+		for _, ur := range urs {
+			NewUser(ur.nick, server).SetHash(ur.hash)
+		}
+		return false
+	})
 	return server
 }
 
 func (server *Server) receiveCommands(commands <-chan Command) {
 	for command := range commands {
 		if DEBUG_SERVER {
-			log.Printf("%s ← %s %s", server, command.Client(), command)
+			log.Printf("%s → %s : %s", command.Client(), server, command)
 		}
 		command.Client().atime = time.Now()
 		command.HandleServer(server)
@@ -278,7 +290,7 @@ func (m *TopicCommand) HandleServer(s *Server) {
 
 	channel := s.channels[m.channel]
 	if channel == nil {
-		user.Replies() <- ErrNoSuchChannel(s, m.channel)
+		m.Client().Replies() <- ErrNoSuchChannel(s, m.channel)
 		return
 	}
 

src/irc/service.go

@@ -29,7 +29,7 @@ type BaseService struct {
 }
 
 func NewService(service EditableService, s *Server, name string) Service {
-	commands := make(chan ServiceCommand)
+	commands := make(chan ServiceCommand, 1)
 	base := &BaseService{
 		server:   s,
 		name:     name,

src/irc/user.go

@@ -46,9 +46,9 @@ func (set UserSet) Nicks() []string {
 	return nicks
 }
 
-func NewUser(nick string, password string, server *Server) *User {
-	commands := make(chan UserCommand)
-	replies := make(chan Reply)
+func NewUser(nick string, server *Server) *User {
+	commands := make(chan UserCommand, 1)
+	replies := make(chan Reply, 1)
 	user := &User{
 		nick:     nick,
 		server:   server,
@@ -56,7 +56,6 @@ func NewUser(nick string, password string, server *Server) *User {
 		channels: make(ChannelSet),
 		replies:  replies,
 	}
-	user.SetPassword(password)
 
 	go user.receiveCommands(commands)
 	go user.receiveReplies(replies)
@@ -81,34 +80,40 @@ func (user *User) Save(q Queryable) bool {
 		}
 	}
 
+	userId := *(user.id)
 	channelIds := user.channels.Ids()
 	if len(channelIds) == 0 {
-		if err := DeleteAllUserChannels(q, *(user.id)); err != nil {
+		if err := DeleteAllUserChannels(q, userId); err != nil {
 			return false
 		}
 	} else {
-		if err := DeleteOtherUserChannels(q, *(user.id), channelIds); err != nil {
+		if err := DeleteOtherUserChannels(q, userId, channelIds); err != nil {
 			return false
 		}
-		if err := InsertUserChannels(q, *(user.id), channelIds); err != nil {
+		if err := InsertUserChannels(q, userId, channelIds); err != nil {
 			return false
 		}
 	}
 	return true
 }
 
-func (user *User) SetPassword(password string) {
+func (user *User) SetPassword(password string) *User {
 	hash, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
 	if err != nil {
 		panic("bcrypt failed; cannot generate password hash")
 	}
+	return user.SetHash(hash)
+}
+
+func (user *User) SetHash(hash []byte) *User {
 	user.hash = hash
+	return user
 }
 
 func (user *User) receiveCommands(commands <-chan UserCommand) {
 	for command := range commands {
 		if DEBUG_USER {
-			log.Printf("%s ← %s %s", user, command.Client(), command)
+			log.Printf("%s → %s : %s", command.Client(), user, command)
 		}
 		command.HandleUser(user)
 	}
@@ -117,7 +122,9 @@ func (user *User) receiveCommands(commands <-chan UserCommand) {
 // Distribute replies to clients.
 func (user *User) receiveReplies(replies <-chan Reply) {
 	for reply := range replies {
-		log.Printf("%s ← %s", user, reply)
+		if DEBUG_USER {
+			log.Printf("%s ← %s : %s", user, reply.Source(), reply)
+		}
 		for client := range user.clients {
 			client.Replies() <- reply
 		}