Monorepo for Tangled forked from tangled.org/core
Something went wrong. Try again.
3.0 kB · 141 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142package db
import ( "strconv" "time"
"tangled.org/core/api/tangled")
type PublicKey struct { Did string tangled.PublicKey}
func (d *DB) AddPublicKeyFromRecord(did string, recordIface map[string]interface{}) error { record := make(map[string]string) for k, v := range recordIface { if str, ok := v.(string); ok { record[k] = str } }
pk := PublicKey{ Did: did, } pk.Key = record["key"] pk.CreatedAt = record["createdAt"]
return d.AddPublicKey(pk)}
func (d *DB) AddPublicKey(pk PublicKey) error { if pk.CreatedAt == "" { pk.CreatedAt = time.Now().Format(time.RFC3339) }
query := `insert or ignore into public_keys (did, key, created) values (?, ?, ?)` _, err := d.db.Exec(query, pk.Did, pk.Key, pk.CreatedAt) return err}
func (d *DB) RemovePublicKey(did string) error { query := `delete from public_keys where did = ?` _, err := d.db.Exec(query, did) return err}
func (pk *PublicKey) JSON() map[string]any { return map[string]any{ "did": pk.Did, "key": pk.Key, "createdAt": pk.CreatedAt, }}
func (d *DB) GetAllPublicKeys() ([]PublicKey, error) { var keys []PublicKey
rows, err := d.db.Query(`select key, did, created from public_keys`) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var publicKey PublicKey if err := rows.Scan(&publicKey.Key, &publicKey.Did, &publicKey.CreatedAt); err != nil { return nil, err } keys = append(keys, publicKey) }
if err := rows.Err(); err != nil { return nil, err }
return keys, nil}
func (d *DB) GetPublicKeys(did string) ([]PublicKey, error) { var keys []PublicKey
rows, err := d.db.Query(`select did, key, created from public_keys where did = ?`, did) if err != nil { return nil, err } defer rows.Close()
for rows.Next() { var publicKey PublicKey if err := rows.Scan(&publicKey.Did, &publicKey.Key, &publicKey.CreatedAt); err != nil { return nil, err } keys = append(keys, publicKey) }
if err := rows.Err(); err != nil { return nil, err }
return keys, nil}
func (d *DB) GetPublicKeysPaginated(limit int, cursor string) ([]PublicKey, string, error) { var keys []PublicKey
offset := 0 if cursor != "" { if o, err := strconv.Atoi(cursor); err == nil && o >= 0 { offset = o } }
query := `select key, did, created from public_keys order by created desc limit ? offset ?` rows, err := d.db.Query(query, limit+1, offset) // +1 to check if there are more results if err != nil { return nil, "", err } defer rows.Close()
for rows.Next() { var publicKey PublicKey if err := rows.Scan(&publicKey.Key, &publicKey.Did, &publicKey.CreatedAt); err != nil { return nil, "", err } keys = append(keys, publicKey) }
if err := rows.Err(); err != nil { return nil, "", err }
// check if there are more results for pagination var nextCursor string if len(keys) > limit { keys = keys[:limit] // remove the extra item nextCursor = strconv.Itoa(offset + limit) }
return keys, nextCursor, nil}