Gabe Farrell 9 months ago
commit 6f54d98ea2

2
.gitignore vendored

@ -0,0 +1,2 @@
kanpeki.db
media

@ -0,0 +1,13 @@
run:
go run cmd/api/main.go
test:
go test ./...
db.up:
goose -dir db/migrations sqlite3 kanpeki.db up
db.rm:
rm kanpeki.db
db.reset: db.rm db.up

@ -0,0 +1 @@
# kanpeki

@ -0,0 +1,67 @@
package app
import (
"context"
"fmt"
"io"
"net/http"
"os"
"os/signal"
"syscall"
"time"
"github.com/gabehf/kanpeki/app/middleware"
"github.com/gabehf/kanpeki/internal/cfg"
"github.com/gabehf/kanpeki/internal/db"
"github.com/gabehf/kanpeki/internal/logger"
"github.com/go-chi/chi/v5"
chimiddleware "github.com/go-chi/chi/v5/middleware"
"github.com/rs/zerolog"
)
func NewServer(logger *zerolog.Logger, cfg *cfg.Config, db db.DB) http.Handler {
mux := chi.NewRouter()
// bind general middleware to mux
mux.Use(middleware.Logger(logger))
mux.Use(chimiddleware.RealIP)
mux.Use(chimiddleware.Recoverer)
// call router binds on mux
bindRoutes(mux, logger, cfg, db)
return mux
}
func Run(
ctx context.Context,
getenv func(string) string,
stdout io.Writer,
db db.DB,
) error {
cfg, err := cfg.Load(getenv)
if err != nil {
return fmt.Errorf("failed to load configuration: %v", err)
}
logger := logger.Get(cfg).Output(stdout)
srv := NewServer(&logger, cfg, db)
httpServer := &http.Server{
Addr: cfg.ListenAddr,
Handler: srv,
}
logger.Info().Msg("listening on " + cfg.ListenAddr)
if err := httpServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
return err
}
// Wait for interrupt signal to gracefully shutdown the server with a timeout of 10 seconds.
// Use a buffered channel to avoid missing signals as recommended for signal.Notify
quit := make(chan os.Signal, 1)
signal.Notify(quit, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
<-quit
logger.Info().Msg("recieved server shutdown notice")
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
logger.Info().Msg("waiting for all processes to finish...")
if err := httpServer.Shutdown(ctx); err != nil {
return err
}
logger.Info().Msg("shutdown successful")
return nil
}

@ -0,0 +1,49 @@
package middleware
import (
"net/http"
"runtime/debug"
"time"
"github.com/go-chi/chi/v5/middleware"
"github.com/rs/zerolog"
)
func Logger(logger *zerolog.Logger) func(next http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) {
log := logger.With().Logger()
ww := middleware.NewWrapResponseWriter(w, r.ProtoMajor)
t1 := time.Now()
defer func() {
t2 := time.Now()
if rec := recover(); rec != nil {
log.Error().
Str("type", "error").
Timestamp().
Interface("recover_info", rec).
Bytes("debug_stack", debug.Stack()).
Msg("log system error")
http.Error(ww, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
}
log.Info().
Str("type", "access").
Timestamp().
Fields(map[string]interface{}{
"remote_ip": r.RemoteAddr,
"url": r.URL.Path,
"proto": r.Proto,
"method": r.Method,
"user_agent": r.Header.Get("User-Agent"),
"status": ww.Status(),
"latency_ms": float64(t2.Sub(t1).Nanoseconds()) / 1000000.0,
"bytes_in": r.Header.Get("Content-Length"),
"bytes_out": ww.BytesWritten(),
}).
Msg("incoming_request")
}()
next.ServeHTTP(ww, r)
}
return http.HandlerFunc(fn)
}
}

@ -0,0 +1,19 @@
package app
import (
"github.com/gabehf/kanpeki/app/routes"
"github.com/gabehf/kanpeki/internal/cfg"
"github.com/gabehf/kanpeki/internal/db"
"github.com/go-chi/chi/v5"
"github.com/rs/zerolog"
)
func bindRoutes(r *chi.Mux, logger *zerolog.Logger, cfg *cfg.Config, db db.DB) {
r.Route("/api/v1", func(r chi.Router) {
r.Route("/library", func(r chi.Router) {
r.Get("/", routes.GetLibraryInfo(logger, db))
r.Post("/", routes.CreateLibrary(logger, db))
r.Delete("/", routes.DeleteLibrary(logger, db))
})
})
}

