1 files changed, 45 insertions, 2 deletions
diff --git a/main.go b/main.go
index 94f1e3a..beb3aa1 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,8 @@ import (
 
 	_ "database/sql"
 	_ "github.com/go-sql-driver/mysql"
+	_ "github.com/mattn/go-sqlite3"
+
 	"github.com/jmoiron/sqlx"
 )
 
@@ -43,34 +45,75 @@ func main() {
 	app := App{Config: &Config{}}
 	app.Config.parseConfig("")
 
-	// Hard coded mysql driver.
-	app.DB, err = sqlx.Connect("mysql", app.Config.DBUser+":"+app.Config.DBPassword+"@"+app.Config.DBHost+"/"+app.Config.DBDBName+app.Config.DBOptions)
+	if "sqlite3" == app.Config.DBDriver {
+		//app.DB, err = sqlx.Connect(app.Config.DBDriver, app.Config.DBPath)
+		app.DB, err = sqlx.Connect(app.Config.DBDriver, "./alkobote.db")
+	} else {
+		app.DB, err = sqlx.Connect(app.Config.DBDriver, app.Config.DBUser+":"+app.Config.DBPassword+"@"+app.Config.DBHost+"/"+app.Config.DBDBName+app.Config.DBOptions)
+	}
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	err = app.createTables()
 	if err != nil {
 		log.Fatal(err)
 	}
 
 	W := ScrapeBottleWord()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "BottleWorld")
 
 	W = ScrapeMCWhisky()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "MC Whisky")
 
 	W = ScrapeRumundCo()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Rum und Co")
 
 	W = ScrapeWhic()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Whic")
 
 	W = ScrapeWhiskyde()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Whisky.de")
 
 	W = ScrapeWhiskysitenl()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Whiskysite.nl")
 
 	W = ScrapeWhiskyworld()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Whiskyworld")
 
 	W = ScrapeWhiskyzone()
+	err = app.save_offer(W)
+	if err != nil {
+		log.Fatal(err)
+	}
 	printName(W, "Whiskyzone")
 }