commit 557eea024a3bd507a3120bda8e894af459b090c1 · edavis.dev/monarch

+33 -1
cmd/monarch/handlers.go
···

       1
       1
        
       package main

     

       2
       2
        
       

     

       3
       3
        
       import (

     

       4
       4
       +
       	"bytes"

     

       4
       5
        
       	"context"

     

       5
       6
        
       	"encoding/json"

     

       7
       7
       +
       	"errors"

     

       6
       8
        
       	"fmt"

     

       9
       9
       +
       	"io"

     

       7
       10
        
       

     

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

     

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

     

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

     

       10
       14
        
       	"gorm.io/gorm"

     
···

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

     

       44
       48
        
       }

     

       45
       49
        
       

     

       50
       50
       +
       type bskycbor interface {

     

       51
       51
       +
       	UnmarshalCBOR(io.Reader) error

     

       52
       52
       +
       }

     

       53
       53
       +
       

     

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

     

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

     

       48
       56
        
       	if err != nil {

     
···

       54
       62
        
       		return nil

     

       55
       63
        
       	}

     

       56
       64
        
       

     

       65
       65
       +
       	var out any

     

       66
       66
       +
       	switch uri.Collection() {

     

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

     

       68
       68
       +
       		out = appbsky.ActorProfile{}

     

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

     

       70
       70
       +
       		out = appbsky.FeedGenerator{}

     

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

     

       72
       72
       +
       		out = appbsky.LabelerService{}

     

       73
       73
       +
       	}

     

       74
       74
       +
       

     

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

     

       76
       76
       +
       	if !ok {

     

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

     

       78
       78
       +
       	}

     

       79
       79
       +
       

     

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

     

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

     

       82
       82
       +
       	}

     

       83
       83
       +
       

     

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

     

       85
       85
       +
       	if err != nil {

     

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

     

       87
       87
       +
       	}

     

       88
       88
       +
       

     

       57
       89
        
       	switch action {

     

       58
       90
        
       	case ActionCreate:

     

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

     

       60
       92
        
       			Repo:       repo,

     

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

     

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

     

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

     

       95
       95
       +
       			Data:       json.RawMessage(body),

     

       64
       96
        
       		}).Error; err != nil {

     

       65
       97
        
       			return err

     

       66
       98
        
       		}