commit 1020aee9e430b20d7f39dfbdf7c02fe8499cf9d5 · edavis.dev/monarch

cmd/backfiller/backfill.go cmd/monarch/backfill.go

cmd/backfiller/census.go cmd/monarch/census.go

cmd/backfiller/cursors.go cmd/monarch/cursors.go

cmd/backfiller/database.go cmd/monarch/database.go

cmd/backfiller/firehose.go cmd/monarch/firehose.go

-58

cmd/backfiller/handlers.go

···

       1
       -
       package main

     

       2
       -
       

     

       3
       -
       import (

     

       4
       -
       	"context"

     

       5
       -
       	"fmt"

     

       6
       -
       

     

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

     

       8
       -
       

     

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

     

       10
       -
       	"gorm.io/gorm"

     

       11
       -
       )

     

       12
       -
       

     

       13
       -
       type HandlerService struct {

     

       14
       -
       	store *gorm.DB

     

       15
       -
       }

     

       16
       -
       

     

       17
       -
       func NewHandlerService(store *gorm.DB) *HandlerService {

     

       18
       -
       	store.AutoMigrate(&Account{})

     

       19
       -
       	store.AutoMigrate(&Profile{})

     

       20
       -
       	store.AutoMigrate(&List{})

     

       21
       -
       	store.AutoMigrate(&Labeler{})

     

       22
       -
       	store.AutoMigrate(&FeedGenerator{})

     

       23
       -
       	store.AutoMigrate(&Post{})

     

       24
       -
       	store.AutoMigrate(&Like{})

     

       25
       -
       	store.AutoMigrate(&StarterPack{})

     

       26
       -
       	store.AutoMigrate(&Verification{})

     

       27
       -
       	store.AutoMigrate(&Lexicon{})

     

       28
       -
       

     

       29
       -
       	return &HandlerService{

     

       30
       -
       		store: store,

     

       31
       -
       	}

     

       32
       -
       }

     

       33
       -
       

     

       34
       -
       // handles both creates and updates

     

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

     

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

     

       37
       -
       	if err != nil {

     

       38
       -
       		return err

     

       39
       -
       	}

     

       40
       -
       

     

       41
       -
       	switch uri.Collection() {

     

       42
       -
       	case syntax.NSID("app.bsky.actor.profile"):

     

       43
       -
       		profile := NewProfile(*rec)

     

       44
       -
       		profile.AtUri = string(uri)

     

       45
       -
       		hs.store.Create(profile)

     

       46
       -
       

     

       47
       -
       	case syntax.NSID("app.bsky.feed.generator"):

     

       48
       -
       		feedgen := NewFeedGenerator(*rec)

     

       49
       -
       		feedgen.AtUri = string(uri)

     

       50
       -
       		hs.store.Create(feedgen)

     

       51
       -
       	}

     

       52
       -
       

     

       53
       -
       	return nil

     

       54
       -
       }

     

       55
       -
       

     

       56
       -
       func (hs *HandlerService) HandleDelete(ctx context.Context, repo string, rev string, path string) error {

     

       57
       -
       	return nil

     

       58
       -
       }

cmd/backfiller/main.go cmd/monarch/main.go

-80

cmd/backfiller/models.go