@ -0,0 +1,26 @@
package routes
import (
"net/http"
"github.com/gabehf/kanpeki/internal/db"
"github.com/rs/zerolog"
)
func CreateLibrary(logger *zerolog.Logger, db db.DB) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
}
}
func GetLibraryInfo(logger *zerolog.Logger, db db.DB) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
}
}
func DeleteLibrary(logger *zerolog.Logger, db db.DB) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
}
}

@ -0,0 +1,24 @@
package main
import (
"context"
"log"
"os"
"github.com/gabehf/kanpeki/app"
"github.com/gabehf/kanpeki/internal/cfg"
"github.com/gabehf/kanpeki/internal/db/sqlite"
)
func main() {
ctx := context.Background()
db, err := sqlite.New(&cfg.Config{})
if err != nil {
log.Fatalf("%s\n", err)
}
defer db.Close()
if err := app.Run(ctx, os.Getenv, os.Stdout, db); err != nil {
log.Fatalf("%s\n", err)
os.Exit(1)
}
}

@ -0,0 +1,64 @@
-- +goose Up
CREATE TABLE libraries (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT UNIQUE NOT NULL,
path TEXT NOT NULL,
type TEXT NOT NULL, -- ('tv' or 'movies')
num_files INTEGER,
size INTEGER
);
CREATE TABLE tv_shows (
id INTEGER PRIMARY KEY AUTOINCREMENT,
library_id INTEGER NOT NULL,
name TEXT NOT NULL,
num_files INTEGER,
size INTEGER,
FOREIGN KEY (library_id) REFERENCES libraries(id)
);
CREATE TABLE seasons (
id INTEGER PRIMARY KEY AUTOINCREMENT,
tv_show_id INTEGER NOT NULL,
season_number INTEGER NOT NULL,
num_files INTEGER,
size INTEGER,
FOREIGN KEY (tv_show_id) REFERENCES tv_shows(id)
);
CREATE TABLE movies (
id INTEGER PRIMARY KEY AUTOINCREMENT,
library_id INTEGER NOT NULL,
title TEXT NOT NULL,
last_scanned INTEGER,
FOREIGN KEY (library_id) REFERENCES libraries(id)
);
CREATE TABLE files (
id INTEGER PRIMARY KEY AUTOINCREMENT,
library_id INTEGER NOT NULL,
tv_show_id INTEGER, -- Nullable for movies
season_id INTEGER, -- Nullable for movies
movie_id INTEGER, -- Nullable for TV episodes
filename TEXT NOT NULL,
last_scanned INTEGER,
FOREIGN KEY (library_id) REFERENCES libraries(id),
FOREIGN KEY (tv_show_id) REFERENCES tv_shows(id),
FOREIGN KEY (season_id) REFERENCES seasons(id),
FOREIGN KEY (movie_id) REFERENCES movies(id)
);
CREATE TABLE ffprobe_metadata (
id INTEGER PRIMARY KEY AUTOINCREMENT,
file_id INTEGER NOT NULL,
data JSON NOT NULL,
FOREIGN KEY (file_id) REFERENCES files(id)
);
-- +goose Down
DROP TABLE ffprobe_metadata;
DROP TABLE files;
DROP TABLE libraries;
DROP TABLE tv_shows;
DROP TABLE movies;
DROP TABLE seasons;

@ -0,0 +1,2 @@
# Install pressly/goose for DB migrations
```go install github.com/pressly/goose/v3/cmd/goose@latest```

@ -0,0 +1,14 @@
module github.com/gabehf/kanpeki
go 1.23.7
require (
github.com/go-chi/chi/v5 v5.2.1
github.com/rs/zerolog v1.33.0
)
require (
github.com/mattn/go-colorable v0.1.13 // indirect
github.com/mattn/go-isatty v0.0.19 // indirect
golang.org/x/sys v0.12.0 // indirect
)

@ -0,0 +1,17 @@
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
github.com/go-chi/chi/v5 v5.2.1 h1:KOIHODQj58PmL80G2Eak4WdvUzjSJSm0vG72crDCqb8=
github.com/go-chi/chi/v5 v5.2.1/go.mod h1:L2yAIGWB3H+phAw1NxKwWM+7eUH/lU8pOMm5hHcoops=
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
github.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8=
github.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o=
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=

