Tony
3bb4ab494e
- Added `internal/db/store.go` to define the Store interface for database operations, including user, organization, app, channel, release, patch, and artifact management. - Introduced `internal/models/models.go` changes to reflect updated organization and artifact response structures. - Created `internal/storage/factory.go`, `local.go`, and `s3.go` to implement storage backends for local filesystem and S3-compatible services. - Removed deprecated `internal/storage/storage.go` and refactored storage interface into `internal/storage/store.go`. - Updated frontend JavaScript in `web/js/app.js` to display server health information, including storage and database backend details.
92 lines
3.0 KiB
Go
92 lines
3.0 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
// LocalStore implements Store using the local filesystem.
|
|
// Release artifacts go under <baseDir>/releases/; patch artifacts
|
|
// under <baseDir>/patches/ (publicly served by the built-in upload
|
|
// handler — no presigned URLs needed).
|
|
type LocalStore struct {
|
|
baseDir string
|
|
// serverBaseURL is the external URL of this server (e.g. http://localhost:8080).
|
|
// Used when generating download URLs for device-side patch checks.
|
|
serverBaseURL string
|
|
}
|
|
|
|
// NewLocalStore creates a local-filesystem storage backend.
|
|
// baseDir is created automatically if it does not exist.
|
|
func NewLocalStore(baseDir, serverBaseURL string) (*LocalStore, error) {
|
|
abs, err := filepath.Abs(baseDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("local storage: %w", err)
|
|
}
|
|
if err := os.MkdirAll(filepath.Join(abs, "releases"), 0o755); err != nil {
|
|
return nil, fmt.Errorf("local storage: create releases dir: %w", err)
|
|
}
|
|
if err := os.MkdirAll(filepath.Join(abs, "patches"), 0o755); err != nil {
|
|
return nil, fmt.Errorf("local storage: create patches dir: %w", err)
|
|
}
|
|
if serverBaseURL == "" {
|
|
serverBaseURL = "http://localhost:8080"
|
|
}
|
|
return &LocalStore{baseDir: abs, serverBaseURL: serverBaseURL}, nil
|
|
}
|
|
|
|
func (s *LocalStore) BackendName() string { return "Local filesystem" }
|
|
|
|
// GeneratePresignedUploadURL returns a server endpoint URL that the
|
|
// CLI can POST the artifact to. The upload handler (registered on
|
|
// the router) writes the file to disk.
|
|
func (s *LocalStore) GeneratePresignedUploadURL(_ context.Context, objectKey string, isPublic bool, _ time.Duration) (string, error) {
|
|
scope := "releases"
|
|
if isPublic {
|
|
scope = "patches"
|
|
}
|
|
return fmt.Sprintf("%s/storage/upload/%s/%s", s.serverBaseURL, scope, objectKey), nil
|
|
}
|
|
|
|
// GeneratePublicDownloadURL returns a public download URL for a patch.
|
|
func (s *LocalStore) GeneratePublicDownloadURL(_ context.Context, objectKey string) (string, error) {
|
|
return fmt.Sprintf("%s/storage/dl/patches/%s", s.serverBaseURL, objectKey), nil
|
|
}
|
|
|
|
// UploadObject writes data directly to the local filesystem.
|
|
func (s *LocalStore) UploadObject(_ context.Context, objectKey string, reader io.Reader, _ int64, _ string, isPublic bool) error {
|
|
scope := "releases"
|
|
if isPublic {
|
|
scope = "patches"
|
|
}
|
|
target := filepath.Join(s.baseDir, scope, objectKey)
|
|
if err := os.MkdirAll(filepath.Dir(target), 0o755); err != nil {
|
|
return fmt.Errorf("local storage: %w", err)
|
|
}
|
|
f, err := os.Create(target)
|
|
if err != nil {
|
|
return fmt.Errorf("local storage: %w", err)
|
|
}
|
|
defer f.Close()
|
|
if _, err := io.Copy(f, reader); err != nil {
|
|
return fmt.Errorf("local storage: write: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetObject reads an object from the local filesystem.
|
|
func (s *LocalStore) GetObject(_ context.Context, objectKey string, isPublic bool) (io.ReadCloser, error) {
|
|
scope := "releases"
|
|
if isPublic {
|
|
scope = "patches"
|
|
}
|
|
return os.Open(filepath.Join(s.baseDir, scope, objectKey))
|
|
}
|
|
|
|
// BaseDir returns the absolute base directory.
|
|
func (s *LocalStore) BaseDir() string { return s.baseDir }
|