···

       
1
       
 
       
package main

     

       
2
       
 
       


     

       
3
       
 
       
import (

     

       
0
       
 
       

     

       
4
       
 
       
	"context"

     

       
5
       
 
       
	"encoding/json"

     

       
0
       
 
       

     

       
6
       
 
       
	"fmt"

     

       
0
       
 
       

     

       
7
       
 
       


     

       
0
       
 
       

     

       
8
       
 
       
	"github.com/bluesky-social/indigo/atproto/syntax"

     

       
9
       
 
       
	"github.com/ipfs/go-cid"

     

       
10
       
 
       
	"gorm.io/gorm"

     
···

       
43
       
 
       
	syntax.NSID("app.bsky.labeler.service"): true,

     

       
44
       
 
       
}

     

       
45
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
46
       
 
       
func (hs *HandlerService) HandleUpsert(ctx context.Context, repo string, rev string, path string, rec *[]byte, cid *cid.Cid, action Action) error {

     

       
47
       
 
       
	uri, err := syntax.ParseATURI(fmt.Sprintf("at://%s/%s", repo, path))

     

       
48
       
 
       
	if err != nil {

     
···

       
54
       
 
       
		return nil

     

       
55
       
 
       
	}

     

       
56
       
 
       


     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
0
       
 
       

     

       
57
       
 
       
	switch action {

     

       
58
       
 
       
	case ActionCreate:

     

       
59
       
 
       
		if err := hs.store.Create(&Record{

     

       
60
       
 
       
			Repo:       repo,

     

       
61
       
 
       
			Collection: string(uri.Collection()),

     

       
62
       
 
       
			Key:        string(uri.RecordKey()),

     

       
63
       
-
       
			Data:       json.RawMessage(*rec),

     

       
64
       
 
       
		}).Error; err != nil {

     

       
65
       
 
       
			return err

     

       
66
       
 
       
		}

     

···

       
1
       
 
       
package main

     

       
2
       
 
       


     

       
3
       
 
       
import (

     

       
4
       
+
       
	"bytes"

     

       
5
       
 
       
	"context"

     

       
6
       
 
       
	"encoding/json"

     

       
7
       
+
       
	"errors"

     

       
8
       
 
       
	"fmt"

     

       
9
       
+
       
	"io"

     

       
10
       
 
       


     

       
11
       
+
       
	appbsky "github.com/bluesky-social/indigo/api/bsky"

     

       
12
       
 
       
	"github.com/bluesky-social/indigo/atproto/syntax"

     

       
13
       
 
       
	"github.com/ipfs/go-cid"

     

       
14
       
 
       
	"gorm.io/gorm"

     
···

       
47
       
 
       
	syntax.NSID("app.bsky.labeler.service"): true,

     

       
48
       
 
       
}

     

       
49
       
 
       


     

       
50
       
+
       
type bskycbor interface {

     

       
51
       
+
       
	UnmarshalCBOR(io.Reader) error

     

       
52
       
+
       
}

     

       
53
       
+
       


     

       
54
       
 
       
func (hs *HandlerService) HandleUpsert(ctx context.Context, repo string, rev string, path string, rec *[]byte, cid *cid.Cid, action Action) error {

     

       
55
       
 
       
	uri, err := syntax.ParseATURI(fmt.Sprintf("at://%s/%s", repo, path))

     

       
56
       
 
       
	if err != nil {

     
···

       
62
       
 
       
		return nil

     

       
63
       
 
       
	}

     

       
64
       
 
       


     

       
65
       
+
       
	var out any

     

       
66
       
+
       
	switch uri.Collection() {

     

       
67
       
+
       
	case syntax.NSID("app.bsky.actor.profile"):

     

       
68
       
+
       
		out = appbsky.ActorProfile{}

     

       
69
       
+
       
	case syntax.NSID("app.bsky.feed.generator"):

     

       
70
       
+
       
		out = appbsky.FeedGenerator{}

     

       
71
       
+
       
	case syntax.NSID("app.bsky.labeler.service"):

     

       
72
       
+
       
		out = appbsky.LabelerService{}

     

       
73
       
+
       
	}

     

       
74
       
+
       


     

       
75
       
+
       
	model, ok := out.(bskycbor)

     

       
76
       
+
       
	if !ok {

     

       
77
       
+
       
		return errors.New("collection model cannot unmarshal CBOR")

     

       
78
       
+
       
	}

     

       
79
       
+
       


     

       
80
       
+
       
	if err := model.UnmarshalCBOR(bytes.NewReader(*rec)); err != nil {

     

       
81
       
+
       
		return fmt.Errorf("error cbor unmarshaling: %w", err)

     

       
82
       
+
       
	}

     

       
83
       
+
       


     

       
84
       
+
       
	body, err := json.Marshal(model)

     

       
85
       
+
       
	if err != nil {

     

       
86
       
+
       
		return fmt.Errorf("error json marshaling: %w", err)

     

       
87
       
+
       
	}

     

       
88
       
+
       


     

       
89
       
 
       
	switch action {

     

       
90
       
 
       
	case ActionCreate:

     

       
91
       
 
       
		if err := hs.store.Create(&Record{

     

       
92
       
 
       
			Repo:       repo,

     

       
93
       
 
       
			Collection: string(uri.Collection()),

     

       
94
       
 
       
			Key:        string(uri.RecordKey()),

     

       
95
       
+
       
			Data:       json.RawMessage(body),

     

       
96
       
 
       
		}).Error; err != nil {

     

       
97
       
 
       
			return err

     

       
98
       
 
       
		}