From 074ec158b3a51d2aedbbdaca944f8e8de08506fa Mon Sep 17 00:00:00 2001
From: michealroberts <michael@observerly.com>
Date: Sun, 16 Feb 2025 20:11:45 +0000
Subject: [PATCH] feat: add NewSolveServer() in @observerly/nova

feat: add NewSolveServer() in @observerly/nova
---
 service/solve/server.go | 94 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 service/solve/server.go

diff --git a/service/solve/server.go b/service/solve/server.go
new file mode 100644
index 0000000..954b2c7
--- /dev/null
+++ b/service/solve/server.go
@@ -0,0 +1,94 @@
+/*****************************************************************************************************************/
+
+//	@author		Michael Roberts <michael@observerly.com>
+//	@package	@observerly/nova
+//	@license	Copyright © 2021-2024 observerly
+
+/*****************************************************************************************************************/
+
+package solve
+
+/*****************************************************************************************************************/
+
+import (
+	"context"
+	"fmt"
+	pb "nova/internal/gen/solve/v1/solvev1connect"
+
+	firebase "firebase.google.com/go/v4"
+	"firebase.google.com/go/v4/storage"
+	"github.com/observerly/iris/pkg/fits"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+
+	"nova/internal/stores"
+)
+
+/*****************************************************************************************************************/
+
+type server struct {
+	pb.UnimplementedSolveServiceHandler
+	App     *firebase.App
+	Logger  zerolog.Logger
+	Storage *storage.Client
+	Client  *stores.FirebaseStore
+}
+
+/*****************************************************************************************************************/
+
+func NewSolveServer(app *firebase.App, storage *storage.Client) *server {
+	// Create a new logger:
+	logger := log.With().Str("@observerly/solver", "rpc").Logger()
+
+	// Create a new Firebase Storage client:
+	client := stores.NewFirebaseStorageClient(storage)
+
+	return &server{
+		App:     app,
+		Logger:  logger,
+		Storage: storage,
+		Client:  client,
+	}
+}
+
+/*****************************************************************************************************************/
+
+func (s *server) RetrieveFITSFromStorage(
+	ctx context.Context,
+	bucketName string,
+	location string,
+) (*fits.FITSImage, error) {
+	// Assume an image of 2x2 pixels with 16-bit depth, and no offset:
+	fit := fits.NewFITSImage(2, 0, 0, 65535)
+
+	// Get the buffer from the storage client:
+	buff, err := s.Client.RetriveBuffer(ctx, bucketName, location)
+
+	if err != nil {
+		s.Logger.Error().Err(err).Msg("Failed to get buffer")
+		return nil, err
+	}
+
+	// Read in our exposure data into the image:
+	err = fit.Read(buff)
+
+	if err != nil {
+		s.Logger.Error().Err(err).Msg("Failed to read exposure data")
+		return nil, fmt.Errorf("failed to read exposure data: %w", err)
+	}
+
+	// We know the image is 2D, so we can extract the width from the fits image:
+	width := fit.Header.Naxis1
+
+	// We know the image is 2D, so we can extract the height from the fits image:
+	height := fit.Header.Naxis2
+
+	if fit.Pixels != width*height {
+		s.Logger.Error().Msg("Failed to read exposure data")
+		return nil, fmt.Errorf("failed to read exposure data as the number of pixels does not match the width and height")
+	}
+
+	return fit, nil
+}
+
+/*****************************************************************************************************************/