refactor: rewrite project from Rust to Go

- Replace Axum with Chi router - Replace sqlx with pgx for PostgreSQL - Replace tonic/prost with grpc-go - Replace tracing with zerolog - Update flake.nix for Go build with protoc generation - Preserve all existing endpoints and functionality Stack: Chi, pgx, grpc-go, zerolog, yaml.v3
2026-04-29 10:45:05 +02:00
parent f24543f401
commit 41fb033d30
48 changed files with 2306 additions and 6652 deletions
@@ -0,0 +1,280 @@
+package api
+
+import (
+	"encoding/json"
+	"net/http"
+
+	"github.com/fujin/music-agregator/internal/database"
+	"github.com/fujin/music-agregator/internal/indexer"
+	"github.com/fujin/music-agregator/internal/metadata"
+	"github.com/fujin/music-agregator/internal/services"
+	"github.com/go-chi/chi/v5"
+)
+
+type Handlers struct {
+	IndexerService *services.IndexerService
+	TorrentService *services.TorrentService
+	MetadataClient *metadata.Client
+	DB             *database.DB
+}
+
+func (h *Handlers) Health(w http.ResponseWriter, r *http.Request) {
+	writeJSON(w, http.StatusOK, map[string]string{"status": "ok"})
+}
+
+func (h *Handlers) ListIndexers(w http.ResponseWriter, r *http.Request) {
+	indexers := h.IndexerService.GetIndexers(r.Context())
+	writeJSON(w, http.StatusOK, indexers)
+}
+
+type searchRequest struct {
+	Artist string  `json:"artist"`
+	Album  *string `json:"album,omitempty"`
+	Year   *uint32 `json:"year,omitempty"`
+	Limit  int     `json:"limit,omitempty"`
+	Offset int     `json:"offset,omitempty"`
+}
+
+func (h *Handlers) SearchIndexers(w http.ResponseWriter, r *http.Request) {
+	var req searchRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		writeError(w, http.StatusBadRequest, "invalid request body")
+		return
+	}
+
+	if req.Limit == 0 {
+		req.Limit = 20
+	}
+
+	criteria := &indexer.MusicSearchCriteria{
+		Artist: req.Artist,
+		Album:  req.Album,
+		Year:   req.Year,
+		Limit:  req.Limit,
+		Offset: req.Offset,
+	}
+
+	results, err := h.IndexerService.Search(r.Context(), criteria, nil)
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, results)
+}
+
+func (h *Handlers) ListTorrents(w http.ResponseWriter, r *http.Request) {
+	torrents, err := h.TorrentService.ListTorrents(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, torrents)
+}
+
+func (h *Handlers) GetTorrent(w http.ResponseWriter, r *http.Request) {
+	hash := chi.URLParam(r, "hash")
+	torrent, err := h.TorrentService.GetTorrent(r.Context(), hash)
+	if err != nil {
+		writeError(w, http.StatusNotFound, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, torrent)
+}
+
+type addTorrentRequest struct {
+	URL      string  `json:"url"`
+	SavePath *string `json:"save_path,omitempty"`
+}
+
+func (h *Handlers) AddTorrent(w http.ResponseWriter, r *http.Request) {
+	var req addTorrentRequest
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		writeError(w, http.StatusBadRequest, "invalid request body")
+		return
+	}
+
+	if err := h.TorrentService.AddTorrentURL(r.Context(), req.URL, req.SavePath); err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, map[string]string{"status": "added"})
+}
+
+type removeTorrentRequest struct {
+	DeleteFiles bool `json:"delete_files"`
+}
+
+func (h *Handlers) RemoveTorrent(w http.ResponseWriter, r *http.Request) {
+	hash := chi.URLParam(r, "hash")
+
+	var req removeTorrentRequest
+	json.NewDecoder(r.Body).Decode(&req)
+
+	if err := h.TorrentService.RemoveTorrent(r.Context(), hash, req.DeleteFiles); err != nil {
+		writeError(w, http.StatusNotFound, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, map[string]string{"status": "removed"})
+}
+
+func (h *Handlers) PauseTorrent(w http.ResponseWriter, r *http.Request) {
+	hash := chi.URLParam(r, "hash")
+	if err := h.TorrentService.PauseTorrent(r.Context(), hash); err != nil {
+		writeError(w, http.StatusNotFound, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, map[string]string{"status": "paused"})
+}
+
+func (h *Handlers) ResumeTorrent(w http.ResponseWriter, r *http.Request) {
+	hash := chi.URLParam(r, "hash")
+	if err := h.TorrentService.ResumeTorrent(r.Context(), hash); err != nil {
+		writeError(w, http.StatusNotFound, err.Error())
+		return
+	}
+	writeJSON(w, http.StatusOK, map[string]string{"status": "resumed"})
+}
+
+func (h *Handlers) SearchArtists(w http.ResponseWriter, r *http.Request) {
+	var req struct {
+		Query  string `json:"query"`
+		Limit  int32  `json:"limit,omitempty"`
+		Offset int32  `json:"offset,omitempty"`
+	}
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		writeError(w, http.StatusBadRequest, "invalid request body")
+		return
+	}
+
+	if req.Limit == 0 {
+		req.Limit = 10
+	}
+
+	result, err := h.MetadataClient.SearchArtists(r.Context(), req.Query, req.Limit, req.Offset)
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, result)
+}
+
+func (h *Handlers) GetArtistAlbums(w http.ResponseWriter, r *http.Request) {
+	artistID := chi.URLParam(r, "id")
+
+	result, err := h.MetadataClient.GetArtistAlbums(r.Context(), artistID, 500, 0)
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, result)
+}
+
+func (h *Handlers) Sync(w http.ResponseWriter, r *http.Request) {
+	var req struct {
+		Artist   string  `json:"artist"`
+		Album    *string `json:"album,omitempty"`
+		Download *bool   `json:"download,omitempty"`
+		Store    *bool   `json:"store,omitempty"`
+	}
+	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+		writeError(w, http.StatusBadRequest, "invalid request body")
+		return
+	}
+
+	download := true
+	if req.Download != nil {
+		download = *req.Download
+	}
+	store := true
+	if req.Store != nil {
+		store = *req.Store
+	}
+
+	options := services.SyncOptions{
+		Artist:   req.Artist,
+		Album:    req.Album,
+		Download: download,
+		Store:    store,
+	}
+
+	result, err := services.Sync(r.Context(), options, h.MetadataClient, h.IndexerService, h.TorrentService, h.DB)
+	if err != nil {
+		if _, ok := err.(*services.NotFoundError); ok {
+			writeError(w, http.StatusNotFound, err.Error())
+			return
+		}
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, result)
+}
+
+func (h *Handlers) ListLibraryArtists(w http.ResponseWriter, r *http.Request) {
+	if h.DB == nil {
+		writeError(w, http.StatusServiceUnavailable, "database not connected")
+		return
+	}
+
+	artists, err := h.DB.ListArtists(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, artists)
+}
+
+func (h *Handlers) ListLibraryAlbums(w http.ResponseWriter, r *http.Request) {
+	if h.DB == nil {
+		writeError(w, http.StatusServiceUnavailable, "database not connected")
+		return
+	}
+
+	albums, err := h.DB.ListAllAlbums(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, albums)
+}
+
+func (h *Handlers) LibraryStats(w http.ResponseWriter, r *http.Request) {
+	if h.DB == nil {
+		writeError(w, http.StatusServiceUnavailable, "database not connected")
+		return
+	}
+
+	artistCount, err := h.DB.CountArtists(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	albumCount, err := h.DB.CountAlbums(r.Context())
+	if err != nil {
+		writeError(w, http.StatusInternalServerError, err.Error())
+		return
+	}
+
+	writeJSON(w, http.StatusOK, map[string]int64{
+		"artists": artistCount,
+		"albums":  albumCount,
+	})
+}
+
+func writeJSON(w http.ResponseWriter, status int, v any) {
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(status)
+	json.NewEncoder(w).Encode(v)
+}
+
+func writeError(w http.ResponseWriter, status int, message string) {
+	writeJSON(w, status, map[string]string{"error": message})
+}