@ -0,0 +1,28 @@
package cfg
const (
BASE_URL_ENV = "KANPEKI_BASE_URL"
DATABASE_URL_ENV = "KANPEKI_DATABASE_URL"
LISTEN_ADDR_ENV = "KANPEKI_LISTEN_ADDR"
)
type Config struct {
ListenAddr string
BaseUrl string
DatabaseUrl string
LogLevel int
}
func Load(getenv func(string) string) (*Config, error) {
cfg := new(Config)
cfg.BaseUrl = getenv(BASE_URL_ENV)
if cfg.BaseUrl == "" {
cfg.BaseUrl = "http://localhost"
}
cfg.DatabaseUrl = getenv(DATABASE_URL_ENV)
cfg.ListenAddr = getenv(LISTEN_ADDR_ENV)
if cfg.ListenAddr == "" {
cfg.ListenAddr = ":3000"
}
return cfg, nil
}

@ -0,0 +1,14 @@
// package db defines the database interface
package db
import "github.com/gabehf/kanpeki/internal/models"
type DB interface {
Close()
SaveLibrary(*models.LibraryInfo) error
GetLibraryInfo(id string) (*models.LibraryInfo, error)
DeleteLibrary(id string) error
SaveMediaInfo(*models.MediaInfo) error
GetMediaInfo(path string) (*models.MediaInfo, error)
DeleteMediaInfo(path string) error
}

@ -0,0 +1,19 @@
package sqlite
import (
"errors"
"github.com/gabehf/kanpeki/internal/models"
)
func (s *SqliteDB) SaveLibrary(library *models.LibraryInfo) error {
return errors.New("not implemented")
}
func (s *SqliteDB) GetLibraryInfo(id string) (*models.LibraryInfo, error) {
return nil, errors.New("not implemented")
}
func (s *SqliteDB) DeleteLibrary(id string) error {
return errors.New("not implemented")
}

@ -0,0 +1,19 @@
package sqlite
import (
"errors"
"github.com/gabehf/kanpeki/internal/models"
)
func (s *SqliteDB) SaveMediaInfo(MediaInfo *models.MediaInfo) error {
return errors.New("not implemented")
}
func (s *SqliteDB) GetMediaInfo(id string) (*models.MediaInfo, error) {
return nil, errors.New("not implemented")
}
func (s *SqliteDB) DeleteMediaInfo(id string) error {
return errors.New("not implemented")
}

@ -0,0 +1,26 @@
// package sqlite implements the db.DB interface using sqlite3
package sqlite
import (
"database/sql"
"github.com/gabehf/kanpeki/internal/cfg"
)
type SqliteDB struct {
db *sql.DB
}
func New(cfg *cfg.Config) (*SqliteDB, error) {
db, err := sql.Open("sqlite3", "kanpeki.db")
if err != nil {
return &SqliteDB{}, err
}
sqldb := new(SqliteDB)
sqldb.db = db
return sqldb, nil
}
func (s *SqliteDB) Close() {
s.db.Close()
}

@ -0,0 +1,33 @@
package logger
import (
"os"
"sync"
"github.com/gabehf/kanpeki/internal/cfg"
"github.com/rs/zerolog"
)
var once sync.Once
var log zerolog.Logger
func Get(cfg *cfg.Config) zerolog.Logger {
once.Do(func() {
zerolog.TimeFieldFormat = zerolog.TimeFormatUnixMs
logLevel := cfg.LogLevel
log = zerolog.New(os.Stdout).
Level(zerolog.Level(logLevel)).
With().
Timestamp().
Logger()
if os.Getenv("LOG_LEVEL") == "" {
log.Info().Msg("environment variable LOG_LEVEL unset, using default value 0 (debug)")
}
})
return log
}

@ -0,0 +1,16 @@
package models
type LibraryType int
const (
LibraryTypeMovies LibraryType = iota
LibraryTypeTV
)
type LibraryInfo struct {
Id string
Path string
Type LibraryType
Size int64
NumFiles int
}

@ -0,0 +1,7 @@
package models
type MediaInfo struct {
Filename string
Path string
FfprobeData []byte
}

@ -0,0 +1,2 @@
// package models defines the schema of objects in the database
package models
Loading…
Cancel
Save