chore: init

internal/tasks/federate_follow.go

@@ -0,0 +1,11 @@
+package tasks
+
+import "context"
+
+type FederateFollowData struct {
+	FollowID string `json:"followID"`
+}
+
+func (t *Handler) FederateFollow(ctx context.Context, data FederateNoteData) error {
+	return nil
+}

internal/tasks/federate_note.go

@@ -0,0 +1,48 @@
+package tasks
+
+import (
+	"context"
+	"github.com/lysand-org/versia-go/internal/repository"
+
+	"github.com/google/uuid"
+	"github.com/lysand-org/versia-go/internal/entity"
+)
+
+type FederateNoteData struct {
+	NoteID uuid.UUID `json:"noteID"`
+}
+
+func (t *Handler) FederateNote(ctx context.Context, data FederateNoteData) error {
+	s := t.telemetry.StartSpan(ctx, "function", "tasks/Handler.FederateNote")
+	defer s.End()
+	ctx = s.Context()
+
+	var n *entity.Note
+	if err := t.repositories.Atomic(ctx, func(ctx context.Context, tx repository.Manager) error {
+		var err error
+		n, err = tx.Notes().GetByID(ctx, data.NoteID)
+		if err != nil {
+			return err
+		}
+
+		for _, uu := range n.Mentions {
+			if !uu.IsRemote {
+				t.log.V(2).Info("User is not remote", "user", uu.ID)
+				continue
+			}
+
+			res, err := t.federationService.SendToInbox(ctx, n.Author, &uu, n.ToLysand())
+			if err != nil {
+				t.log.Error(err, "Failed to send note to remote user", "res", res, "user", uu.ID)
+			} else {
+				t.log.V(2).Info("Sent note to remote user", "res", res, "user", uu.ID)
+			}
+		}
+
+		return nil
+	}); err != nil {
+		return err
+	}
+
+	return nil
+}

internal/tasks/handler.go

@@ -0,0 +1,53 @@
+package tasks
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/lysand-org/versia-go/internal/repository"
+	"github.com/lysand-org/versia-go/internal/service"
+
+	"git.devminer.xyz/devminer/unitel"
+	"github.com/go-logr/logr"
+	"github.com/lysand-org/versia-go/pkg/taskqueue"
+)
+
+const (
+	FederateNote   = "federate_note"
+	FederateFollow = "federate_follow"
+)
+
+type Handler struct {
+	federationService service.FederationService
+
+	repositories repository.Manager
+
+	telemetry *unitel.Telemetry
+	log       logr.Logger
+}
+
+func NewHandler(federationService service.FederationService, repositories repository.Manager, telemetry *unitel.Telemetry, log logr.Logger) *Handler {
+	return &Handler{
+		federationService: federationService,
+
+		repositories: repositories,
+
+		telemetry: telemetry,
+		log:       log,
+	}
+}
+
+func (t *Handler) Register(tq *taskqueue.Client) {
+	tq.RegisterHandler(FederateNote, parse(t.FederateNote))
+	tq.RegisterHandler(FederateFollow, parse(t.FederateFollow))
+}
+
+func parse[T any](handler func(context.Context, T) error) func(context.Context, taskqueue.Task) error {
+	return func(ctx context.Context, task taskqueue.Task) error {
+		var data T
+		if err := json.Unmarshal(task.Payload, &data); err != nil {
+			return err
+		}
+
+		return handler(ctx, data)
+	}
+}