Adds repair function. (crawler)

1 files changed, 13 insertions, 9 deletions

diff --git a/crawler/config.go b/crawler/config.go
index c9b17a4..f74253e 100644
--- a/crawler/config.go
+++ b/crawler/config.go
@@ -5,19 +5,21 @@ import (
 )
 
 type Config struct {
-	DBDriver          string
-	DBDBName          string
-	DBHost            string
-	DBPort            string
-	DBUser            string
-	DBPassword        string
-	DBOptions         string
-	DBPath            string // for sqlite
+	DBDriver   string
+	DBDBName   string
+	DBHost     string
+	DBPort     string
+	DBUser     string
+	DBPassword string
+	DBOptions  string
+	DBPath     string // for sqlite
+
 	DisableURLShorter bool
 	Polr_URL          string
 	Polr_API_Key      string
 
-	Debug bool
+	Debug       bool
+	FixDatabase bool
 }
 
 // Parses the configuration and sets the configuration struct.
@@ -31,6 +33,7 @@ func (c *Config) parseConfig(configFile string) {
 	viper.SetDefault("DB_Path", "./alkobote.db")
 
 	viper.SetDefault("Debug", false)
+	viper.SetDefault("FixDatabase", false)
 	viper.SetDefault("DisableURLShorter", false)
 
 	// Name of the configuration file
@@ -72,6 +75,7 @@ func (c *Config) setsConfig() {
 	c.DBOptions = viper.GetString("DB_Options")
 	c.DBPath = viper.GetString("DB_Path")
 	c.Debug = viper.GetBool("Debug")
+	c.FixDatabase = viper.GetBool("FixDatabase")
 	c.DisableURLShorter = viper.GetBool("DisableURLShorter")
 	c.Polr_URL = viper.GetString("Polr_URL")
 	c.Polr_API_Key = viper.GetString("Polr_API_Key")