···

       1
       -
       package main

     

       2
       -
       

     

       3
       -
       import (

     

       4
       -
       	"bytes"

     

       5
       -
       	"log/slog"

     

       6
       -
       

     

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

     

       8
       -
       )

     

       9
       -
       

     

       10
       -
       type Account struct{}

     

       11
       -
       

     

       12
       -
       // func NewAccount

     

       13
       -
       

     

       14
       -
       type Profile struct {

     

       15
       -
       	ID          uint `gorm:"primaryKey"`

     

       16
       -
       	AtUri       string

     

       17
       -
       	DisplayName *string

     

       18
       -
       }

     

       19
       -
       

     

       20
       -
       func NewProfile(rec []byte) *Profile {

     

       21
       -
       	var out appbsky.ActorProfile

     

       22
       -
       	if err := out.UnmarshalCBOR(bytes.NewReader(rec)); err != nil {

     

       23
       -
       		slog.Error("could not unmarshal profile CBOR", "err", err)

     

       24
       -
       		return nil

     

       25
       -
       	}

     

       26
       -
       	return &Profile{

     

       27
       -
       		DisplayName: out.DisplayName,

     

       28
       -
       	}

     

       29
       -
       }

     

       30
       -
       

     

       31
       -
       type List struct{}

     

       32
       -
       

     

       33
       -
       // func NewList

     

       34
       -
       

     

       35
       -
       type Labeler struct{}

     

       36
       -
       

     

       37
       -
       // func NewFeedGenerator

     

       38
       -
       

     

       39
       -
       type FeedGenerator struct {

     

       40
       -
       	ID          uint `gorm:"primaryKey"`

     

       41
       -
       	AtUri       string

     

       42
       -
       	DisplayName string

     

       43
       -
       }

     

       44
       -
       

     

       45
       -
       func NewFeedGenerator(rec []byte) *FeedGenerator {

     

       46
       -
       	var out appbsky.FeedGenerator

     

       47
       -
       	if err := out.UnmarshalCBOR(bytes.NewReader(rec)); err != nil {

     

       48
       -
       		slog.Error("could not unmarshal feedgen CBOR", "err", err)

     

       49
       -
       		return nil

     

       50
       -
       	}

     

       51
       -
       	return &FeedGenerator{

     

       52
       -
       		DisplayName: out.DisplayName,

     

       53
       -
       	}

     

       54
       -
       }

     

       55
       -
       

     

       56
       -
       type Post struct {

     

       57
       -
       	ID    uint `gorm:"primaryKey"`

     

       58
       -
       	AtUri string

     

       59
       -
       	Text  string

     

       60
       -
       }

     

       61
       -
       

     

       62
       -
       func NewPost(rec []byte) *Post {

     

       63
       -
       	return nil

     

       64
       -
       }

     

       65
       -
       

     

       66
       -
       type Like struct{}

     

       67
       -
       

     

       68
       -
       // func NewLike

     

       69
       -
       

     

       70
       -
       type StarterPack struct{}

     

       71
       -
       

     

       72
       -
       // func NewStarterPack

     

       73
       -
       

     

       74
       -
       type Verification struct{}

     

       75
       -
       

     

       76
       -
       // func NewVerification

     

       77
       -
       

     

       78
       -
       type Lexicon struct{}

     

       79
       -
       

     

       80
       -
       // func NewLexicon

+45

cmd/monarch/handlers.go

···

       1
       +
       package main

     

       2
       +
       

     

       3
       +
       import (

     

       4
       +
       	"context"

     

       5
       +
       	"fmt"

     

       6
       +
       

     

       7
       +
       	"github.com/bluesky-social/indigo/atproto/syntax"

     

       8
       +
       	"github.com/ipfs/go-cid"

     

       9
       +
       	"gorm.io/gorm"

     

       10
       +
       	"tangled.sh/edavis.dev/monarch/models"

     

       11
       +
       )

     

       12
       +
       

     

       13
       +
       type HandlerService struct {

     

       14
       +
       	store *gorm.DB

     

       15
       +
       }

     

       16
       +
       

     

       17
       +
       func NewHandlerService(store *gorm.DB) *HandlerService {

     

       18
       +
       	store.AutoMigrate(&models.ActorProfile{})

     

       19
       +
       

     

       20
       +
       	return &HandlerService{

     

       21
       +
       		store: store,

     

       22
       +
       	}

     

       23
       +
       }

     

       24
       +
       

     

       25
       +
       // handles both creates and updates

     

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

     

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

     

       28
       +
       	if err != nil {

     

       29
       +
       		return fmt.Errorf("error parsing at-uri: %w", err)

     

       30
       +
       	}

     

       31
       +
       

     

       32
       +
       	switch uri.Collection() {

     

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

     

       34
       +
       		profile := models.NewActorProfile(uri, *rec)

     

       35
       +
       		if err := hs.store.Save(profile).Error; err != nil {

     

       36
       +
       			return fmt.Errorf("error saving profile: %w", err)

     

       37
       +
       		}

     

       38
       +
       	}

     

       39
       +
       

     

       40
       +
       	return nil

     

       41
       +
       }

     

       42
       +
       

     

       43
       +
       func (hs *HandlerService) HandleDelete(ctx context.Context, repo string, rev string, path string) error {

     

       44
       +
       	return nil

     

       45
       +
       }

