From d6d3a108170127f7f1b9d4b6efcd7aaebd608d65 Mon Sep 17 00:00:00 2001
From: Shivaram Lingamneni <slingamn@cs.stanford.edu>
Date: Mon, 31 Dec 2018 01:17:44 -0500
Subject: [PATCH] implement #322

Automatically create the datastore on `oragono run` if it doesn't exist.
See also #302.
---
 irc/database.go | 30 ++++++++++++++++++------------
 irc/server.go   |  9 +++++++++
 2 files changed, 27 insertions(+), 12 deletions(-)

diff --git a/irc/database.go b/irc/database.go
index 6fef867b..635ffee5 100644
--- a/irc/database.go
+++ b/irc/database.go
@@ -52,14 +52,26 @@ func (err *incompatibleSchemaError) Error() string {
 	return fmt.Sprintf("Database requires update. Expected schema v%s, got v%s", err.requiredVersion, err.currentVersion)
 }
 
-// InitDB creates the database.
+// InitDB creates the database, implementing the `oragono initdb` command.
 func InitDB(path string) {
-	// prepare kvstore db
-	//TODO(dan): fail if already exists instead? don't want to overwrite good data
-	os.Remove(path)
+	_, err := os.Stat(path)
+	if err == nil {
+		log.Fatal("Datastore already exists (delete it manually to continue): ", path)
+	} else if !os.IsNotExist(err) {
+		log.Fatal("Datastore path is inaccessible: ", err.Error())
+	}
+
+	err = initializeDB(path)
+	if err != nil {
+		log.Fatal("Could not save datastore: ", err.Error())
+	}
+}
+
+// internal database initialization code
+func initializeDB(path string) error {
 	store, err := buntdb.Open(path)
 	if err != nil {
-		log.Fatal(fmt.Sprintf("Failed to open datastore: %s", err.Error()))
+		return err
 	}
 	defer store.Close()
 
@@ -69,9 +81,7 @@ func InitDB(path string) {
 		return nil
 	})
 
-	if err != nil {
-		log.Fatal("Could not save datastore:", err.Error())
-	}
+	return err
 }
 
 // OpenDatabase returns an existing database, performing a schema version check.
@@ -81,10 +91,6 @@ func OpenDatabase(config *Config) (*buntdb.DB, error) {
 
 // open the database, giving it at most one chance to auto-upgrade the schema
 func openDatabaseInternal(config *Config, allowAutoupgrade bool) (db *buntdb.DB, err error) {
-	_, err = os.Stat(config.Datastore.Path)
-	if os.IsNotExist(err) {
-		return
-	}
 	db, err = buntdb.Open(config.Datastore.Path)
 	if err != nil {
 		return
diff --git a/irc/server.go b/irc/server.go
index 1afee925..e1e29ae0 100644
--- a/irc/server.go
+++ b/irc/server.go
@@ -885,6 +885,15 @@ func (server *Server) loadDatastore(config *Config) error {
 	// open the datastore and load server state for which it (rather than config)
 	// is the source of truth
 
+	_, err := os.Stat(config.Datastore.Path)
+	if os.IsNotExist(err) {
+		server.logger.Warning("startup", "database does not exist, creating it", config.Datastore.Path)
+		err = initializeDB(config.Datastore.Path)
+		if err != nil {
+			return err
+		}
+	}
+
 	db, err := OpenDatabase(config)
 	if err == nil {
 		server.store = db