storage.go

package client

import (
	"net/http"
	"os"
	"path/filepath"
	"sync"
	"time"
)

// BuildStorage defines the behavior for compiling and serving the WASM client.
type BuildStorage interface {
	// Compile performs the compilation.
	// For Memory: compiles to buffer.
	// For Disk: compiles to disk.
	Compile() error

	// RegisterRoutes registers the WASM file handler on the mux.
	RegisterRoutes(mux *http.ServeMux)

	// Name returns the Storage name for logging/debugging
	Name() string
}

// MemoryStorage compiles WASM to memory and serves it directly.
type MemoryStorage struct {
	Client *WasmClient // Access to config and logger

	Mu          sync.RWMutex
	WasmContent []byte
	LastCompile time.Time
}

func (s *MemoryStorage) Name() string {
	return "In-Memory"
}

func (s *MemoryStorage) Compile() error {
	// Delegate to active builder's CompileToMemory
	// Note: activeSizeBuilder is in WasmClient
	content, err := s.Client.activeSizeBuilder.CompileToMemory()
	if err != nil {
		return err
	}

	s.Mu.Lock()
	s.WasmContent = content
	s.LastCompile = time.Now()
	s.Mu.Unlock()

	return nil
}

// DiskStorage compiles WASM to disk and serves the static file.
type DiskStorage struct {
	Client *WasmClient
}

func (s *DiskStorage) Name() string {
	return "External"
}

func (s *DiskStorage) Compile() error {
	// Ensure directory exists
	outDir := filepath.Join(s.Client.AppRootDir, s.Client.Config.OutputDir())
	if err := os.MkdirAll(outDir, 0755); err != nil {
		return err
	}

	// Use existing CompileProgram which writes to config.OutputDir
	return s.Client.activeSizeBuilder.CompileProgram()
}

func (s *DiskStorage) RegisterRoutes(mux *http.ServeMux) {
	routePath := s.Client.wasmRoutePath()
	result := filepath.Join(s.Client.Config.OutputDir(), s.Client.OutputName+".wasm")
	// Note: Config.OutputDir is relative to AppRootDir usually, but ServeFile needs OS path.
	// We need absolute path.
	absPath := filepath.Join(s.Client.AppRootDir, result)

	mux.HandleFunc(routePath, func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "application/wasm")
		http.ServeFile(w, r, absPath)
	})
	s.Client.LogSuccessState("http route:", routePath)
}