56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package wallet
|
|
|
|
import (
|
|
"database/sql"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
)
|
|
|
|
type SavedAddress struct {
|
|
Address common.Address `json:"address"`
|
|
// TODO: Add Emoji and Networks
|
|
// Emoji string `json:"emoji"`
|
|
Name string `json:"name"`
|
|
Favourite bool `json:"favourite"`
|
|
ChainID uint64 `json:"chainId"`
|
|
}
|
|
|
|
type SavedAddressesManager struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func (sam *SavedAddressesManager) GetSavedAddresses(chainID uint64) ([]SavedAddress, error) {
|
|
rows, err := sam.db.Query("SELECT address, name, favourite, network_id FROM saved_addresses WHERE network_id = ?", chainID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var rst []SavedAddress
|
|
for rows.Next() {
|
|
sa := SavedAddress{}
|
|
err := rows.Scan(&sa.Address, &sa.Name, &sa.Favourite, &sa.ChainID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rst = append(rst, sa)
|
|
}
|
|
|
|
return rst, nil
|
|
}
|
|
|
|
func (sam *SavedAddressesManager) AddSavedAddress(sa SavedAddress) error {
|
|
insert, err := sam.db.Prepare("INSERT OR REPLACE INTO saved_addresses (network_id, address, name, favourite) VALUES (?, ?, ?, ?)")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = insert.Exec(sa.ChainID, sa.Address, sa.Name, sa.Favourite)
|
|
return err
|
|
}
|
|
|
|
func (sam *SavedAddressesManager) DeleteSavedAddress(chainID uint64, address common.Address) error {
|
|
_, err := sam.db.Exec(`DELETE FROM saved_addresses WHERE address = ? AND network_id = ?`, address, chainID)
|
|
return err
|
|
}
|