package accounts import ( "database/sql" "encoding/json" "errors" "github.com/status-im/status-go/eth-node/types" "github.com/status-im/status-go/params" "github.com/status-im/status-go/sqlite" ) const ( uniqueChatConstraint = "UNIQUE constraint failed: accounts.chat" uniqueWalletConstraint = "UNIQUE constraint failed: accounts.wallet" ) var ( // ErrWalletNotUnique returned if another account has `wallet` field set to true. ErrWalletNotUnique = errors.New("another account is set to be default wallet. disable it before using new") // ErrChatNotUnique returned if another account has `chat` field set to true. ErrChatNotUnique = errors.New("another account is set to be default chat. disable it before using new") // ErrInvalidConfig returned if config isn't allowed ErrInvalidConfig = errors.New("configuration value not allowed") ) type Account struct { Address types.Address `json:"address"` Wallet bool `json:"wallet"` Chat bool `json:"chat"` Type string `json:"type,omitempty"` Storage string `json:"storage,omitempty"` Path string `json:"path,omitempty"` PublicKey types.HexBytes `json:"public-key,omitempty"` Name string `json:"name"` Color string `json:"color"` } type Settings struct { // required Address types.Address `json:"address"` ChaosMode bool `json:"chaos-mode?,omitempty"` Currency string `json:"currency,omitempty"` CurrentNetwork string `json:"networks/current-network"` CustomBootnodes *json.RawMessage `json:"custom-boot-nodes,omitempty"` CustomBootnodesEnabled *json.RawMessage `json:"custom-boot-nodes-enabled,omitempty"` DappsAddress types.Address `json:"dapps-address"` EIP1581Address types.Address `json:"eip1581-address"` Fleet *string `json:"fleet,omitempty"` HideHomeTooltip bool `json:"hide-home-tooltip?,omitempty"` InstallationID string `json:"installation-id"` KeyUID string `json:"key-uid"` KeycardInstanceUID string `json:"keycard-instance-uid,omitempty"` KeycardPAiredOn int64 `json:"keycard-paired-on,omitempty"` KeycardPairing string `json:"keycard-pairing,omitempty"` LastUpdated *int64 `json:"last-updated,omitempty"` LatestDerivedPath uint `json:"last-derived-path"` LogLevel *string `json:"log-level,omitempty"` Mnemonic string `json:"mnemonic,omitempty"` Name string `json:"name,omitempty"` Networks *json.RawMessage `json:"networks/networks"` NotificationsEnabled bool `json:"notifications-enabled?,omitempty"` PhotoPath string `json:"photo-path"` PinnedMailserver *json.RawMessage `json:"pinned-mailservers,omitempty"` PreferredName *string `json:"preferred-name,omitempty"` PreviewPrivacy bool `json:"preview-privacy?"` PublicKey string `json:"public-key"` RememberSyncingChoice bool `json:"remember-syncing-choice?,omitempty"` ShowName *bool `json:"show-name?,omitempty"` SigningPhrase string `json:"signing-phrase"` StickerPacksInstalled *json.RawMessage `json:"stickers-packs-installed,omitempty"` StickersRecentStickers *json.RawMessage `json:"stickers-recent-stickers,omitempty"` SyncingOnMobileNetwork bool `json:"syncing-on-mobile-network?,omitempty"` Usernames *[]string `json:"usernames,omitempty"` WalletRootAddress types.Address `json:"wallet-root-address,omitempty"` WalletSetupPassed bool `json:"wallet-setup-passed?,omitempty"` WalletVisibleTokens *json.RawMessage `json:"wallet/visible-tokens,omitempty"` } func NewDB(db *sql.DB) *Database { return &Database{db: db} } // Database sql wrapper for operations with browser objects. type Database struct { db *sql.DB } // Close closes database. func (db Database) Close() error { return db.db.Close() } func (db *Database) CreateSettings(s Settings, nodecfg params.NodeConfig) error { _, err := db.db.Exec(` INSERT INTO settings ( address, currency, current_network, dapps_address, eip1581_address, installation_id, key_uid, keycard_instance_uid, keycard_paired_on, keycard_pairing, latest_derived_path, mnemonic, name, networks, node_config, photo_path, preview_privacy, public_key, signing_phrase, wallet_root_address, synthetic_id ) VALUES ( ?,?,?,?,?,?,?,?,?,?, ?,?,?,?,?,?,?,?,?,?, 'id')`, s.Address, s.Currency, s.CurrentNetwork, s.DappsAddress, s.EIP1581Address, s.InstallationID, s.KeyUID, s.KeycardInstanceUID, s.KeycardPAiredOn, s.KeycardPairing, s.LatestDerivedPath, s.Mnemonic, s.Name, s.Networks, &sqlite.JSONBlob{nodecfg}, s.PhotoPath, s.PreviewPrivacy, s.PublicKey, s.SigningPhrase, s.WalletRootAddress) return err } // nolint: gocyclo func (db *Database) SaveSetting(setting string, value interface{}) error { var ( update *sql.Stmt err error ) switch setting { case "chaos-mode?": update, err = db.db.Prepare("UPDATE settings SET chaos_mode = ? WHERE synthetic_id = 'id'") case "currency": update, err = db.db.Prepare("UPDATE settings SET currency = ? WHERE synthetic_id = 'id'") case "custom-bootnodes": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET custom_bootnodes = ? WHERE synthetic_id = 'id'") case "custom-bootnodes-enabled": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET custom_bootnodes_enabled = ? WHERE synthetic_id = 'id'") case "dapps-address": update, err = db.db.Prepare("UPDATE settings SET dapps_address = ? WHERE synthetic_id = 'id'") case "eip1581-address": update, err = db.db.Prepare("UPDATE settings SET eip1581_address = ? WHERE synthetic_id = 'id'") case "fleet": update, err = db.db.Prepare("UPDATE settings SET fleet = ? WHERE synthetic_id = 'id'") case "hide-home-tooltip?": update, err = db.db.Prepare("UPDATE settings SET hide_home_tooltip = ? WHERE synthetic_id = 'id'") case "keycard-instance_uid": update, err = db.db.Prepare("UPDATE settings SET keycard_instance_uid = ? WHERE synthetic_id = 'id'") case "keycard-paired_on": update, err = db.db.Prepare("UPDATE settings SET keycard_paired_on = ? WHERE synthetic_id = 'id'") case "keycard-pairing": update, err = db.db.Prepare("UPDATE settings SET keycard_pairing = ? WHERE synthetic_id = 'id'") case "last-updated": update, err = db.db.Prepare("UPDATE settings SET last_updated = ? WHERE synthetic_id = 'id'") case "latest-derived-path": update, err = db.db.Prepare("UPDATE settings SET latest_derived_path = ? WHERE synthetic_id = 'id'") case "log-level": update, err = db.db.Prepare("UPDATE settings SET log_level = ? WHERE synthetic_id = 'id'") case "mnemonic": update, err = db.db.Prepare("UPDATE settings SET mnemonic = ? WHERE synthetic_id = 'id'") case "name": update, err = db.db.Prepare("UPDATE settings SET name = ? WHERE synthetic_id = 'id'") case "networks/current-network": update, err = db.db.Prepare("UPDATE settings SET current_network = ? WHERE synthetic_id = 'id'") case "networks/networks": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET networks = ? WHERE synthetic_id = 'id'") case "node-config": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET node_config = ? WHERE synthetic_id = 'id'") case "notifications-enabled?": update, err = db.db.Prepare("UPDATE settings SET notifications_enabled = ? WHERE synthetic_id = 'id'") case "photo-path": update, err = db.db.Prepare("UPDATE settings SET photo_path = ? WHERE synthetic_id = 'id'") case "pinned-mailservers": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET pinned_mailservers = ? WHERE synthetic_id = 'id'") case "preferred-name": update, err = db.db.Prepare("UPDATE settings SET preferred_name = ? WHERE synthetic_id = 'id'") case "preview-privacy?": update, err = db.db.Prepare("UPDATE settings SET preview_privacy = ? WHERE synthetic_id = 'id'") case "public-key": update, err = db.db.Prepare("UPDATE settings SET public_key = ? WHERE synthetic_id = 'id'") case "remember-syncing-choice?": update, err = db.db.Prepare("UPDATE settings SET remember_syncing_choice = ? WHERE synthetic_id = 'id'") case "show-name?": update, err = db.db.Prepare("UPDATE settings SET show_name = ? WHERE synthetic_id = 'id'") case "stickers-packs-installed": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET stickers_packs_installed = ? WHERE synthetic_id = 'id'") case "stickers-recent-stickers": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET stickers_recent_stickers = ? WHERE synthetic_id = 'id'") case "syncing-on-mobile-network?": update, err = db.db.Prepare("UPDATE settings SET syncing_on_mobile_network = ? WHERE synthetic_id = 'id'") case "usernames": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET usernames = ? WHERE synthetic_id = 'id'") case "wallet-set-up-passed?": update, err = db.db.Prepare("UPDATE settings SET wallet_set_up_passed = ? WHERE synthetic_id = 'id'") case "wallet/visible-tokens": value = &sqlite.JSONBlob{value} update, err = db.db.Prepare("UPDATE settings SET wallet_visible_tokens = ? WHERE synthetic_id = 'id'") default: return ErrInvalidConfig } if err != nil { return err } _, err = update.Exec(value) return err } func (db *Database) GetNodeConfig(nodecfg interface{}) error { return db.db.QueryRow("SELECT node_config FROM settings WHERE synthetic_id = 'id'").Scan(&sqlite.JSONBlob{nodecfg}) } func (db *Database) GetSettings() (Settings, error) { var s Settings err := db.db.QueryRow("SELECT address, chaos_mode, currency, current_network, custom_bootnodes, custom_bootnodes_enabled, dapps_address, eip1581_address, fleet, hide_home_tooltip, installation_id, key_uid, keycard_instance_uid, keycard_paired_on, keycard_pairing, last_updated, latest_derived_path, log_level, mnemonic, name, networks, notifications_enabled, photo_path, pinned_mailservers, preferred_name, preview_privacy, public_key, remember_syncing_choice, show_name, signing_phrase, stickers_packs_installed, stickers_recent_stickers, syncing_on_mobile_network, usernames, wallet_root_address, wallet_set_up_passed, wallet_visible_tokens FROM settings WHERE synthetic_id = 'id'").Scan( &s.Address, &s.ChaosMode, &s.Currency, &s.CurrentNetwork, &s.CustomBootnodes, &s.CustomBootnodesEnabled, &s.DappsAddress, &s.EIP1581Address, &s.Fleet, &s.HideHomeTooltip, &s.InstallationID, &s.KeyUID, &s.KeycardInstanceUID, &s.KeycardPAiredOn, &s.KeycardPairing, &s.LastUpdated, &s.LatestDerivedPath, &s.LogLevel, &s.Mnemonic, &s.Name, &s.Networks, &s.NotificationsEnabled, &s.PhotoPath, &s.PinnedMailserver, &s.PreferredName, &s.PreviewPrivacy, &s.PublicKey, &s.RememberSyncingChoice, &s.ShowName, &s.SigningPhrase, &s.StickerPacksInstalled, &s.StickersRecentStickers, &s.SyncingOnMobileNetwork, &s.Usernames, &s.WalletRootAddress, &s.WalletSetupPassed, &s.WalletVisibleTokens) return s, err } func (db *Database) GetAccounts() ([]Account, error) { rows, err := db.db.Query("SELECT address, wallet, chat, type, storage, pubkey, path, name, color FROM accounts ORDER BY created_at") if err != nil { return nil, err } accounts := []Account{} pubkey := []byte{} for rows.Next() { acc := Account{} err := rows.Scan( &acc.Address, &acc.Wallet, &acc.Chat, &acc.Type, &acc.Storage, &pubkey, &acc.Path, &acc.Name, &acc.Color) if err != nil { return nil, err } if lth := len(pubkey); lth > 0 { acc.PublicKey = make(types.HexBytes, lth) copy(acc.PublicKey, pubkey) } accounts = append(accounts, acc) } return accounts, nil } func (db *Database) SaveAccounts(accounts []Account) (err error) { var ( tx *sql.Tx insert *sql.Stmt update *sql.Stmt ) tx, err = db.db.Begin() if err != nil { return } defer func() { if err == nil { err = tx.Commit() return } _ = tx.Rollback() }() // NOTE(dshulyak) replace all record values using address (primary key) // can't use `insert or replace` because of the additional constraints (wallet and chat) insert, err = tx.Prepare("INSERT OR IGNORE INTO accounts (address, created_at, updated_at) VALUES (?, datetime('now'), datetime('now'))") if err != nil { return err } update, err = tx.Prepare("UPDATE accounts SET wallet = ?, chat = ?, type = ?, storage = ?, pubkey = ?, path = ?, name = ?, color = ?, updated_at = datetime('now') WHERE address = ?") if err != nil { return err } for i := range accounts { acc := &accounts[i] _, err = insert.Exec(acc.Address) if err != nil { return } _, err = update.Exec(acc.Wallet, acc.Chat, acc.Type, acc.Storage, acc.PublicKey, acc.Path, acc.Name, acc.Color, acc.Address) if err != nil { switch err.Error() { case uniqueChatConstraint: err = ErrChatNotUnique case uniqueWalletConstraint: err = ErrWalletNotUnique } return } } return } func (db *Database) DeleteAccount(address types.Address) error { _, err := db.db.Exec("DELETE FROM accounts WHERE address = ?", address) return err } func (db *Database) GetWalletAddress() (rst types.Address, err error) { err = db.db.QueryRow("SELECT address FROM accounts WHERE wallet = 1").Scan(&rst) return } func (db *Database) GetChatAddress() (rst types.Address, err error) { err = db.db.QueryRow("SELECT address FROM accounts WHERE chat = 1").Scan(&rst) return } func (db *Database) GetAddresses() (rst []types.Address, err error) { rows, err := db.db.Query("SELECT address FROM accounts ORDER BY created_at") if err != nil { return nil, err } for rows.Next() { addr := types.Address{} err = rows.Scan(&addr) if err != nil { return nil, err } rst = append(rst, addr) } return rst, nil } // AddressExists returns true if given address is stored in database. func (db *Database) AddressExists(address types.Address) (exists bool, err error) { err = db.db.QueryRow("SELECT EXISTS (SELECT 1 FROM accounts WHERE address = ?)", address).Scan(&exists) return exists, err }