mirror of https://github.com/gophish/gophish
45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"github.com/jordan-wright/gophish/config"
|
||
|
|
_ "github.com/mattn/go-sqlite3"
|
||
|
|
)
|
||
|
|
|
||
|
|
var Conn *sql.DB
|
||
|
|
|
||
|
|
// Setup initializes the Conn object
|
||
|
|
// It also populates the Gophish Config object
|
||
|
|
func Setup() error {
|
||
|
|
//If the file already exists, delete it and recreate it
|
||
|
|
_, err := os.Stat(config.Conf.DBPath)
|
||
|
|
if err == nil {
|
||
|
|
os.Remove(config.Conf.DBPath)
|
||
|
|
}
|
||
|
|
fmt.Println("Creating db at " + config.Conf.DBPath)
|
||
|
|
Conn, err = sql.Open("sqlite3", config.Conf.DBPath)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
//Create the tables needed
|
||
|
|
_, err = Conn.Exec(
|
||
|
|
`CREATE TABLE Users (id INTEGER PRIMARY KEY AUTOINCREMENT, username TEXT, hash VARCHAR(60), apikey VARCHAR(32));`)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
//Create the default user
|
||
|
|
stmt, err := Conn.Prepare(`INSERT INTO Users (username, hash, apikey) VALUES (?, ?, ?);`)
|
||
|
|
defer stmt.Close()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
_, err = stmt.Exec("jordan", "$2a$10$d4OtT.RkEOQn.iruVWIQ5u8CeV/85ZYF41y8wKeUwsAPqPNFvTccW", "12345678901234567890123456789012")
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|