|
1 | 1 | package db |
2 | 2 |
|
3 | 3 | import ( |
4 | | - "errors" |
5 | | - "fmt" |
| 4 | + "database/sql" |
6 | 5 | "os" |
7 | 6 | "path" |
8 | | - "time" |
9 | 7 |
|
10 | 8 | "github.com/CosmicPredator/chibi/internal" |
11 | | - bolt "go.etcd.io/bbolt" |
| 9 | + _ "github.com/mattn/go-sqlite3" |
12 | 10 | ) |
13 | 11 |
|
14 | | -// define DbContext type to make CRUD operations |
15 | | -type DbContext struct {} |
16 | | - |
17 | | -// creates a new bucket and if exists, it'll skip |
18 | | -func (dc DbContext) createBucket() error { |
19 | | - db, err := dc.openDbConn() |
20 | | - defer db.Close() |
21 | | - if err != nil { |
22 | | - return err |
23 | | - } |
24 | | - return db.Update(func(tx *bolt.Tx) error { |
25 | | - _, err := tx.CreateBucketIfNotExists([]byte(internal.BOLT_BUCKET_NAME)) |
26 | | - if err != nil { |
27 | | - return err |
28 | | - } |
29 | | - return nil |
30 | | - }) |
| 12 | +type DbContext struct { |
| 13 | + dbConn *sql.DB |
31 | 14 | } |
32 | 15 |
|
33 | | -// cleans up and prepares os sepcific config folder |
34 | | -// to save .db file |
35 | | -func (dc DbContext) InitDB() error { |
36 | | - osConfigPath, _ := os.UserConfigDir() |
37 | | - configDir := path.Join(osConfigPath, "chibi") |
38 | | - _, err := os.Stat(configDir) |
| 16 | +// creates required SQL tables |
| 17 | +func (dc *DbContext) createRequiredTables() error { |
| 18 | + tx, err := dc.dbConn.Begin() |
| 19 | + if err != nil { |
| 20 | + return err |
| 21 | + } |
| 22 | + |
| 23 | + if _, err = tx.Exec(QUERY_CREATE_TABLE); err != nil { |
| 24 | + return err |
| 25 | + } |
39 | 26 |
|
40 | | - if err == nil { |
41 | | - os.RemoveAll(configDir) |
| 27 | + if err = tx.Commit(); err != nil { |
| 28 | + tx.Rollback() |
| 29 | + return err |
42 | 30 | } |
43 | | - os.MkdirAll(configDir, 0755) |
44 | | - return dc.createBucket() |
| 31 | + return nil |
45 | 32 | } |
46 | 33 |
|
47 | | -// opens the .db file and returns the instance |
48 | | -func (dc *DbContext) openDbConn() (*bolt.DB, error) { |
49 | | - osConfigDir, _ := os.UserConfigDir() |
50 | | - configFilePath := path.Join(osConfigDir, "chibi", internal.BOLT_DB_NAME) |
| 34 | +// initialize an SQL instance. |
| 35 | +// if isFirstTime is true, it'll clean the os config dir |
| 36 | +// and creates a brand new table |
| 37 | +func (dc *DbContext) Init(isFirstTime bool) error { |
| 38 | + osConfigPath, _ := os.UserConfigDir() |
| 39 | + configDir := path.Join(osConfigPath, "chibi") |
| 40 | + |
| 41 | + if isFirstTime { |
| 42 | + internal.CreateConfigDir() |
| 43 | + } |
| 44 | + dbPath := path.Join(configDir, "chibi_config.db") |
51 | 45 |
|
52 | | - db, err := bolt.Open(configFilePath, 0755, &bolt.Options{ |
53 | | - Timeout: 5 * time.Second, |
54 | | - }) |
| 46 | + db, err := sql.Open("sqlite3", dbPath) |
55 | 47 | if err != nil { |
56 | | - return nil, err |
| 48 | + return err |
57 | 49 | } |
58 | | - return db, nil |
| 50 | + dc.dbConn = db |
| 51 | + if isFirstTime { |
| 52 | + err := dc.createRequiredTables() |
| 53 | + if err != nil { |
| 54 | + return err |
| 55 | + } |
| 56 | + } |
| 57 | + return nil |
59 | 58 | } |
60 | 59 |
|
61 | | -// writes the key value pair to db |
62 | | -func (dc DbContext) Set(key string, value string) error { |
63 | | - db, err := dc.openDbConn() |
64 | | - defer db.Close() |
65 | | - if err != nil { |
66 | | - return err |
67 | | - } |
68 | | - err = db.Update(func(tx *bolt.Tx) error { |
69 | | - b := tx.Bucket([]byte(internal.BOLT_BUCKET_NAME)) |
70 | | - if b == nil { |
71 | | - return errors.New("DB Bucket does not exist") |
72 | | - } |
73 | | - err := b.Put([]byte(key), []byte(value)) |
| 60 | +// add a key value pair to config table |
| 61 | +func (dc *DbContext) SetConfig(key string, value string) error { |
| 62 | + tx, err := dc.dbConn.Begin() |
| 63 | + if err != nil { |
| 64 | + return err |
| 65 | + } |
| 66 | + |
| 67 | + if _, err = tx.Exec(QUERY_INSERT_CONFIG, key, value); err != nil { |
74 | 68 | return err |
75 | | - }) |
76 | | - return err |
| 69 | + } |
| 70 | + |
| 71 | + if err = tx.Commit(); err != nil { |
| 72 | + tx.Rollback() |
| 73 | + return err |
| 74 | + } |
| 75 | + |
| 76 | + return nil |
77 | 77 | } |
78 | 78 |
|
79 | | -// reads the value of specified key from db |
80 | | -func (dc DbContext) Get(key string) (string, error) { |
81 | | - db, err := dc.openDbConn() |
82 | | - defer db.Close() |
83 | | - if err != nil { |
84 | | - return "", err |
85 | | - } |
| 79 | +// gets a key's value from the config table |
| 80 | +func (dc *DbContext) GetConfig(key string) (*string, error) { |
| 81 | + tx, err := dc.dbConn.Begin() |
| 82 | + if err != nil { |
| 83 | + return nil, err |
| 84 | + } |
86 | 85 |
|
| 86 | + row := tx.QueryRow(QUERY_GET_CONFIG, key) |
87 | 87 | var value string |
88 | | - err = db.View(func(tx *bolt.Tx) error { |
89 | | - b := tx.Bucket([]byte(internal.BOLT_BUCKET_NAME)) |
90 | | - if b == nil { |
91 | | - return errors.New("DB Bucket does not exist") |
92 | | - } |
93 | | - value = string(b.Get([]byte(key))) |
94 | | - if value == "" { |
95 | | - return fmt.Errorf("no value found for the key %s", key) |
96 | | - } |
97 | | - return nil |
98 | | - }) |
99 | | - if err != nil { |
100 | | - return "", err |
| 88 | + |
| 89 | + if err = row.Scan(&value); err != nil { |
| 90 | + return nil, err |
101 | 91 | } |
102 | | - return value, err |
| 92 | + |
| 93 | + if err = tx.Commit(); err != nil { |
| 94 | + return nil, err |
| 95 | + } |
| 96 | + return &value, nil |
| 97 | +} |
| 98 | + |
| 99 | +// closes the DB instance after usage |
| 100 | +func (dc *DbContext) Close() { |
| 101 | + dc.dbConn.Close() |
103 | 102 | } |
104 | 103 |
|
| 104 | +// returns a brand new DbContext instance |
105 | 105 | func NewDbConn() *DbContext { |
106 | 106 | return &DbContext{} |
107 | 107 | } |
0 commit comments