+1 -1

go.mod

···

       1
       -
       module github.com/edavis/backfiller

     

       2
        
       

     

       3
        
       go 1.24

     

       4

···

       1
       +
       module tangled.sh/edavis.dev/monarch

     

       2
        
       

     

       3
        
       go 1.24

     

       4

+73

models/models.go

···

       1
       +
       package models

     

       2
       +
       

     

       3
       +
       import (

     

       4
       +
       	"bytes"

     

       5
       +
       	"log/slog"

     

       6
       +
       	"time"

     

       7
       +
       

     

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

     

       9
       +
       	"github.com/bluesky-social/indigo/atproto/syntax"

     

       10
       +
       )

     

       11
       +
       

     

       12
       +
       type StrongRef struct {

     

       13
       +
       	Uri string

     

       14
       +
       	Cid string

     

       15
       +
       }

     

       16
       +
       

     

       17
       +
       type ActorProfile struct {

     

       18
       +
       	ID string `gorm:"primaryKey"`

     

       19
       +
       

     

       20
       +
       	// Avatar

     

       21
       +
       	// Banner

     

       22
       +
       	CreatedAt            *string

     

       23
       +
       	Description          *string

     

       24
       +
       	DisplayName          *string

     

       25
       +
       	JoinedViaStarterPack StrongRef `gorm:"embedded;embeddedPrefix:joinedviastarterpack_"`

     

       26
       +
       	// Labels

     

       27
       +
       	// PinnedPost StrongRef

     

       28
       +
       

     

       29
       +
       	AutoCreatedAt time.Time `gorm:"autoCreateTime"`

     

       30
       +
       	AutoUpdatedAt time.Time `gorm:"autoUpdateTime"`

     

       31
       +
       }

     

       32
       +
       

     

       33
       +
       func NewActorProfile(uri syntax.ATURI, rec []byte) *ActorProfile {

     

       34
       +
       	var out appbsky.ActorProfile

     

       35
       +
       	if err := out.UnmarshalCBOR(bytes.NewReader(rec)); err != nil {

     

       36
       +
       		slog.Error("could not unmarshal profile CBOR", "err", err)

     

       37
       +
       		return nil

     

       38
       +
       	}

     

       39
       +
       

     

       40
       +
       	profile := ActorProfile{

     

       41
       +
       		ID:          string(uri),

     

       42
       +
       		CreatedAt:   out.CreatedAt,

     

       43
       +
       		Description: out.Description,

     

       44
       +
       		DisplayName: out.DisplayName,

     

       45
       +
       	}

     

       46
       +
       

     

       47
       +
       	if out.JoinedViaStarterPack != nil {

     

       48
       +
       		profile.JoinedViaStarterPack = StrongRef{

     

       49
       +
       			Uri: out.JoinedViaStarterPack.Uri,

     

       50
       +
       			Cid: out.JoinedViaStarterPack.Cid,

     

       51
       +
       		}

     

       52
       +
       	}

     

       53
       +
       

     

       54
       +
       	return &profile

     

       55
       +
       }

     

       56
       +
       

     

       57
       +
       // ActorStatus

     

       58
       +
       // FeedGenerator

     

       59
       +
       // FeedLike

     

       60
       +
       // FeedPost

     

       61
       +
       // FeedPostgate

     

       62
       +
       // FeedRepost

     

       63
       +
       // FeedThreadgate

     

       64
       +
       // GraphBlock

     

       65
       +
       // GraphFollow

     

       66
       +
       // GraphList

     

       67
       +
       // GraphListblock

     

       68
       +
       // GraphListitem

     

       69
       +
       // GraphStarterpack

     

       70
       +
       // GraphVerification

     

       71
       +
       // LabelerService

     

       72
       +
       // ActorDeclaration

     

       73
       +
       // LexiconSchema