清理代码库

db.go

@@ -1,13 +1,14 @@
 package main
 
 import (
-	"butterfliu/internal/repository"
 	"butterfliu/migrations"
 	"database/sql"
 	"fmt"
 	"log"
 	"os"
 	"path/filepath"
+
+	_ "github.com/mattn/go-sqlite3"
 )
 
 var DB *sql.DB
@@ -25,10 +26,14 @@ func InitDB(dbPath string) error {
 	}
 
 	var err error
+	DB, err = sql.Open("sqlite3", dbPath)
+	if err != nil {
+		return fmt.Errorf("failed to open database: %v", err)
+	}
 
-	DB, _ = sql.Open("sqlite3", dbPath)
-
-	DB.Ping()
+	if err = DB.Ping(); err != nil {
+		return fmt.Errorf("failed to ping database: %v", err)
+	}
 
 	if err = migrations.InitMigrationTable(DB); err != nil {
 		return fmt.Errorf("failed to initialize migration table: %v", err)
@@ -55,390 +60,3 @@ func CloseDB() error {
 	}
 	return nil
 }
-
-func AddLibrary(name, path string) (repository.Library, error) {
-	absPath, err := filepath.Abs(path)
-	if err != nil {
-		return repository.Library{}, err
-	}
-
-	result, err := DB.Exec(
-		"INSERT INTO libraries (name, path) VALUES (?, ?)",
-		name, absPath,
-	)
-	if err != nil {
-		return repository.Library{}, err
-	}
-
-	id, err := result.LastInsertId()
-	if err != nil {
-		return repository.Library{}, err
-	}
-
-	return repository.Library{
-		ID:   int(id),
-		Name: name,
-		Path: absPath,
-	}, nil
-}
-
-func GetLibraries() ([]repository.Library, error) {
-	rows, err := DB.Query("SELECT id, name, path FROM libraries")
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-
-	libraries := []repository.Library{}
-	for rows.Next() {
-		var lib repository.Library
-		if err := rows.Scan(&lib.ID, &lib.Name, &lib.Path); err != nil {
-			return nil, err
-		}
-		libraries = append(libraries, lib)
-	}
-
-	return libraries, nil
-}
-
-func GetLibraryByID(id int) (repository.Library, error) {
-	row := DB.QueryRow("SELECT id, name, path FROM libraries WHERE id = ?", id)
-	var lib repository.Library
-	if err := row.Scan(&lib.ID, &lib.Name, &lib.Path); err != nil {
-		return repository.Library{}, err
-	}
-	return lib, nil
-}
-
-func UpdateLibraryName(id int, name string) error {
-	_, err := DB.Exec(
-		"UPDATE libraries SET name = ? WHERE id = ?",
-		name, id,
-	)
-	return err
-}
-func UpdateLibraryPath(id int, path string) error {
-	_, err := DB.Exec(
-		"UPDATE libraries SET path = ? WHERE id = ?",
-		path, id,
-	)
-	return err
-}
-func DeleteLibrary(id int) error {
-	_, err := DB.Exec("DELETE FROM libraries WHERE id = ?", id)
-	return err
-}
-
-func AddMediaFile(path string, libraryID int) (repository.MediaFile, error) {
-	result, err := DB.Exec("INSERT INTO media_files (path, library_id) VALUES (?, ?)", path, libraryID)
-	if err != nil {
-		return repository.MediaFile{}, err
-	}
-
-	id, err := result.LastInsertId()
-	if err != nil {
-		return repository.MediaFile{}, err
-	}
-
-	return repository.MediaFile{
-		ID:   int(id),
-		Path: path,
-	}, nil
-}
-
-func GetMediaFileByPath(path string) (repository.MediaFile, error) {
-	row := DB.QueryRow("SELECT id, path FROM media_files WHERE path = ?", path)
-	var mediaFile repository.MediaFile
-	if err := row.Scan(&mediaFile.ID, &mediaFile.Path); err != nil {
-		return repository.MediaFile{}, err
-	}
-	return mediaFile, nil
-}
-
-func AddArtist(name string) (repository.Artist, error) {
-	result, err := DB.Exec("INSERT INTO artists (name) VALUES (?)", name)
-	if err != nil {
-		return repository.Artist{}, err
-	}
-
-	id, err := result.LastInsertId()
-	if err != nil {
-		return repository.Artist{}, err
-	}
-
-	return repository.Artist{
-		ID:   int(id),
-		Name: name,
-	}, nil
-}
-
-func GetArtists() ([]repository.Artist, error) {
-	rows, err := DB.Query("SELECT id, name FROM artists")
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	artists := []repository.Artist{}
-	for rows.Next() {
-		var artist repository.Artist
-		if err := rows.Scan(&artist.ID, &artist.Name); err != nil {
-			return nil, err
-		}
-		artists = append(artists, artist)
-	}
-	return artists, nil
-}
-func GetArtistByID(id int) (repository.Artist, error) {
-	row := DB.QueryRow("SELECT id, name FROM artists WHERE id = ?", id)
-	var artist repository.Artist
-	if err := row.Scan(&artist.ID, &artist.Name); err != nil {
-		return repository.Artist{}, err
-	}
-	return artist, nil
-}
-func GetArtistByName(name string) (repository.Artist, error) {
-	row := DB.QueryRow("SELECT id, name FROM artists WHERE name = ?", name)
-	var artist repository.Artist
-	if err := row.Scan(&artist.ID, &artist.Name); err != nil {
-		return repository.Artist{}, err
-	}
-	return artist, nil
-}
-func UpdateArtistName(id int, name string) error {
-	_, err := DB.Exec("UPDATE artists SET name = ? WHERE id = ?", name, id)
-	return err
-}
-
-func DeleteArtist(id int) error {
-	_, err := DB.Exec("DELETE FROM artists WHERE id = ?", id)
-	return err
-}
-
-func AddAlbum(title string, artistID int) (repository.Album, error) {
-	result, err := DB.Exec("INSERT INTO albums (title, artist_id) VALUES (?, ?)", title, artistID)
-	if err != nil {
-		return repository.Album{}, err
-	}
-	id, err := result.LastInsertId()
-	if err != nil {
-		return repository.Album{}, err
-	}
-
-	return repository.Album{
-		ID:       int(id),
-		Title:    title,
-		ArtistID: artistID,
-	}, nil
-}
-
-func GetAlbums() ([]repository.Album, error) {
-	rows, err := DB.Query("SELECT id, title, artist_id FROM albums")
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	albums := []repository.Album{}
-	for rows.Next() {
-		var album repository.Album
-		if err := rows.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
-			return nil, err
-		}
-		albums = append(albums, album)
-	}
-	return albums, nil
-}
-
-func GetAlbumByID(id int) (repository.Album, error) {
-	row := DB.QueryRow("SELECT id, title, artist_id FROM albums WHERE id = ?", id)
-	var album repository.Album
-	if err := row.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
-		return repository.Album{}, err
-	}
-	return album, nil
-}
-
-func GetAlbumByTitle(title string) (repository.Album, error) {
-	row := DB.QueryRow("SELECT id, title, artist_id FROM albums WHERE title = ?", title)
-	var album repository.Album
-	if err := row.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
-		return repository.Album{}, err
-	}
-	return album, nil
-}
-
-func GetAlbumByTitleAndArtist(title string, artistID int) (repository.Album, error) {
-	row := DB.QueryRow("SELECT id, title, artist_id FROM albums WHERE title = ? AND artist_id = ?", title, artistID)
-	var album repository.Album
-	if err := row.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
-		return repository.Album{}, err
-	}
-	return album, nil
-}
-
-func GetAlbumsByArtist(artistID int) ([]repository.Album, error) {
-	rows, err := DB.Query("SELECT id, title, artist_id FROM albums WHERE artist_id = ?", artistID)
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	albums := []repository.Album{}
-	for rows.Next() {
-		var album repository.Album
-		if err := rows.Scan(&album.ID, &album.Title, &album.ArtistID); err != nil {
-			return nil, err
-		}
-		albums = append(albums, album)
-	}
-	return albums, nil
-}
-
-func UpdateAlbumTitle(id int, title string) error {
-	_, err := DB.Exec("UPDATE albums SET title = ? WHERE id = ?", title, id)
-	return err
-}
-func UpdateAlbumArtistID(id, artistID int) error {
-	_, err := DB.Exec("UPDATE albums SET artist_id = ?", artistID)
-	return err
-}
-func DeleteAlbum(id int) error {
-	_, err := DB.Exec("DELETE FROM albums WHERE id = ?", id)
-	return err
-}
-
-func AddSong(title string, artistID, albumID, duration, mediaFileID int) (repository.Song, error) {
-	result, err := DB.Exec("INSERT INTO songs (title, artist_id, album_id, duration, media_file_id) VALUES (?, ?, ?, ?, ?)", title, artistID, albumID, duration, mediaFileID)
-	if err != nil {
-		return repository.Song{}, err
-	}
-
-	id, err := result.LastInsertId()
-	if err != nil {
-		return repository.Song{}, err
-	}
-
-	return repository.Song{
-		ID:          int(id),
-		Title:       title,
-		ArtistID:    artistID,
-		AlbumID:     albumID,
-		Duration:    duration,
-		MediaFileID: mediaFileID,
-	}, nil
-}
-func GetSongs() ([]repository.Song, error) {
-	rows, err := DB.Query("SELECT id, artist_id, album_id, duration, media_file_id FROM songs")
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	songs := []repository.Song{}
-	for rows.Next() {
-		var song repository.Song
-		if err := rows.Scan(&song.ID, &song.ArtistID, &song.AlbumID, &song.Duration, &song.MediaFileID); err != nil {
-			return nil, err
-		}
-		songs = append(songs, song)
-	}
-	return songs, nil
-}
-func UpdateSongTitle(id int, title string) error {
-	_, err := DB.Exec("UPDATE songs SET title = ? WHERE id = ?", title, id)
-	return err
-}
-func UpdateSongArtistID(id int, artistID int) error {
-	_, err := DB.Exec("UPDATE songs SET artist_id = ?", artistID)
-	return err
-}
-func UpdateSongAlbumID(id, albumID int) error {
-	_, err := DB.Exec("UPDATE songs SET album_id = ?", albumID)
-	return err
-}
-func UpdateSongDuration(id, duration int) error {
-	_, err := DB.Exec("UPDATE songs SET duration = ?", duration)
-	return err
-}
-func DeleteSong(id int) error {
-	_, err := DB.Exec("DELETE FROM songs WHERE id = ?", id)
-	return err
-}
-
-func GetSongsByLibrary(libraryID int) ([]repository.Song, error) {
-	rows, err := DB.Query(`
-		SELECT s.id, s.title, s.artist_id, s.album_id, s.duration, s.media_file_id
-		FROM songs s
-		INNER JOIN media_files mf ON s.media_file_id = mf.id
-		WHERE mf.library_id = ?
-		ORDER BY s.title
-	`, libraryID)
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	songs := []repository.Song{}
-	for rows.Next() {
-		var song repository.Song
-		if err := rows.Scan(&song.ID, &song.Title, &song.ArtistID, &song.AlbumID, &song.Duration, &song.MediaFileID); err != nil {
-			return nil, err
-		}
-		songs = append(songs, song)
-	}
-	return songs, nil
-}
-
-type SongDetail struct {
-	ID       int    `json:"id"`
-	Title    string `json:"title"`
-	Artist   string `json:"artist"`
-	Album    string `json:"album"`
-	Duration int    `json:"duration"`
-	Path     string `json:"path"`
-}
-
-func GetSongsByLibraryWithDetails(libraryID int) ([]SongDetail, error) {
-	rows, err := DB.Query(`
-		SELECT s.id, s.title, a.name as artist_name, al.title as album_title, s.duration, mf.path
-		FROM songs s
-		INNER JOIN media_files mf ON s.media_file_id = mf.id
-		INNER JOIN artists a ON s.artist_id = a.id
-		INNER JOIN albums al ON s.album_id = al.id
-		WHERE mf.library_id = ?
-		ORDER BY s.title
-	`, libraryID)
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	songs := []SongDetail{}
-	for rows.Next() {
-		var song SongDetail
-		if err := rows.Scan(&song.ID, &song.Title, &song.Artist, &song.Album, &song.Duration, &song.Path); err != nil {
-			return nil, err
-		}
-		songs = append(songs, song)
-	}
-	return songs, nil
-}
-
-func GetAllSongsWithDetails() ([]SongDetail, error) {
-	rows, err := DB.Query(`
-		SELECT s.id, s.title, a.name as artist_name, al.title as album_title, s.duration, mf.path
-		FROM songs s
-		INNER JOIN media_files mf ON s.media_file_id = mf.id
-		INNER JOIN artists a ON s.artist_id = a.id
-		INNER JOIN albums al ON s.album_id = al.id
-		ORDER BY s.title
-	`)
-	if err != nil {
-		return nil, err
-	}
-	defer rows.Close()
-	songs := []SongDetail{}
-	for rows.Next() {
-		var song SongDetail
-		if err := rows.Scan(&song.ID, &song.Title, &song.Artist, &song.Album, &song.Duration, &song.Path); err != nil {
-			return nil, err
-		}
-		songs = append(songs, song)
-	}
-	return songs, nil
-}