Added support for sqlite as database backend

app.go

@@ -22,6 +22,8 @@ type Config struct {
 	PrivateKeyPath string
 	DerpMap        *tailcfg.DERPMap
 
+	DBtype string
+	DBpath string
 	DBhost string
 	DBport int
 	DBname string
@@ -60,11 +62,22 @@ func NewHeadscale(cfg Config) (*Headscale, error) {
 		return nil, err
 	}
 	pubKey := privKey.Public()
+
+	var dbString string
+	switch cfg.DBtype {
+	case "postgres":
+		dbString = fmt.Sprintf("host=%s port=%d dbname=%s user=%s password=%s sslmode=disable", cfg.DBhost,
+			cfg.DBport, cfg.DBname, cfg.DBuser, cfg.DBpass)
+	case "sqlite3":
+		dbString = cfg.DBpath
+	default:
+		return nil, errors.New("Unsupported DB")
+	}
+
 	h := Headscale{
 		cfg:        cfg,
-		dbType: "postgres",
+		dbType:     cfg.DBtype,
-		dbString: fmt.Sprintf("host=%s port=%d dbname=%s user=%s password=%s sslmode=disable", cfg.DBhost,
+		dbString:   dbString,
-			cfg.DBport, cfg.DBname, cfg.DBuser, cfg.DBpass),
 		privateKey: privKey,
 		publicKey:  &pubKey,
 	}

cmd/headscale/cli/utils.go

@@ -43,6 +43,8 @@ func getHeadscaleApp() (*headscale.Headscale, error) {
 		PrivateKeyPath: absPath(viper.GetString("private_key_path")),
 		DerpMap:        derpMap,
 
+		DBtype: viper.GetString("db_type"),
+		DBpath: viper.GetString("db_path"),
 		DBhost: viper.GetString("db_host"),
 		DBport: viper.GetInt("db_port"),
 		DBname: viper.GetString("db_name"),

db.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/postgres" // sql driver
+	_ "github.com/jinzhu/gorm/dialects/sqlite"   // sql driver
 )
 
 const dbVersion = "1"