80 lines
2.5 KiB
Go
80 lines
2.5 KiB
Go
package multiaccounts
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/status-im/status-go/multiaccounts/migrations"
|
|
"github.com/status-im/status-go/sqlite"
|
|
)
|
|
|
|
// Account stores public information about account.
|
|
type Account struct {
|
|
Name string `json:"name"`
|
|
Address common.Address `json:"address"`
|
|
Timestamp int64 `json:"timestamp"`
|
|
PhotoPath string `json:"photo-path"`
|
|
KeycardPairing string `json:"keycard-pairing"`
|
|
KeycardKeyUID string `json:"keycard-key-uid"`
|
|
}
|
|
|
|
// InitializeDB creates db file at a given path and applies migrations.
|
|
func InitializeDB(path string) (*Database, error) {
|
|
db, err := sqlite.OpenUnecryptedDB(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
err = migrations.Migrate(db)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Database{db: db}, nil
|
|
}
|
|
|
|
type Database struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func (db *Database) Close() error {
|
|
return db.db.Close()
|
|
}
|
|
|
|
func (db *Database) GetAccounts() ([]Account, error) {
|
|
rows, err := db.db.Query("SELECT address, name, loginTimestamp, photoPath, keycardPairing, keycardKeyUid from accounts ORDER BY loginTimestamp DESC")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rst := []Account{}
|
|
inthelper := sql.NullInt64{}
|
|
for rows.Next() {
|
|
acc := Account{}
|
|
err = rows.Scan(&acc.Address, &acc.Name, &inthelper, &acc.PhotoPath, &acc.KeycardPairing, &acc.KeycardKeyUID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
acc.Timestamp = inthelper.Int64
|
|
rst = append(rst, acc)
|
|
}
|
|
return rst, nil
|
|
}
|
|
|
|
func (db *Database) SaveAccount(account Account) error {
|
|
_, err := db.db.Exec("INSERT OR REPLACE INTO accounts (address, name, photoPath, keycardPairing, keycardKeyUid) VALUES (?, ?, ?, ?, ?)", account.Address, account.Name, account.PhotoPath, account.KeycardPairing, account.KeycardKeyUID)
|
|
return err
|
|
}
|
|
|
|
func (db *Database) UpdateAccount(account Account) error {
|
|
_, err := db.db.Exec("UPDATE accounts SET name = ?, photoPath = ?, keycardPairing = ?, keycardKeyUid = ? WHERE address = ?", account.Name, account.PhotoPath, account.KeycardPairing, account.KeycardKeyUID, account.Address)
|
|
return err
|
|
}
|
|
|
|
func (db *Database) UpdateAccountTimestamp(address common.Address, loginTimestamp int64) error {
|
|
_, err := db.db.Exec("UPDATE accounts SET loginTimestamp = ? WHERE address = ?", loginTimestamp, address)
|
|
return err
|
|
}
|
|
|
|
func (db *Database) DeleteAccount(address common.Address) error {
|
|
_, err := db.db.Exec("DELETE FROM accounts WHERE address = ?", address)
|
|
return err
|
|
}
|