this repo has no description
tangled.org
1package db
2
3import (
4 "encoding/json"
5 "time"
6)
7
8func (d *DB) AddPublicKey(did, name, key string) error {
9 query := `insert into public_keys (did, name, key) values (?, ?, ?)`
10 _, err := d.db.Exec(query, did, name, key)
11 return err
12}
13
14func (d *DB) RemovePublicKey(did string) error {
15 query := `delete from public_keys where did = ?`
16 _, err := d.db.Exec(query, did)
17 return err
18}
19
20type PublicKey struct {
21 Did string `json:"did"`
22 Key string `json:"key"`
23 Name string `json:"name"`
24 Created time.Time
25}
26
27func (p PublicKey) MarshalJSON() ([]byte, error) {
28 type Alias PublicKey
29 return json.Marshal(&struct {
30 Created int64 `json:"created"`
31 *Alias
32 }{
33 Created: p.Created.Unix(),
34 Alias: (*Alias)(&p),
35 })
36}
37
38func (d *DB) GetAllPublicKeys() ([]PublicKey, error) {
39 var keys []PublicKey
40
41 rows, err := d.db.Query(`select key, name, did, created from public_keys`)
42 if err != nil {
43 return nil, err
44 }
45 defer rows.Close()
46
47 for rows.Next() {
48 var publicKey PublicKey
49 var createdAt *int64
50 if err := rows.Scan(&publicKey.Key, &publicKey.Name, &publicKey.Did, &createdAt); err != nil {
51 return nil, err
52 }
53 publicKey.Created = time.Unix(*createdAt, 0)
54 keys = append(keys, publicKey)
55 }
56
57 if err := rows.Err(); err != nil {
58 return nil, err
59 }
60
61 return keys, nil
62}
63
64func (d *DB) GetPublicKeys(did string) ([]PublicKey, error) {
65 var keys []PublicKey
66
67 rows, err := d.db.Query(`select did, key, name, created from public_keys where did = ?`, did)
68 if err != nil {
69 return nil, err
70 }
71 defer rows.Close()
72
73 for rows.Next() {
74 var publicKey PublicKey
75 var createdAt *int64
76 if err := rows.Scan(&publicKey.Did, &publicKey.Key, &publicKey.Name, &createdAt); err != nil {
77 return nil, err
78 }
79 publicKey.Created = time.Unix(*createdAt, 0)
80 keys = append(keys, publicKey)
81 }
82
83 if err := rows.Err(); err != nil {
84 return nil, err
85 }
86
87 return keys, nil
88}