lewis.moe
Monorepo for Tangled
appview: show banner for users with read-only knots
Signed-off-by: oppiliappan <me@oppi.li>
+67
-133
appview/db/registration.go
+67
-133
appview/db/registration.go
···
1
package db
2
3
import (
4
-
"crypto/rand"
5
"database/sql"
6
-
"encoding/hex"
7
"fmt"
8
-
"log"
9
"strings"
10
"time"
11
)
···
18
ByDid string
19
Created *time.Time
20
Registered *time.Time
21
}
22
23
func (r *Registration) Status() Status {
24
-
if r.Registered != nil {
25
return Registered
26
} else {
27
return Pending
28
}
29
}
30
31
type Status uint32
32
33
const (
34
Registered Status = iota
35
Pending
36
)
37
38
-
// returns registered status, did of owner, error
39
-
func RegistrationsByDid(e Execer, did string) ([]Registration, error) {
40
var registrations []Registration
41
42
-
rows, err := e.Query(`
43
-
select id, domain, did, created, registered from registrations
44
-
where did = ?
45
-
`, did)
46
-
if err != nil {
47
-
return nil, err
48
}
49
50
-
for rows.Next() {
51
-
var createdAt *string
52
-
var registeredAt *string
53
-
var registration Registration
54
-
err = rows.Scan(®istration.Id, ®istration.Domain, ®istration.ByDid, &createdAt, ®isteredAt)
55
-
56
-
if err != nil {
57
-
log.Println(err)
58
-
} else {
59
-
createdAtTime, _ := time.Parse(time.RFC3339, *createdAt)
60
-
var registeredAtTime *time.Time
61
-
if registeredAt != nil {
62
-
x, _ := time.Parse(time.RFC3339, *registeredAt)
63
-
registeredAtTime = &x
64
-
}
65
-
66
-
registration.Created = &createdAtTime
67
-
registration.Registered = registeredAtTime
68
-
registrations = append(registrations, registration)
69
-
}
70
}
71
72
-
return registrations, nil
73
-
}
74
-
75
-
// returns registered status, did of owner, error
76
-
func RegistrationByDomain(e Execer, domain string) (*Registration, error) {
77
-
var createdAt *string
78
-
var registeredAt *string
79
-
var registration Registration
80
-
81
-
err := e.QueryRow(`
82
-
select id, domain, did, created, registered from registrations
83
-
where domain = ?
84
-
`, domain).Scan(®istration.Id, ®istration.Domain, ®istration.ByDid, &createdAt, ®isteredAt)
85
86
if err != nil {
87
-
if err == sql.ErrNoRows {
88
-
return nil, nil
89
-
} else {
90
-
return nil, err
91
-
}
92
}
93
94
-
createdAtTime, _ := time.Parse(time.RFC3339, *createdAt)
95
-
var registeredAtTime *time.Time
96
-
if registeredAt != nil {
97
-
x, _ := time.Parse(time.RFC3339, *registeredAt)
98
-
registeredAtTime = &x
99
-
}
100
-
101
-
registration.Created = &createdAtTime
102
-
registration.Registered = registeredAtTime
103
-
104
-
return ®istration, nil
105
-
}
106
-
107
-
func genSecret() string {
108
-
key := make([]byte, 32)
109
-
rand.Read(key)
110
-
return hex.EncodeToString(key)
111
-
}
112
113
-
func GenerateRegistrationKey(e Execer, domain, did string) (string, error) {
114
-
// sanity check: does this domain already have a registration?
115
-
reg, err := RegistrationByDomain(e, domain)
116
-
if err != nil {
117
-
return "", err
118
-
}
119
-
120
-
// registration is open
121
-
if reg != nil {
122
-
switch reg.Status() {
123
-
case Registered:
124
-
// already registered by `owner`
125
-
return "", fmt.Errorf("%s already registered by %s", domain, reg.ByDid)
126
-
case Pending:
127
-
// TODO: be loud about this
128
-
log.Printf("%s registered by %s, status pending", domain, reg.ByDid)
129
}
130
-
}
131
132
-
secret := genSecret()
133
134
-
_, err = e.Exec(`
135
-
insert into registrations (domain, did, secret)
136
-
values (?, ?, ?)
137
-
on conflict(domain) do update set did = excluded.did, secret = excluded.secret, created = excluded.created
138
-
`, domain, did, secret)
139
-
140
-
if err != nil {
141
-
return "", err
142
-
}
143
-
144
-
return secret, nil
145
-
}
146
147
-
func GetRegistrationKey(e Execer, domain string) (string, error) {
148
-
res := e.QueryRow(`select secret from registrations where domain = ?`, domain)
149
150
-
var secret string
151
-
err := res.Scan(&secret)
152
-
if err != nil || secret == "" {
153
-
return "", err
154
}
155
156
-
return secret, nil
157
}
158
159
-
func GetCompletedRegistrations(e Execer) ([]string, error) {
160
-
rows, err := e.Query(`select domain from registrations where registered not null`)
161
-
if err != nil {
162
-
return nil, err
163
-
}
164
-
165
-
var domains []string
166
-
for rows.Next() {
167
-
var domain string
168
-
err = rows.Scan(&domain)
169
-
170
-
if err != nil {
171
-
log.Println(err)
172
-
} else {
173
-
domains = append(domains, domain)
174
-
}
175
}
176
177
-
if err = rows.Err(); err != nil {
178
-
return nil, err
179
}
180
181
-
return domains, nil
182
-
}
183
-
184
-
func Register(e Execer, domain string) error {
185
-
_, err := e.Exec(`
186
-
update registrations
187
-
set registered = strftime('%Y-%m-%dT%H:%M:%SZ', 'now')
188
-
where domain = ?;
189
-
`, domain)
190
-
191
return err
192
}
193
···
1
package db
2
3
import (
4
"database/sql"
5
"fmt"
6
"strings"
7
"time"
8
)
···
15
ByDid string
16
Created *time.Time
17
Registered *time.Time
18
+
ReadOnly bool
19
}
20
21
func (r *Registration) Status() Status {
22
+
if r.ReadOnly {
23
+
return ReadOnly
24
+
} else if r.Registered != nil {
25
return Registered
26
} else {
27
return Pending
28
}
29
}
30
31
+
func (r *Registration) IsRegistered() bool {
32
+
return r.Status() == Registered
33
+
}
34
+
35
+
func (r *Registration) IsReadOnly() bool {
36
+
return r.Status() == ReadOnly
37
+
}
38
+
39
+
func (r *Registration) IsPending() bool {
40
+
return r.Status() == Pending
41
+
}
42
+
43
type Status uint32
44
45
const (
46
Registered Status = iota
47
Pending
48
+
ReadOnly
49
)
50
51
+
func GetRegistrations(e Execer, filters ...filter) ([]Registration, error) {
52
var registrations []Registration
53
54
+
var conditions []string
55
+
var args []any
56
+
for _, filter := range filters {
57
+
conditions = append(conditions, filter.Condition())
58
+
args = append(args, filter.Arg()...)
59
}
60
61
+
whereClause := ""
62
+
if conditions != nil {
63
+
whereClause = " where " + strings.Join(conditions, " and ")
64
}
65
66
+
query := fmt.Sprintf(`
67
+
select id, domain, did, created, registered, read_only
68
+
from registrations
69
+
%s
70
+
order by created
71
+
`,
72
+
whereClause,
73
+
)
74
75
+
rows, err := e.Query(query, args...)
76
if err != nil {
77
+
return nil, err
78
}
79
80
+
for rows.Next() {
81
+
var createdAt string
82
+
var registeredAt sql.Null[string]
83
+
var readOnly int
84
+
var reg Registration
85
86
+
err = rows.Scan(®.Id, ®.Domain, ®.ByDid, &createdAt, ®isteredAt, &readOnly)
87
+
if err != nil {
88
+
return nil, err
89
}
90
91
+
if t, err := time.Parse(time.RFC3339, createdAt); err == nil {
92
+
reg.Created = &t
93
+
}
94
95
+
if registeredAt.Valid {
96
+
if t, err := time.Parse(time.RFC3339, registeredAt.V); err == nil {
97
+
reg.Registered = &t
98
+
}
99
+
}
100
101
+
if readOnly != 0 {
102
+
reg.ReadOnly = true
103
+
}
104
105
+
registrations = append(registrations, reg)
106
}
107
108
+
return registrations, nil
109
}
110
111
+
func MarkRegistered(e Execer, filters ...filter) error {
112
+
var conditions []string
113
+
var args []any
114
+
for _, filter := range filters {
115
+
conditions = append(conditions, filter.Condition())
116
+
args = append(args, filter.Arg()...)
117
}
118
119
+
query := "update registrations set registered = strftime('%Y-%m-%dT%H:%M:%SZ', 'now'), read_only = 0"
120
+
if len(conditions) > 0 {
121
+
query += " where " + strings.Join(conditions, " and ")
122
}
123
124
+
_, err := e.Exec(query, args...)
125
return err
126
}
127
+40
-16
appview/knots/knots.go
+40
-16
appview/knots/knots.go
···
3
import (
4
"errors"
5
"fmt"
6
"log/slog"
7
"net/http"
8
"slices"
···
49
r.With(middleware.AuthMiddleware(k.OAuth)).Post("/{domain}/add", k.addMember)
50
r.With(middleware.AuthMiddleware(k.OAuth)).Post("/{domain}/remove", k.removeMember)
51
52
return r
53
}
54
55
func (k *Knots) knots(w http.ResponseWriter, r *http.Request) {
56
user := k.OAuth.GetUser(r)
57
-
registrations, err := db.RegistrationsByDid(k.Db, user.Did)
58
if err != nil {
59
k.Logger.Error("failed to fetch knot registrations", "err", err)
60
w.WriteHeader(http.StatusInternalServerError)
···
89
http.Error(w, "Not found", http.StatusNotFound)
90
return
91
}
92
-
93
-
// Find the specific registration for this domain
94
-
var registration *db.Registration
95
-
for _, reg := range registrations {
96
-
if reg.Domain == domain && reg.ByDid == user.Did && reg.Registered != nil {
97
-
registration = ®
98
-
break
99
-
}
100
-
}
101
-
102
-
if registration == nil {
103
-
l.Error("registration not found or not verified")
104
-
http.Error(w, "Not found", http.StatusNotFound)
105
return
106
}
107
registration := registrations[0]
···
518
db.FilterIsNot("registered", "null"),
519
)
520
if err != nil {
521
-
l.Error("failed to retrieve domain registration", "err", err)
522
-
http.Error(w, "Not found", http.StatusNotFound)
523
return
524
}
525
526
noticeId := fmt.Sprintf("add-member-error-%d", registration.Id)
527
defaultErr := "Failed to add member. Try again later."
···
679
// ok
680
k.Pages.HxRefresh(w)
681
}
···
3
import (
4
"errors"
5
"fmt"
6
+
"log"
7
"log/slog"
8
"net/http"
9
"slices"
···
50
r.With(middleware.AuthMiddleware(k.OAuth)).Post("/{domain}/add", k.addMember)
51
r.With(middleware.AuthMiddleware(k.OAuth)).Post("/{domain}/remove", k.removeMember)
52
53
+
r.With(middleware.AuthMiddleware(k.OAuth)).Get("/upgradeBanner", k.banner)
54
+
55
return r
56
}
57
58
func (k *Knots) knots(w http.ResponseWriter, r *http.Request) {
59
user := k.OAuth.GetUser(r)
60
+
registrations, err := db.GetRegistrations(
61
+
k.Db,
62
+
db.FilterEq("did", user.Did),
63
+
)
64
if err != nil {
65
k.Logger.Error("failed to fetch knot registrations", "err", err)
66
w.WriteHeader(http.StatusInternalServerError)
···
95
http.Error(w, "Not found", http.StatusNotFound)
96
return
97
}
98
+
if len(registrations) != 1 {
99
+
l.Error("got incorret number of registrations", "got", len(registrations), "expected", 1)
100
return
101
}
102
registration := registrations[0]
···
513
db.FilterIsNot("registered", "null"),
514
)
515
if err != nil {
516
+
l.Error("failed to get registration", "err", err)
517
return
518
}
519
+
if len(registrations) != 1 {
520
+
l.Error("got incorret number of registrations", "got", len(registrations), "expected", 1)
521
+
return
522
+
}
523
+
registration := registrations[0]
524
525
noticeId := fmt.Sprintf("add-member-error-%d", registration.Id)
526
defaultErr := "Failed to add member. Try again later."
···
678
// ok
679
k.Pages.HxRefresh(w)
680
}
681
+
682
+
func (k *Knots) banner(w http.ResponseWriter, r *http.Request) {
683
+
user := k.OAuth.GetUser(r)
684
+
l := k.Logger.With("handler", "removeMember")
685
+
l = l.With("did", user.Did)
686
+
l = l.With("handle", user.Handle)
687
+
688
+
registrations, err := db.GetRegistrations(
689
+
k.Db,
690
+
db.FilterEq("did", user.Did),
691
+
db.FilterEq("read_only", 1),
692
+
)
693
+
if err != nil {
694
+
l.Error("non-fatal: failed to get registrations")
695
+
return
696
+
}
697
+
698
+
if registrations == nil {
699
+
return
700
+
}
701
+
702
+
k.Pages.KnotBanner(w, pages.KnotBannerParams{
703
+
Registrations: registrations,
704
+
})
705
+
}
+9
-1
appview/pages/pages.go
+9
-1
appview/pages/pages.go
···
338
return p.execute("user/settings/emails", w, params)
339
}
340
341
type KnotsParams struct {
342
LoggedInUser *oauth.User
343
Registrations []db.Registration
···
360
}
361
362
type KnotListingParams struct {
363
-
db.Registration
364
}
365
366
func (p *Pages) KnotListing(w io.Writer, params KnotListingParams) error {
···
338
return p.execute("user/settings/emails", w, params)
339
}
340
341
+
type KnotBannerParams struct {
342
+
Registrations []db.Registration
343
+
}
344
+
345
+
func (p *Pages) KnotBanner(w io.Writer, params KnotBannerParams) error {
346
+
return p.executePlain("knots/fragments/banner", w, params)
347
+
}
348
+
349
type KnotsParams struct {
350
LoggedInUser *oauth.User
351
Registrations []db.Registration
···
368
}
369
370
type KnotListingParams struct {
371
+
*db.Registration
372
}
373
374
func (p *Pages) KnotListing(w io.Writer, params KnotListingParams) error {
+11
-2
appview/pages/templates/knots/fragments/knotListing.html
+11
-2
appview/pages/templates/knots/fragments/knotListing.html
···
30
{{ define "knotRightSide" }}
31
<div id="right-side" class="flex gap-2">
32
{{ $style := "px-2 py-1 rounded flex items-center flex-shrink-0 gap-2 text-sm" }}
33
-
{{ if .Registered }}
34
-
<span class="bg-green-100 text-green-800 dark:bg-green-900 dark:text-green-200 {{$style}}">{{ i "shield-check" "w-4 h-4" }} verified</span>
35
{{ template "knots/fragments/addMemberModal" . }}
36
{{ else }}
37
<span class="bg-red-100 text-red-800 dark:bg-red-900 dark:text-red-200 {{$style}}">
38
{{ i "shield-off" "w-4 h-4" }} unverified
···
30
{{ define "knotRightSide" }}
31
<div id="right-side" class="flex gap-2">
32
{{ $style := "px-2 py-1 rounded flex items-center flex-shrink-0 gap-2 text-sm" }}
33
+
{{ if .IsRegistered }}
34
+
<span class="bg-green-100 text-green-800 dark:bg-green-900 dark:text-green-200 {{$style}}">
35
+
{{ i "shield-check" "w-4 h-4" }} verified
36
+
</span>
37
{{ template "knots/fragments/addMemberModal" . }}
38
+
{{ block "knotDeleteButton" . }} {{ end }}
39
+
{{ else if .IsReadOnly }}
40
+
<span class="bg-yellow-100 text-yellow-800 dark:bg-yellow-900 dark:text-yellow-200 {{$style}}">
41
+
{{ i "shield-alert" "w-4 h-4" }} read-only
42
+
</span>
43
+
{{ block "knotRetryButton" . }} {{ end }}
44
+
{{ block "knotDeleteButton" . }} {{ end }}
45
{{ else }}
46
<span class="bg-red-100 text-red-800 dark:bg-red-900 dark:text-red-200 {{$style}}">
47
{{ i "shield-off" "w-4 h-4" }} unverified
+1
-1
appview/pages/templates/knots/index.html
+1
-1
appview/pages/templates/knots/index.html
+7
appview/pages/templates/layouts/topbar.html
+7
appview/pages/templates/layouts/topbar.html
+3
-1
appview/pages/templates/spindles/fragments/spindleListing.html
+3
-1
appview/pages/templates/spindles/fragments/spindleListing.html
···
9
{{ if .Verified }}
10
<a href="/spindles/{{ .Instance }}" class="hover:no-underline flex items-center gap-2 min-w-0 max-w-[60%]">
11
{{ i "hard-drive" "w-4 h-4" }}
12
+
<span class="hover:underline">
13
+
{{ .Instance }}
14
+
</span>
15
<span class="text-gray-500">
16
{{ template "repo/fragments/shortTimeAgo" .Created }}
17
</span>
+5
-2
appview/state/knotstream.go
+5
-2
appview/state/knotstream.go
···
24
)
25
26
func Knotstream(ctx context.Context, c *config.Config, d *db.DB, enforcer *rbac.Enforcer, posthog posthog.Client) (*ec.Consumer, error) {
27
-
knots, err := db.GetCompletedRegistrations(d)
28
if err != nil {
29
return nil, err
30
}
31
32
srcs := make(map[ec.Source]struct{})
33
for _, k := range knots {
34
-
s := ec.NewKnotSource(k)
35
srcs[s] = struct{}{}
36
}
37
···
24
)
25
26
func Knotstream(ctx context.Context, c *config.Config, d *db.DB, enforcer *rbac.Enforcer, posthog posthog.Client) (*ec.Consumer, error) {
27
+
knots, err := db.GetRegistrations(
28
+
d,
29
+
db.FilterIsNot("registered", "null"),
30
+
)
31
if err != nil {
32
return nil, err
33
}
34
35
srcs := make(map[ec.Source]struct{})
36
for _, k := range knots {
37
+
s := ec.NewKnotSource(k.Domain)
38
srcs[s] = struct{}{}
39
}
40
+3
-3
appview/state/state.go
+3
-3
appview/state/state.go
+25
appview/xrpcclient/xrpc.go
+25
appview/xrpcclient/xrpc.go
···
3
import (
4
"bytes"
5
"context"
6
+
"errors"
7
+
"fmt"
8
"io"
9
+
"net/http"
10
11
"github.com/bluesky-social/indigo/api/atproto"
12
"github.com/bluesky-social/indigo/xrpc"
13
+
indigoxrpc "github.com/bluesky-social/indigo/xrpc"
14
oauth "tangled.sh/icyphox.sh/atproto-oauth"
15
)
16
···
106
107
return &out, nil
108
}
109
+
110
+
// produces a more manageable error
111
+
func HandleXrpcErr(err error) error {
112
+
if err == nil {
113
+
return nil
114
+
}
115
+
116
+
var xrpcerr *indigoxrpc.Error
117
+
if ok := errors.As(err, &xrpcerr); !ok {
118
+
return fmt.Errorf("Recieved invalid XRPC error response.")
119
+
}
120
+
121
+
switch xrpcerr.StatusCode {
122
+
case http.StatusNotFound:
123
+
return fmt.Errorf("XRPC is unsupported on this knot, consider upgrading your knot.")
124
+
case http.StatusUnauthorized:
125
+
return fmt.Errorf("Unauthorized XRPC request.")
126
+
default:
127
+
return fmt.Errorf("Failed to perform operation. Try again later.")
128
+
}
129
+
}
+3
-3
knotserver/ingester.go
+3
-3
knotserver/ingester.go
···
73
}
74
l.Info("added member from firehose", "member", record.Subject)
75
76
-
if err := h.db.AddDid(did); err != nil {
77
l.Error("failed to add did", "error", err)
78
return fmt.Errorf("failed to add did: %w", err)
79
}
80
-
h.jc.AddDid(did)
81
82
-
if err := h.fetchAndAddKeys(ctx, did); err != nil {
83
return fmt.Errorf("failed to fetch and add keys: %w", err)
84
}
85
···
73
}
74
l.Info("added member from firehose", "member", record.Subject)
75
76
+
if err := h.db.AddDid(record.Subject); err != nil {
77
l.Error("failed to add did", "error", err)
78
return fmt.Errorf("failed to add did: %w", err)
79
}
80
+
h.jc.AddDid(record.Subject)
81
82
+
if err := h.fetchAndAddKeys(ctx, record.Subject); err != nil {
83
return fmt.Errorf("failed to fetch and add keys: %w", err)
84
}
85