migrate orion repo into monorepo structure

2026-02-21 08:48:34 -08:00
parent 144596fcaa
commit b5f90709de
91 changed files with 19775 additions and 0 deletions
@@ -0,0 +1,151 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net/http"
+	"os"
+
+	"cloud.google.com/go/storage"
+	pbaero "github.com/flowy-live/llink/genproto/aero"
+	"github.com/flowy-live/llink/internal"
+	"github.com/flowy-live/llink/internal/auth"
+	"github.com/flowy-live/llink/internal/db"
+	"github.com/flowy-live/llink/internal/depot"
+	"github.com/flowy-live/llink/internal/handler"
+	"github.com/flowy-live/llink/internal/human"
+	"github.com/flowy-live/llink/internal/middleware"
+	"github.com/flowy-live/llink/internal/network"
+	"github.com/flowy-live/llink/internal/particle"
+	"github.com/flowy-live/llink/internal/utils"
+	"github.com/redis/go-redis/v9"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+const (
+	// FIX: Use separate redis instance. We start with higher number because use this same instance in helios.
+	REDIS_DATABASE_FOR_AUTH int = 4
+)
+
+func redisForAuth() *redis.Client {
+	return internal.ConnectAndTestRedis(REDIS_DATABASE_FOR_AUTH)
+}
+
+func main() {
+	port := utils.MustGetEnv("PORT")
+	gcsBucket := utils.MustGetEnv("GCS_BUCKET")
+
+	// Initialize database
+	db.Init()
+	defer db.Cleanup()
+
+	// Initialize Redis for auth
+	redisClient := redisForAuth()
+
+	// Initialize GCS client
+	ctx := context.Background()
+	storageClient, err := storage.NewClient(ctx)
+	if err != nil {
+		slog.Error("failed to create GCS client", "error", err)
+		os.Exit(1)
+	}
+	defer storageClient.Close()
+
+	aeroAddr := utils.MustGetEnv("AERO_ADDR")
+	if aeroAddr == "" {
+		slog.Error("must provide AERO_ADDR")
+		os.Exit(1)
+	}
+	aeroServer, err := grpc.NewClient(aeroAddr, grpc.WithTransportCredentials(insecure.NewCredentials()))
+	if err != nil {
+		slog.Error("connection to aero server invalid", "error", err)
+		os.Exit(1)
+	}
+	defer aeroServer.Close()
+	aeroSvc := pbaero.NewPrimaryClient(aeroServer)
+
+	// Initialize services
+	authSvc := auth.NewAuthService(redisClient, aeroSvc)
+	humanSvc := human.NewService(db.Pool())
+	networkSvc := network.NewService(db.Pool())
+	particleSvc := particle.NewService(db.Pool(), networkSvc)
+	depotSvc := depot.NewService(db.Pool(), storageClient, depot.Config{
+		GoogleServiceAccountEmail: utils.MustGetEnv("GOOGLE_SERVICE_ACCOUNT_EMAIL"),
+		BucketName:                gcsBucket,
+	})
+
+	// Initialize handler
+	h := handler.NewHandler(authSvc, humanSvc, networkSvc, particleSvc, depotSvc)
+
+	// Helper to wrap handlers with auth middleware
+	withAuth := func(hf http.HandlerFunc) http.Handler {
+		return middleware.Auth(authSvc)(http.HandlerFunc(hf))
+	}
+
+	mux := http.NewServeMux()
+
+	// ==========================================================================
+	// Public routes (no auth)
+	// ==========================================================================
+	mux.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+	})
+	mux.HandleFunc("GET /health", func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+	})
+	mux.HandleFunc("POST /auth/request-code", h.RequestSignInCode)
+	mux.HandleFunc("POST /auth/sign-in", h.SignIn)
+
+	// ==========================================================================
+	// Protected routes (auth required)
+	// ==========================================================================
+
+	// Auth
+	mux.Handle("POST /auth/sign-out", withAuth(h.SignOut))
+	mux.Handle("GET /auth/me", withAuth(h.GetCurrentHuman))
+
+	// Bootstrap startup data
+	mux.Handle("GET /startup", withAuth(h.StartupData))
+
+	// Networks
+	mux.Handle("POST /networks", withAuth(h.CreateNetwork))
+	mux.Handle("GET /networks", withAuth(h.ListNetworks))
+	mux.Handle("GET /networks/{id}", withAuth(h.GetNetwork))
+	mux.Handle("POST /networks/{id}/members", withAuth(h.AddMembersToNetwork))
+	// TODO: what about members who are part of streams visibility within this network?
+	mux.Handle("DELETE /networks/{id}/members/{email}", withAuth(h.RemoveMemberFromNetwork))
+	mux.Handle("PUT /networks/{id}/capacity", withAuth(h.SetOpenStreamCapacity))
+
+	// Streams
+	mux.Handle("POST /networks/{network_id}/streams", withAuth(h.CreateStream))
+	mux.Handle("GET /streams/{id}", withAuth(h.GetStream))
+	mux.Handle("PATCH /streams/{id}", withAuth(h.UpdateStream))
+	mux.Handle("POST /streams/{id}/particles", withAuth(h.CreateStreamParticle))
+	mux.Handle("POST /streams/{id}/open", withAuth(h.OpenStream))
+	mux.Handle("POST /streams/{id}/close", withAuth(h.CloseStream))
+	mux.Handle("POST /streams/{id}/members", withAuth(h.AddMembers))
+	mux.Handle("DELETE /streams/{id}/members", withAuth(h.RemoveMembers))
+
+	// Particles
+	mux.Handle("GET /networks/{network_id}/particles", withAuth(h.ListParticles))
+	mux.Handle("GET /particles/{id}", withAuth(h.GetParticle))
+	mux.Handle("PATCH /particles/{id}", withAuth(h.UpdateParticle))
+	mux.Handle("DELETE /particles/{id}", withAuth(h.DeleteParticle))
+	mux.Handle("POST /particles/{id}/seen", withAuth(h.MarkSeen))
+	mux.Handle("POST /particles/{id}/ack", withAuth(h.AckParticle))
+	mux.Handle("GET /particles/{id}/download", withAuth(h.DownloadParticle))
+	mux.Handle("POST /particles/seen", withAuth(h.MarkSeenBatch))
+
+	// Depot
+	mux.Handle("POST /depot/upload", withAuth(h.PrepareUpload))
+	mux.Handle("POST /depot/objects/{id}/confirm", withAuth(h.ConfirmUpload))
+
+	addr := fmt.Sprintf("0.0.0.0:%s", port)
+	slog.Info("running server", "addr", addr)
+	if err := http.ListenAndServe(addr, mux); err != nil {
+		slog.Error("server failed", "error", err)
+		os.Exit(1)
+	}
+}