1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
package internal
import (
"database/sql"
"fmt"
"os"
"golang.org/x/crypto/bcrypt"
)
func Migrate(db *sql.DB, webmaster string, passOne string, passTwo string) {
_, table_check := db.Query("SELECT * FROM logins;")
if table_check != nil {
_, err := db.Exec("CREATE TABLE logins (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, pass_one TEXT NOT NULL, pass_two TEXT NOT NULL);")
if err != nil {
fmt.Fprintf(os.Stderr, "Unable to create logins table: %v\n", err)
os.Exit(1)
}
hashOne, errHashOne := bcrypt.GenerateFromPassword([]byte(passOne), 12)
hashTwo, errHashTwo := bcrypt.GenerateFromPassword([]byte(passTwo), 12)
if errHashOne != nil || errHashTwo != nil {
fmt.Fprintf(os.Stderr, "Failed to hash password")
os.Exit(1)
}
_, err = db.Exec(fmt.Sprintf("INSERT INTO logins (name, pass_one, pass_two) VALUES ('%s', '%s', '%s');", webmaster, hashOne, hashTwo))
if err != nil {
fmt.Fprintf(os.Stderr, "Unable to add webmaster to logins table: %v\n", err)
os.Exit(1)
}
}
_, table_check = db.Query("SELECT * FROM posts;")
if table_check != nil {
_, err := db.Exec("CREATE TABLE posts (id SERIAL PRIMARY KEY, user_id INTEGER REFERENCES logins(id), time TIMESTAMP DEFAULT CURRENT_TIMESTAMP, brief TEXT NOT NULL, content TEXT NOT NULL);")
if err != nil {
fmt.Fprintf(os.Stderr, "Unable to create posts table: %v\n", err)
os.Exit(1)
}
}
_, table_check = db.Query("SELECT * FROM feeds;")
if table_check != nil {
_, err := db.Exec("CREATE TABLE feeds (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, user_id INTEGER REFERENCES logins(id), url VARCHAR(255) NOT NULL, category INTEGER);")
if err != nil {
fmt.Fprintf(os.Stderr, "Unable to create feeds table: %v\n", err)
os.Exit(1)
}
}
_, table_check = db.Query("SELECT * FROM cookies;")
if table_check != nil {
_, err := db.Exec("CREATE TABLE cookies (id SERIAL PRIMARY KEY, content VARCHAR(255) NOT NULL, user_id INTEGER REFERENCES logins(id), expiration TIMESTAMP);")
if err != nil {
fmt.Fprintf(os.Stderr, "Unable to create cookies table: %v\n", err)
}
}
}
|
