Add database support, basic client-side event processing

asink.go

@@ -2,13 +2,24 @@ package main
 
 import (
 	"code.google.com/p/goconf/conf"
+	"database/sql"
 	"flag"
 	"fmt"
+	"os"
 	"os/user"
 	"path"
 )
 
-var configFileName string
+type AsinkGlobals struct {
+	configFileName string
+	syncDir        string
+	cacheDir       string
+	tmpDir         string
+	db             *sql.DB
+	storage        Storage
+}
+
+var globals AsinkGlobals
 
 func init() {
 	const config_usage = "Config File to use"
@@ -19,50 +30,121 @@ func init() {
 		userHomeDir = u.HomeDir
 	}
 
-	flag.StringVar(&configFileName, "config", path.Join(userHomeDir, ".asink", "config"), config_usage)
-	flag.StringVar(&configFileName, "c", path.Join(userHomeDir, ".asink", "config"), config_usage+" (shorthand)")
+	flag.StringVar(&globals.configFileName, "config", path.Join(userHomeDir, ".asink", "config"), config_usage)
+	flag.StringVar(&globals.configFileName, "c", path.Join(userHomeDir, ".asink", "config"), config_usage+" (shorthand)")
 }
 
 func main() {
 	flag.Parse()
-	fmt.Println("config file:", configFileName)
+	fmt.Println("config file:", globals.configFileName)
 
-	config, err := conf.ReadConfigFile(configFileName)
+	config, err := conf.ReadConfigFile(globals.configFileName)
 	if err != nil {
 		fmt.Println(err)
-		fmt.Println("Error reading config file at ", configFileName, ". Does it exist?")
+		fmt.Println("Error reading config file at ", globals.configFileName, ". Does it exist?")
 		return
 	}
 
-	storage, err := GetStorage(config)
+	globals.storage, err = GetStorage(config)
 	if err != nil {
 		fmt.Println(err)
 		return
 	}
 
-	syncdir, err := config.GetString("local", "syncdir")
-	cachedir, err := config.GetString("local", "cachedir")
+	globals.syncDir, err = config.GetString("local", "syncdir")
+	globals.cacheDir, err = config.GetString("local", "cachedir")
+	globals.tmpDir, err = config.GetString("local", "tmpdir")
 
-	fmt.Println(syncdir)
-	fmt.Println(cachedir)
-	fmt.Println(storage)
+	//make sure all the necessary directories exist
+	err = ensureDirExists(globals.syncDir)
+	if err != nil {
+		panic(err)
+	}
+	err = ensureDirExists(globals.cacheDir)
+	if err != nil {
+		panic(err)
+	}
+	err = ensureDirExists(globals.tmpDir)
+	if err != nil {
+		panic(err)
+	}
+
+	//TODO FIXME REMOVEME
+	fmt.Println(globals.syncDir)
+	fmt.Println(globals.cacheDir)
+	fmt.Println(globals.tmpDir)
+	fmt.Println(globals.storage)
+	//TODO FIXME REMOVEME
 
 	fileUpdates := make(chan *Event)
-	go StartWatching(syncdir, fileUpdates)
+	go StartWatching(globals.syncDir, fileUpdates)
+
+	globals.db, err = GetAndInitDB(config)
+	if err != nil {
+		panic(err)
+		return
+	}
 
 	for {
 		event := <-fileUpdates
-		ProcessEvent(storage, event)
+		ProcessEvent(globals, event)
 	}
 }
 
-func ProcessEvent(storage Storage, event *Event) {
-	fmt.Println(event)
+func ProcessEvent(globals AsinkGlobals, event *Event) {
+	//add to database
+	err := DatabaseAddEvent(globals.db, event)
+	if err != nil {
+		panic(err)
+	}
 
 	if event.IsUpdate() {
-		err := storage.Put(event.Path, event.Hash)
+		//copy to tmp
+		tmpfilename, err := copyToTmp(event.Path, globals.tmpDir)
 		if err != nil {
 			panic(err)
 		}
+		event.Status |= COPIED_TO_TMP
+
+		//get the file's hash
+		hash, err := HashFile(tmpfilename)
+		event.Hash = hash
+		if err != nil {
+			panic(err)
+		}
+		event.Status |= HASHED
+
+		//rename to local cache w/ filename=hash
+		err = os.Rename(tmpfilename, path.Join(globals.cacheDir, event.Hash))
+		if err != nil {
+			err := os.Remove(tmpfilename)
+			if err != nil {
+				panic(err)
+			}
+		}
+		event.Status |= CACHED
+
+		//update database
+		err = DatabaseUpdateEvent(globals.db, event)
+		if err != nil {
+			panic(err)
+		}
+
+		//upload file to remote storage
+		err = globals.storage.Put(event.Path, event.Hash)
+		if err != nil {
+			panic(err)
+		}
+		event.Status |= UPLOADED
+
+		//update database again
+		err = DatabaseUpdateEvent(globals.db, event)
+		if err != nil {
+			panic(err)
+		}
+
 	}
+	fmt.Println(event)
+
+	//TODO notify server of new file
 }