feat: initial commit

This commit is contained in:
Khairul Hidayat 2024-08-25 00:15:23 +00:00
commit 4532b7a87a
40 changed files with 3796 additions and 0 deletions

51
.air.toml Normal file
View File

@ -0,0 +1,51 @@
root = "."
testdata_dir = "testdata"
tmp_dir = "tmp"
[build]
args_bin = []
bin = "./tmp/main"
cmd = "go build -o ./tmp/main ."
delay = 1000
exclude_dir = ["assets", "tmp", "vendor", "testdata"]
exclude_file = []
exclude_regex = ["_test.go"]
exclude_unchanged = false
follow_symlink = false
full_bin = ""
include_dir = []
include_ext = ["go", "tpl", "tmpl", "html"]
include_file = []
kill_delay = "0s"
log = "build-errors.log"
poll = false
poll_interval = 0
post_cmd = []
pre_cmd = []
rerun = false
rerun_delay = 500
send_interrupt = false
stop_on_error = false
[color]
app = ""
build = "yellow"
main = "magenta"
runner = "green"
watcher = "cyan"
[log]
main_only = false
time = false
[misc]
clean_on_exit = false
[proxy]
app_port = 0
enabled = false
proxy_port = 0
[screen]
clear_on_rebuild = false
keep_scroll = true

5
.gitignore vendored Normal file
View File

@ -0,0 +1,5 @@
out/
tmp/
main
._.DS_Store
.DS_Store

10
Makefile Normal file
View File

@ -0,0 +1,10 @@
all: build-ui backend
build-ui:
cd ui && npm run build
backend:
CGO_ENABLED=0 go build -o main -tags="prod" main.go
clean:
rm -f main && rm -rf ui/dist

22
go.mod Normal file
View File

@ -0,0 +1,22 @@
module rul.sh/go-ytmp3
go 1.23.0
require github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646
require (
github.com/disintegration/imaging v1.6.2 // indirect
github.com/joho/godotenv v1.5.1 // indirect
)
require (
github.com/anthonynsimon/bild v0.14.0
github.com/aws/aws-sdk-go v1.55.5 // indirect
github.com/gosimple/slug v1.14.0 // indirect
github.com/gosimple/unidecode v1.0.1 // indirect
github.com/jmespath/go-jmespath v0.4.0 // indirect
github.com/u2takey/ffmpeg-go v0.5.0 // indirect
github.com/u2takey/go-utils v0.3.1 // indirect
github.com/wader/goutubedl v0.0.0-20240818101919-a623bde37ba9 // indirect
golang.org/x/image v0.19.0 // indirect
)

73
go.sum Normal file
View File

@ -0,0 +1,73 @@
github.com/anthonynsimon/bild v0.14.0 h1:IFRkmKdNdqmexXHfEU7rPlAmdUZ8BDZEGtGHDnGWync=
github.com/anthonynsimon/bild v0.14.0/go.mod h1:hcvEAyBjTW69qkKJTfpcDQ83sSZHxwOunsseDfeQhUs=
github.com/aws/aws-sdk-go v1.38.20/go.mod h1:hcU610XS61/+aQV88ixoOzUoG7v3b31pl2zKMmprdro=
github.com/aws/aws-sdk-go v1.55.5 h1:KKUZBfBoyqy5d3swXyiC7Q76ic40rYcbqH7qjh59kzU=
github.com/aws/aws-sdk-go v1.55.5/go.mod h1:eRwEWoyTWFMVYVQzKMNHWP5/RV4xIUGMQfXQHfHkpNU=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/disintegration/imaging v1.6.2 h1:w1LecBlG2Lnp8B3jk5zSuNqd7b4DXhcjwek1ei82L+c=
github.com/disintegration/imaging v1.6.2/go.mod h1:44/5580QXChDfwIclfc/PCwrr44amcmDAg8hxG0Ewe4=
github.com/fortytw2/leaktest v1.3.0/go.mod h1:jDsjWgpAGjm2CA7WthBh/CdZYEPF31XHquHwclZch5g=
github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
github.com/go-logr/logr v0.1.0/go.mod h1:ixOQHD9gLJUVQQ2ZOR7zLEifBX6tGkNJF4QyIY7sIas=
github.com/gogo/protobuf v1.3.1/go.mod h1:SlYgWuQ5SjCEi6WLHjHCa1yvBfUnHcTbrrZtXPKa29o=
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/gosimple/slug v1.14.0 h1:RtTL/71mJNDfpUbCOmnf/XFkzKRtD6wL6Uy+3akm4Es=
github.com/gosimple/slug v1.14.0/go.mod h1:UiRaFH+GEilHstLUmcBgWcI42viBN7mAb818JrYOeFQ=
github.com/gosimple/unidecode v1.0.1 h1:hZzFTMMqSswvf0LBJZCZgThIZrpDHFXux9KeGmn6T/o=
github.com/gosimple/unidecode v1.0.1/go.mod h1:CP0Cr1Y1kogOtx0bJblKzsVWrqYaqfNOnHzpgWw4Awc=
github.com/hashicorp/golang-lru v0.5.4/go.mod h1:iADmTwqILo4mZ8BN3D2Q6+9jd8WM5uGBxy+E8yxSoD4=
github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg=
github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo=
github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U=
github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
github.com/kisielk/errcheck v1.2.0/go.mod h1:/BMXB+zMLi60iA8Vv6Ksmxu/1UDYcXs4uQLJ+jE2L00=
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646 h1:zYyBkD/k9seD2A7fsi6Oo2LfFZAehjjQMERAvZLEDnQ=
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646/go.mod h1:jpp1/29i3P1S/RLdc7JQKbRpFeM1dOBd8T9ki5s+AY8=
github.com/panjf2000/ants/v2 v2.4.2/go.mod h1:f6F0NZVFsGCp5A7QW/Zj/m92atWwOkY0OIhFxRNFr4A=
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/spf13/afero v1.2.2/go.mod h1:9ZxEEn6pIJ8Rxe320qSDBk6AsU0r9pR7Q4OcevTdifk=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
github.com/u2takey/ffmpeg-go v0.5.0 h1:r7d86XuL7uLWJ5mzSeQ03uvjfIhiJYvsRAJFCW4uklU=
github.com/u2takey/ffmpeg-go v0.5.0/go.mod h1:ruZWkvC1FEiUNjmROowOAps3ZcWxEiOpFoHCvk97kGc=
github.com/u2takey/go-utils v0.3.1 h1:TaQTgmEZZeDHQFYfd+AdUT1cT4QJgJn/XVPELhHw4ys=
github.com/u2takey/go-utils v0.3.1/go.mod h1:6e+v5vEZ/6gu12w/DC2ixZdZtCrNokVxD0JUklcqdCs=
github.com/wader/goutubedl v0.0.0-20240818101919-a623bde37ba9 h1:RJcftQ9fAsNachIVAxtZw1nF2Tu8bIYNjY0ui4Ku1Ts=
github.com/wader/goutubedl v0.0.0-20240818101919-a623bde37ba9/go.mod h1:5KXd5tImdbmz4JoVhePtbIokCwAfEhUVVx3WLHmjYuw=
github.com/wader/osleaktest v0.0.0-20191111175233-f643b0fed071/go.mod h1:XD6emOFPHVzb0+qQpiNOdPL2XZ0SRUM0N5JHuq6OmXo=
gocv.io/x/gocv v0.25.0/go.mod h1:Rar2PS6DV+T4FL+PM535EImD/h13hGVaHhnCu1xarBs=
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=
golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
golang.org/x/image v0.19.0 h1:D9FX4QWkLfkeqaC62SonffIIuYdOk/UE2XKUBgRIBIQ=
golang.org/x/image v0.19.0/go.mod h1:y0zrRqlQRWQ5PXaYCOMLTW2fpsxZ8Qh9I/ohnInJEys=
golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
golang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20200602225109-6fdc65e7d980/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
golang.org/x/time v0.0.0-20190308202827-9d24e82272b4/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
golang.org/x/tools v0.0.0-20181030221726-6c7e314b6563/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
gopkg.in/yaml.v2 v2.2.7/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
sigs.k8s.io/yaml v1.2.0/go.mod h1:yfXDCHCao9+ENCvLSE62v9VSji2MKu5jeNfTrofGhJc=

85
lib/tasks.go Normal file
View File

@ -0,0 +1,85 @@
package lib
import "rul.sh/go-ytmp3/utils"
type Task struct {
Url string `json:"url"`
Slug string `json:"slug"`
Thumbnail string `json:"thumbnail"`
Title string `json:"title"`
Artist string `json:"artist"`
Album string `json:"album"`
IsPending bool `json:"is_pending"`
Result string `json:"result"`
Error error `json:"error"`
}
var tasks = []*Task{}
var queue = []*Task{}
func NewTask(task Task) *Task {
task.IsPending = true
queue = append(queue, &task)
tasks = append(tasks, &task)
if len(tasks) > 20 {
tasks = tasks[1:]
}
return &task
}
func GetTasks() []*Task {
return tasks
}
type TaskScheduler struct {
Ch chan bool
}
func InitTaskScheduler() *TaskScheduler {
ch := make(chan bool)
outDir := utils.GetEnv("OUT_DIR", "/tmp")
go func() {
for {
select {
case <-ch:
return
default:
if len(queue) == 0 {
continue
}
task := queue[0]
queue = queue[1:]
video, err := YtGetVideo(task.Url)
if err != nil {
task.Error = err
task.IsPending = false
continue
}
result, err := Yt2Mp3(&video, Yt2Mp3Options{
OutDir: outDir,
Slug: task.Slug,
Thumbnail: task.Thumbnail,
Title: task.Title,
Artist: task.Artist,
Album: task.Album,
})
task.IsPending = false
task.Error = err
task.Result = result
}
}
}()
return &TaskScheduler{Ch: ch}
}
func (s *TaskScheduler) Stop() {
s.Ch <- true
}

213
lib/yt2mp3.go Normal file
View File

@ -0,0 +1,213 @@
package lib
import (
"context"
"fmt"
"image"
"image/jpeg"
"io"
"net/http"
"os"
"strings"
"github.com/disintegration/imaging"
"github.com/gosimple/slug"
ffmpeg "github.com/u2takey/ffmpeg-go"
"github.com/wader/goutubedl"
"golang.org/x/image/webp"
"rul.sh/go-ytmp3/utils"
)
func fetchVideo(video *goutubedl.Result, out string, ch chan error) {
dl, err := video.Download(context.Background(), "best")
if err != nil {
ch <- err
return
}
defer dl.Close()
f, err := os.Create(out)
if err != nil {
ch <- err
return
}
defer f.Close()
fmt.Println("Downloading...")
if _, err := io.Copy(f, dl); err != nil {
ch <- err
return
}
ch <- nil
}
func resizeImage(imgName string, src io.Reader, dst *os.File, width int, height int) error {
var img image.Image
var err error
ext := imgName[strings.LastIndex(imgName, "."):]
switch ext {
case "webp":
img, err = webp.Decode(src)
default:
img, _, err = image.Decode(src)
}
if err != nil {
fmt.Println(err)
return err
}
img = imaging.Fill(img, width, height, imaging.Center, imaging.Lanczos)
return jpeg.Encode(dst, img, nil)
}
func fetchThumbnail(video *goutubedl.Result, thumbnail string, ch chan error) {
if video.Info.Thumbnail == "" {
ch <- fmt.Errorf("no thumbnail found")
return
}
fmt.Println("Downloading thumbnail...")
f, err := os.Create(thumbnail)
if err != nil {
ch <- err
return
}
defer f.Close()
resp, err := http.Get(video.Info.Thumbnail)
if err != nil {
ch <- err
return
}
defer resp.Body.Close()
if err := resizeImage(video.Info.Thumbnail, resp.Body, f, 512, 512); err != nil {
ch <- err
return
}
ch <- nil
}
type ConvertOptions struct {
Video string
Thumbnail string
Title string
Artist string
Album string
Output string
}
func convertToMp3(data ConvertOptions, ch chan error) {
fmt.Println("Converting...")
input := []*ffmpeg.Stream{ffmpeg.Input(data.Video).Audio()}
args := ffmpeg.KwArgs{
"format": "mp3",
"id3v2_version": "3",
"write_id3v1": "1",
"metadata": []string{
fmt.Sprintf("title=%s", data.Title),
fmt.Sprintf("artist=%s", data.Artist),
fmt.Sprintf("album=%s", data.Album),
},
}
if data.Thumbnail != "" {
input = append(input, ffmpeg.Input(data.Thumbnail).Video())
}
if err := ffmpeg.Output(input, data.Output, args).OverWriteOutput().Run(); err != nil {
ch <- err
return
}
ch <- nil
}
func YtGetVideo(url string) (goutubedl.Result, error) {
return goutubedl.New(context.Background(), url, goutubedl.Options{})
}
type Yt2Mp3Options struct {
OutDir string
Slug string
Thumbnail string
Title string
Artist string
Album string
}
func Yt2Mp3(video *goutubedl.Result, options Yt2Mp3Options) (string, error) {
if video == nil {
return "", fmt.Errorf("no video found")
}
tmpDir := utils.GetEnv("TMP_DIR", "/tmp")
title := video.Info.Title
artist := video.Info.Channel
album := video.Info.Album
if video.Info.Artist != "" {
artist = video.Info.Artist
}
videoSlug := options.Slug
if len(options.Slug) == 0 {
videoSlug = slug.Make(title)
}
videoSrc := fmt.Sprintf("%s/%s.mp4", tmpDir, videoSlug)
thumbnail := fmt.Sprintf("%s/%s.jpg", tmpDir, videoSlug)
out := fmt.Sprintf("%s/%s.mp3", options.OutDir, videoSlug)
if err := os.MkdirAll(options.OutDir, os.ModePerm); err != nil {
return "", err
}
videoCh := make(chan error)
thumbCh := make(chan error)
go fetchVideo(video, videoSrc, videoCh)
go fetchThumbnail(video, thumbnail, thumbCh)
err := <-videoCh
if err != nil {
return "", err
}
err = <-thumbCh
if err != nil {
thumbnail = ""
}
fmt.Println(artist, album)
convertCh := make(chan error)
go convertToMp3(ConvertOptions{
Video: videoSrc,
Thumbnail: thumbnail,
Title: title,
Artist: artist,
Album: album,
Output: out,
}, convertCh)
err = <-convertCh
if err != nil {
return "", err
}
return out, nil
}

144
main.go Normal file
View File

@ -0,0 +1,144 @@
package main
import (
"encoding/json"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strings"
"github.com/gosimple/slug"
"github.com/joho/godotenv"
"rul.sh/go-ytmp3/lib"
"rul.sh/go-ytmp3/types"
"rul.sh/go-ytmp3/ui"
"rul.sh/go-ytmp3/utils"
)
func main() {
godotenv.Load()
outDir := utils.GetEnv("OUT_DIR", "/tmp")
app := http.NewServeMux()
app.HandleFunc("GET /api/info/", func(w http.ResponseWriter, r *http.Request) {
url := r.URL.Query().Get("url")
if len(url) == 0 {
http.Error(w, "No video url provided", http.StatusBadRequest)
return
}
video, err := lib.YtGetVideo(url)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
data := types.GetVideoInfoRes{
Url: url,
Slug: slug.Make(video.Info.Title),
Thumbnail: video.Info.Thumbnail,
Title: video.Info.Title,
Artist: video.Info.Channel,
Album: video.Info.Album,
}
if video.Info.Artist != "" {
data.Artist = video.Info.Artist
}
json, err := json.Marshal(data)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(json)
})
app.HandleFunc("POST /api/tasks/", func(w http.ResponseWriter, r *http.Request) {
var data types.CreateTaskBody
err := json.NewDecoder(r.Body).Decode(&data)
if err != nil {
http.Error(w, err.Error(), http.StatusBadRequest)
return
}
if len(data.Url) == 0 {
http.Error(w, "No video url provided", http.StatusBadRequest)
return
}
task := lib.NewTask(lib.Task{
Url: data.Url,
Slug: data.Slug,
Thumbnail: data.Thumbnail,
Title: data.Title,
Artist: data.Artist,
Album: data.Album,
})
json, err := json.Marshal(task)
if err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
w.Header().Set("Content-Type", "application/json")
w.Write(json)
})
app.HandleFunc("GET /api/tasks/", func(w http.ResponseWriter, r *http.Request) {
json, err := json.Marshal(lib.GetTasks())
if err != nil {
w.WriteHeader(http.StatusInternalServerError)
return
}
w.Write(json)
})
app.HandleFunc("GET /api/get/", func(w http.ResponseWriter, r *http.Request) {
filename := r.URL.Path[strings.LastIndex(r.URL.Path, "/")+1:]
isDownload := r.URL.Query().Get("dl") == "true"
if len(filename) == 0 {
http.Error(w, "No filename provided", http.StatusBadRequest)
return
}
file, err := os.Open(filepath.Join(outDir, filename))
if err != nil {
http.Error(w, err.Error(), http.StatusNotFound)
return
}
defer file.Close()
w.Header().Set("Content-Type", "audio/mpeg")
if isDownload {
w.Header().Set("Content-Disposition", fmt.Sprintf("attachment; filename=%s", filename))
}
if _, err := io.Copy(w, file); err != nil {
http.Error(w, err.Error(), http.StatusInternalServerError)
return
}
})
ui.ServeUI(app)
scheduler := lib.InitTaskScheduler()
defer scheduler.Stop()
port := utils.GetEnv("PORT", "8080")
fmt.Printf("Listening on http://localhost:%s\n", port)
if err := http.ListenAndServe(fmt.Sprintf(":%s", port), app); err != nil {
panic(err)
}
}

17
rest.http Normal file
View File

@ -0,0 +1,17 @@
GET http://localhost:8080/tasks/ HTTP/1.1
Accept: : application/json
###
GET http://localhost:8080/info?url=https://www.youtube.com/watch?v=lCok_iWOIXw HTTP/1.1
Accept: : application/json
###
POST http://localhost:8080/tasks/ HTTP/1.1
Content-Type: application/json
{
"url": "https://www.youtube.com/watch?v=lCok_iWOIXw"
}

19
types/api.go Normal file
View File

@ -0,0 +1,19 @@
package types
type GetVideoInfoRes struct {
Url string `json:"url"`
Slug string `json:"slug"`
Thumbnail string `json:"thumbnail"`
Title string `json:"title"`
Artist string `json:"artist"`
Album string `json:"album"`
}
type CreateTaskBody struct {
Url string `json:"url"`
Slug string `json:"slug"`
Thumbnail string `json:"thumbnail"`
Title string `json:"title"`
Artist string `json:"artist"`
Album string `json:"album"`
}

24
ui/.gitignore vendored Normal file
View File

@ -0,0 +1,24 @@
# Logs
logs
*.log
npm-debug.log*
yarn-debug.log*
yarn-error.log*
pnpm-debug.log*
lerna-debug.log*
node_modules
dist
dist-ssr
*.local
# Editor directories and files
.vscode/*
!.vscode/extensions.json
.idea
.DS_Store
*.suo
*.ntvs*
*.njsproj
*.sln
*.sw?

50
ui/README.md Normal file
View File

@ -0,0 +1,50 @@
# React + TypeScript + Vite
This template provides a minimal setup to get React working in Vite with HMR and some ESLint rules.
Currently, two official plugins are available:
- [@vitejs/plugin-react](https://github.com/vitejs/vite-plugin-react/blob/main/packages/plugin-react/README.md) uses [Babel](https://babeljs.io/) for Fast Refresh
- [@vitejs/plugin-react-swc](https://github.com/vitejs/vite-plugin-react-swc) uses [SWC](https://swc.rs/) for Fast Refresh
## Expanding the ESLint configuration
If you are developing a production application, we recommend updating the configuration to enable type aware lint rules:
- Configure the top-level `parserOptions` property like this:
```js
export default tseslint.config({
languageOptions: {
// other options...
parserOptions: {
project: ['./tsconfig.node.json', './tsconfig.app.json'],
tsconfigRootDir: import.meta.dirname,
},
},
})
```
- Replace `tseslint.configs.recommended` to `tseslint.configs.recommendedTypeChecked` or `tseslint.configs.strictTypeChecked`
- Optionally add `...tseslint.configs.stylisticTypeChecked`
- Install [eslint-plugin-react](https://github.com/jsx-eslint/eslint-plugin-react) and update the config:
```js
// eslint.config.js
import react from 'eslint-plugin-react'
export default tseslint.config({
// Set the react version
settings: { react: { version: '18.3' } },
plugins: {
// Add the react plugin
react,
},
rules: {
// other rules...
// Enable its recommended rules
...react.configs.recommended.rules,
...react.configs['jsx-runtime'].rules,
},
})
```

28
ui/eslint.config.js Normal file
View File

@ -0,0 +1,28 @@
import js from '@eslint/js'
import globals from 'globals'
import reactHooks from 'eslint-plugin-react-hooks'
import reactRefresh from 'eslint-plugin-react-refresh'
import tseslint from 'typescript-eslint'
export default tseslint.config(
{ ignores: ['dist'] },
{
extends: [js.configs.recommended, ...tseslint.configs.recommended],
files: ['**/*.{ts,tsx}'],
languageOptions: {
ecmaVersion: 2020,
globals: globals.browser,
},
plugins: {
'react-hooks': reactHooks,
'react-refresh': reactRefresh,
},
rules: {
...reactHooks.configs.recommended.rules,
'react-refresh/only-export-components': [
'warn',
{ allowConstantExport: true },
],
},
},
)

17
ui/index.html Normal file
View File

@ -0,0 +1,17 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<link rel="apple-touch-icon" sizes="180x180" href="/apple-touch-icon.png" />
<link rel="icon" type="image/png" sizes="32x32" href="/favicon-32x32.png" />
<link rel="icon" type="image/png" sizes="16x16" href="/favicon-16x16.png" />
<link rel="manifest" href="/site.webmanifest" />
<title>YouTube To MP3</title>
</head>
<body>
<div id="root"></div>
<script type="module" src="/src/main.tsx"></script>
</body>
</html>

35
ui/package.json Normal file
View File

@ -0,0 +1,35 @@
{
"name": "ui",
"private": true,
"version": "0.0.0",
"type": "module",
"scripts": {
"dev": "vite",
"build": "tsc -b && vite build",
"lint": "eslint .",
"preview": "vite preview"
},
"dependencies": {
"lucide-react": "^0.435.0",
"react": "^18.3.1",
"react-dom": "^18.3.1",
"slugify": "^1.6.6"
},
"devDependencies": {
"@eslint/js": "^9.9.0",
"@types/react": "^18.3.3",
"@types/react-dom": "^18.3.0",
"@vitejs/plugin-react-swc": "^3.5.0",
"autoprefixer": "^10.4.20",
"daisyui": "^4.12.10",
"eslint": "^9.9.0",
"eslint-plugin-react-hooks": "^5.1.0-rc.0",
"eslint-plugin-react-refresh": "^0.4.9",
"globals": "^15.9.0",
"postcss": "^8.4.41",
"tailwindcss": "^3.4.10",
"typescript": "^5.5.3",
"typescript-eslint": "^8.0.1",
"vite": "^5.4.1"
}
}

2433
ui/pnpm-lock.yaml generated Normal file

File diff suppressed because it is too large Load Diff

6
ui/postcss.config.js Normal file
View File

@ -0,0 +1,6 @@
export default {
plugins: {
tailwindcss: {},
autoprefixer: {},
},
}

Binary file not shown.

After

Width:  |  Height:  |  Size: 9.5 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

BIN
ui/public/apple-touch-icon.png Executable file

Binary file not shown.

After

Width:  |  Height:  |  Size: 8.6 KiB

BIN
ui/public/favicon-16x16.png Executable file

Binary file not shown.

After

Width:  |  Height:  |  Size: 483 B

BIN
ui/public/favicon-32x32.png Executable file

Binary file not shown.

After

Width:  |  Height:  |  Size: 1.1 KiB

BIN
ui/public/favicon.ico Executable file

Binary file not shown.

After

Width:  |  Height:  |  Size: 15 KiB

1
ui/public/site.webmanifest Executable file
View File

@ -0,0 +1 @@
{"name":"","short_name":"","icons":[{"src":"/android-chrome-192x192.png","sizes":"192x192","type":"image/png"},{"src":"/android-chrome-512x512.png","sizes":"512x512","type":"image/png"}],"theme_color":"#ffffff","background_color":"#ffffff","display":"standalone"}

1
ui/public/vite.svg Normal file
View File

@ -0,0 +1 @@
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="31.88" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 257"><defs><linearGradient id="IconifyId1813088fe1fbc01fb466" x1="-.828%" x2="57.636%" y1="7.652%" y2="78.411%"><stop offset="0%" stop-color="#41D1FF"></stop><stop offset="100%" stop-color="#BD34FE"></stop></linearGradient><linearGradient id="IconifyId1813088fe1fbc01fb467" x1="43.376%" x2="50.316%" y1="2.242%" y2="89.03%"><stop offset="0%" stop-color="#FFEA83"></stop><stop offset="8.333%" stop-color="#FFDD35"></stop><stop offset="100%" stop-color="#FFA800"></stop></linearGradient></defs><path fill="url(#IconifyId1813088fe1fbc01fb466)" d="M255.153 37.938L134.897 252.976c-2.483 4.44-8.862 4.466-11.382.048L.875 37.958c-2.746-4.814 1.371-10.646 6.827-9.67l120.385 21.517a6.537 6.537 0 0 0 2.322-.004l117.867-21.483c5.438-.991 9.574 4.796 6.877 9.62Z"></path><path fill="url(#IconifyId1813088fe1fbc01fb467)" d="M185.432.063L96.44 17.501a3.268 3.268 0 0 0-2.634 3.014l-5.474 92.456a3.268 3.268 0 0 0 3.997 3.378l24.777-5.718c2.318-.535 4.413 1.507 3.936 3.838l-7.361 36.047c-.495 2.426 1.782 4.5 4.151 3.78l15.304-4.649c2.372-.72 4.652 1.36 4.15 3.788l-11.698 56.621c-.732 3.542 3.979 5.473 5.943 2.437l1.313-2.028l72.516-144.72c1.215-2.423-.88-5.186-3.54-4.672l-25.505 4.922c-2.396.462-4.435-1.77-3.759-4.114l16.646-57.705c.677-2.35-1.37-4.583-3.769-4.113Z"></path></svg>

After

Width:  |  Height:  |  Size: 1.5 KiB

347
ui/src/App.tsx Normal file
View File

@ -0,0 +1,347 @@
import { DownloadIcon, LinkIcon, Loader2, SearchIcon } from "lucide-react";
import { useEffect, useMemo, useRef, useState } from "react";
import { useFetch } from "./hooks/useFetch";
import { API_BASE_URL, fetchAPI } from "./api";
import slugify from "slugify";
const App = () => {
const formRef = useRef<HTMLFormElement | null>(null);
const [errors, setErrors] = useState<Record<string, string>>({});
const [video, setVideo] = useState<any>(null);
const [isLoading, setLoading] = useState(false);
const tasks = useFetch("tasks", () => fetchAPI("/tasks/"));
const onFindVideo = async () => {
if (!formRef.current || isLoading) {
return;
}
setErrors({});
setVideo(null);
setLoading(true);
const formData = new FormData(formRef.current);
const url = (formData.get("url") as string).trim();
if (!url?.length || !url.startsWith("http")) {
setErrors({ url: "URL is invalid" });
setLoading(false);
return;
}
try {
const data = await fetchAPI("/info?url=" + encodeURI(url));
setVideo(data);
} catch (err) {
setErrors({ url: (err as Error)?.message || "Unknown error" });
}
setLoading(false);
};
const onSubmit = async (e: React.FormEvent<HTMLFormElement>) => {
e.preventDefault();
if (isLoading) {
return;
}
setErrors({});
if (!video?.url) {
setErrors({ url: "URL is invalid" });
return;
}
const formData = new FormData(e.target as HTMLFormElement);
const artist = (formData.get("artist") as string).trim();
const slug = (formData.get("slug") as string).trim();
const album = (formData.get("album") as string).trim();
const title = (formData.get("title") as string).trim();
const data = {
url: video.url,
thumbnail: video.thumbnail,
title,
slug,
artist,
album,
};
try {
await fetchAPI("/tasks/", {
method: "POST",
body: JSON.stringify(data),
headers: {
"Content-Type": "application/json",
},
});
// reset form
formRef.current?.reset();
setVideo(null);
tasks.refetch();
} catch (err) {
setErrors({ result: (err as Error)?.message || "Unknown error" });
}
};
useEffect(() => {
const hasPendingTask = tasks.data?.find((task: any) => task.is_pending);
if (hasPendingTask) {
const timeout = setTimeout(() => tasks.refetch(), 1000);
return () => clearTimeout(timeout);
}
}, [tasks.data]);
return (
<div className="min-h-screen flex flex-col items-center justify-center p-4 bg-base-300">
<div className="card bg-base-100 w-full max-w-2xl">
<div className="card-body p-4 md:p-8">
<p className="card-title font-normal text-2xl self-center text-center">
YouTube To MP3
</p>
<p className="self-center text-center">
Download and convert YouTube videos to MP3
</p>
<form ref={formRef} onSubmit={onSubmit}>
<label className="input input-bordered flex items-center gap-2 md:gap-3 pl-2 md:pl-4 pr-0 mt-4 overflow-hidden">
<LinkIcon className="shrink-0" size={20} />
<input
className="grow"
name="url"
placeholder="Enter Video URL"
required
onKeyDown={(e) => {
if (e.key === "Enter") {
onFindVideo();
e.preventDefault();
}
}}
/>
<button
type="button"
className="btn btn-primary btn-square shrink-0"
onClick={onFindVideo}
disabled={isLoading}
>
{isLoading ? (
<Loader2 className="animate-spin" />
) : (
<SearchIcon />
)}
</button>
</label>
{errors.url && (
<p className="text-error text-sm mt-1">{errors.url}</p>
)}
{video != null && (
<div className="mt-4 md:mt-8">
<img
src={video.thumbnail}
alt="thumbnail"
className="w-full md:max-w-[50%] rounded-box mx-auto"
/>
<div className="mt-4 grid md:grid-cols-2 gap-2 md:gap-4">
<label className="form-control w-full max-w-xs">
<div className="label pt-0 pb-0.5">
<span className="label-text">Title</span>
</div>
<input
type="text"
name="title"
placeholder="Enter Title"
className="input input-bordered w-full"
defaultValue={video.title}
required
onChange={(e) => {
const slugEl = formRef.current?.querySelector(
'input[name="slug"]'
) as HTMLInputElement | null;
if (slugEl) {
slugEl.value = slugify(e.target.value).toLowerCase();
}
}}
/>
</label>
<label className="form-control w-full max-w-xs">
<div className="label pt-0 pb-0.5">
<span className="label-text">Slug</span>
</div>
<input
type="text"
name="slug"
placeholder="Enter Slug"
className="input input-bordered w-full"
defaultValue={video.slug}
required
/>
</label>
<label className="form-control w-full max-w-xs">
<div className="label pt-0 pb-0.5">
<span className="label-text">Artist</span>
</div>
<input
type="text"
name="artist"
placeholder="Enter Artist"
className="input input-bordered w-full"
defaultValue={video.artist}
/>
</label>
<label className="form-control w-full max-w-xs">
<div className="label pt-0 pb-0.5">
<span className="label-text">Album</span>
</div>
<input
type="text"
name="album"
placeholder="Enter Album"
className="input input-bordered w-full"
defaultValue={video.album}
/>
</label>
<button
type="submit"
className="btn btn-primary mt-4 w-full max-w-xs"
disabled={isLoading}
>
{isLoading ? (
<Loader2 className="animate-spin" />
) : (
<DownloadIcon />
)}
Download
</button>
</div>
</div>
)}
</form>
</div>
<TaskList data={tasks.data} />
</div>
</div>
);
};
type TaskListProps = {
data?: any[];
};
const TaskList = ({ data }: TaskListProps) => {
const items = useMemo(() => {
return data?.reverse() ?? [];
}, [data]);
if (!items?.length) {
return null;
}
return (
<div className="overflow-x-auto p-4 pt-0">
<table className="table w-full">
<thead>
<tr>
<th>Title</th>
<th>Artist</th>
<th>Album</th>
<th>Status</th>
</tr>
</thead>
<tbody>
{items.map((task, idx) => {
let downloadUrl = "";
if (!task.is_pending && task.result) {
const filename = task.result.split("/").pop();
downloadUrl = API_BASE_URL + "/get/" + filename;
}
return (
<tr key={task.slug + idx}>
<td>
<div className="flex flex-row items-center gap-2">
<img
src={task.thumbnail}
alt="thumbnail"
className="w-8 h-8 object-cover"
/>
{downloadUrl ? (
<a
href={downloadUrl}
target="_blank"
className="link truncate max-w-[200px]"
title={task.title}
>
{task.title}
</a>
) : (
<span
className="truncate max-w-[200px]"
title={task.title}
>
{task.title}
</span>
)}
</div>
</td>
<td>
<span
className="inline-block truncate max-w-[160px]"
title={task.artist}
>
{task.artist || "-"}
</span>
</td>
<td>{task.album || "-"}</td>
<td>
<div className="flex flex-row items-center gap-2">
<TaskStatus
isPending={task.is_pending}
error={task.error}
/>
{downloadUrl ? (
<a
href={downloadUrl + "?dl=true"}
className="btn btn-ghost btn-square"
>
<DownloadIcon size={20} />
</a>
) : null}
</div>
</td>
</tr>
);
})}
</tbody>
</table>
</div>
);
};
type TaskStatusProps = {
isPending: boolean;
error?: string | null;
};
const TaskStatus = ({ isPending, error }: TaskStatusProps) => {
if (isPending) {
return <Loader2 className="animate-spin" />;
}
return error ? (
<p className="text-error text-sm" title={error}>
Error!
</p>
) : (
<p className="text-success text-sm">Done</p>
);
};
export default App;

18
ui/src/api.ts Normal file
View File

@ -0,0 +1,18 @@
//
export const API_BASE_URL = "/api";
export const fetchAPI = async <T = any>(
url: string,
init: RequestInit = {}
) => {
const res = await fetch(API_BASE_URL + url, init);
if (!res.ok) {
const data = await res.json().catch(() => null);
const message = data.message ?? res.statusText;
throw new Error(message);
}
return (await res.json()) as T;
};

1
ui/src/assets/react.svg Normal file
View File

@ -0,0 +1 @@
<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="35.93" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 228"><path fill="#00D8FF" d="M210.483 73.824a171.49 171.49 0 0 0-8.24-2.597c.465-1.9.893-3.777 1.273-5.621c6.238-30.281 2.16-54.676-11.769-62.708c-13.355-7.7-35.196.329-57.254 19.526a171.23 171.23 0 0 0-6.375 5.848a155.866 155.866 0 0 0-4.241-3.917C100.759 3.829 77.587-4.822 63.673 3.233C50.33 10.957 46.379 33.89 51.995 62.588a170.974 170.974 0 0 0 1.892 8.48c-3.28.932-6.445 1.924-9.474 2.98C17.309 83.498 0 98.307 0 113.668c0 15.865 18.582 31.778 46.812 41.427a145.52 145.52 0 0 0 6.921 2.165a167.467 167.467 0 0 0-2.01 9.138c-5.354 28.2-1.173 50.591 12.134 58.266c13.744 7.926 36.812-.22 59.273-19.855a145.567 145.567 0 0 0 5.342-4.923a168.064 168.064 0 0 0 6.92 6.314c21.758 18.722 43.246 26.282 56.54 18.586c13.731-7.949 18.194-32.003 12.4-61.268a145.016 145.016 0 0 0-1.535-6.842c1.62-.48 3.21-.974 4.76-1.488c29.348-9.723 48.443-25.443 48.443-41.52c0-15.417-17.868-30.326-45.517-39.844Zm-6.365 70.984c-1.4.463-2.836.91-4.3 1.345c-3.24-10.257-7.612-21.163-12.963-32.432c5.106-11 9.31-21.767 12.459-31.957c2.619.758 5.16 1.557 7.61 2.4c23.69 8.156 38.14 20.213 38.14 29.504c0 9.896-15.606 22.743-40.946 31.14Zm-10.514 20.834c2.562 12.94 2.927 24.64 1.23 33.787c-1.524 8.219-4.59 13.698-8.382 15.893c-8.067 4.67-25.32-1.4-43.927-17.412a156.726 156.726 0 0 1-6.437-5.87c7.214-7.889 14.423-17.06 21.459-27.246c12.376-1.098 24.068-2.894 34.671-5.345a134.17 134.17 0 0 1 1.386 6.193ZM87.276 214.515c-7.882 2.783-14.16 2.863-17.955.675c-8.075-4.657-11.432-22.636-6.853-46.752a156.923 156.923 0 0 1 1.869-8.499c10.486 2.32 22.093 3.988 34.498 4.994c7.084 9.967 14.501 19.128 21.976 27.15a134.668 134.668 0 0 1-4.877 4.492c-9.933 8.682-19.886 14.842-28.658 17.94ZM50.35 144.747c-12.483-4.267-22.792-9.812-29.858-15.863c-6.35-5.437-9.555-10.836-9.555-15.216c0-9.322 13.897-21.212 37.076-29.293c2.813-.98 5.757-1.905 8.812-2.773c3.204 10.42 7.406 21.315 12.477 32.332c-5.137 11.18-9.399 22.249-12.634 32.792a134.718 134.718 0 0 1-6.318-1.979Zm12.378-84.26c-4.811-24.587-1.616-43.134 6.425-47.789c8.564-4.958 27.502 2.111 47.463 19.835a144.318 144.318 0 0 1 3.841 3.545c-7.438 7.987-14.787 17.08-21.808 26.988c-12.04 1.116-23.565 2.908-34.161 5.309a160.342 160.342 0 0 1-1.76-7.887Zm110.427 27.268a347.8 347.8 0 0 0-7.785-12.803c8.168 1.033 15.994 2.404 23.343 4.08c-2.206 7.072-4.956 14.465-8.193 22.045a381.151 381.151 0 0 0-7.365-13.322Zm-45.032-43.861c5.044 5.465 10.096 11.566 15.065 18.186a322.04 322.04 0 0 0-30.257-.006c4.974-6.559 10.069-12.652 15.192-18.18ZM82.802 87.83a323.167 323.167 0 0 0-7.227 13.238c-3.184-7.553-5.909-14.98-8.134-22.152c7.304-1.634 15.093-2.97 23.209-3.984a321.524 321.524 0 0 0-7.848 12.897Zm8.081 65.352c-8.385-.936-16.291-2.203-23.593-3.793c2.26-7.3 5.045-14.885 8.298-22.6a321.187 321.187 0 0 0 7.257 13.246c2.594 4.48 5.28 8.868 8.038 13.147Zm37.542 31.03c-5.184-5.592-10.354-11.779-15.403-18.433c4.902.192 9.899.29 14.978.29c5.218 0 10.376-.117 15.453-.343c-4.985 6.774-10.018 12.97-15.028 18.486Zm52.198-57.817c3.422 7.8 6.306 15.345 8.596 22.52c-7.422 1.694-15.436 3.058-23.88 4.071a382.417 382.417 0 0 0 7.859-13.026a347.403 347.403 0 0 0 7.425-13.565Zm-16.898 8.101a358.557 358.557 0 0 1-12.281 19.815a329.4 329.4 0 0 1-23.444.823c-7.967 0-15.716-.248-23.178-.732a310.202 310.202 0 0 1-12.513-19.846h.001a307.41 307.41 0 0 1-10.923-20.627a310.278 310.278 0 0 1 10.89-20.637l-.001.001a307.318 307.318 0 0 1 12.413-19.761c7.613-.576 15.42-.876 23.31-.876H128c7.926 0 15.743.303 23.354.883a329.357 329.357 0 0 1 12.335 19.695a358.489 358.489 0 0 1 11.036 20.54a329.472 329.472 0 0 1-11 20.722Zm22.56-122.124c8.572 4.944 11.906 24.881 6.52 51.026c-.344 1.668-.73 3.367-1.15 5.09c-10.622-2.452-22.155-4.275-34.23-5.408c-7.034-10.017-14.323-19.124-21.64-27.008a160.789 160.789 0 0 1 5.888-5.4c18.9-16.447 36.564-22.941 44.612-18.3ZM128 90.808c12.625 0 22.86 10.235 22.86 22.86s-10.235 22.86-22.86 22.86s-22.86-10.235-22.86-22.86s10.235-22.86 22.86-22.86Z"></path></svg>

After

Width:  |  Height:  |  Size: 4.0 KiB

48
ui/src/hooks/useFetch.ts Normal file
View File

@ -0,0 +1,48 @@
import { useCallback, useEffect, useRef, useState } from "react";
const cacheStore = new Map<string, any>();
type UseFetchOptions = {
enabled: boolean;
};
export const useFetch = <T = any>(
fetchKey: any,
fetchFn: () => Promise<T>,
options?: Partial<UseFetchOptions>
) => {
const key = JSON.stringify(fetchKey);
const loadingRef = useRef(false);
const [isLoading, setIsLoading] = useState(false);
const [data, setData] = useState<T | undefined>(cacheStore.get(key));
const [error, setError] = useState<Error | undefined>();
const fetchData = useCallback(async () => {
if (loadingRef.current) {
return;
}
try {
loadingRef.current = true;
setIsLoading(true);
const res = await fetchFn();
setData(res);
cacheStore.set(key, res);
} catch (err) {
setError(err instanceof Error ? err : new Error("Unknown error"));
setData(undefined);
} finally {
loadingRef.current = false;
setIsLoading(false);
}
}, [key]);
useEffect(() => {
if (options?.enabled !== false) {
fetchData();
}
}, [fetchData, options?.enabled]);
return { data, isLoading, error, refetch: fetchData };
};

3
ui/src/index.css Normal file
View File

@ -0,0 +1,3 @@
@tailwind base;
@tailwind components;
@tailwind utilities;

10
ui/src/main.tsx Normal file
View File

@ -0,0 +1,10 @@
import { StrictMode } from "react";
import { createRoot } from "react-dom/client";
import App from "./App.tsx";
import "./index.css";
createRoot(document.getElementById("root")!).render(
<StrictMode>
<App />
</StrictMode>
);

1
ui/src/vite-env.d.ts vendored Normal file
View File

@ -0,0 +1 @@
/// <reference types="vite/client" />

15
ui/tailwind.config.js Normal file
View File

@ -0,0 +1,15 @@
/** @type {import('tailwindcss').Config} */
export default {
content: [
"./index.html",
"./src/**/*.{js,ts,jsx,tsx}",
],
theme: {
extend: {},
},
plugins: [require('daisyui')],
daisyui: {
themes: ['dracula'],
},
}

24
ui/tsconfig.app.json Normal file
View File

@ -0,0 +1,24 @@
{
"compilerOptions": {
"target": "ES2020",
"useDefineForClassFields": true,
"lib": ["ES2020", "DOM", "DOM.Iterable"],
"module": "ESNext",
"skipLibCheck": true,
/* Bundler mode */
"moduleResolution": "bundler",
"allowImportingTsExtensions": true,
"isolatedModules": true,
"moduleDetection": "force",
"noEmit": true,
"jsx": "react-jsx",
/* Linting */
"strict": true,
"noUnusedLocals": true,
"noUnusedParameters": true,
"noFallthroughCasesInSwitch": true
},
"include": ["src"]
}

7
ui/tsconfig.json Normal file
View File

@ -0,0 +1,7 @@
{
"files": [],
"references": [
{ "path": "./tsconfig.app.json" },
{ "path": "./tsconfig.node.json" }
]
}

22
ui/tsconfig.node.json Normal file
View File

@ -0,0 +1,22 @@
{
"compilerOptions": {
"target": "ES2022",
"lib": ["ES2023"],
"module": "ESNext",
"skipLibCheck": true,
/* Bundler mode */
"moduleResolution": "bundler",
"allowImportingTsExtensions": true,
"isolatedModules": true,
"moduleDetection": "force",
"noEmit": true,
/* Linting */
"strict": true,
"noUnusedLocals": true,
"noUnusedParameters": true,
"noFallthroughCasesInSwitch": true
},
"include": ["vite.config.ts"]
}

10
ui/ui.go Normal file
View File

@ -0,0 +1,10 @@
//go:build !prod
// +build !prod
package ui
import "net/http"
func ServeUI(app *http.ServeMux) {
//
}

34
ui/ui_prod.go Normal file
View File

@ -0,0 +1,34 @@
//go:build prod
// +build prod
package ui
import (
"embed"
"io/fs"
"net/http"
"path"
)
//go:embed dist
var embeddedFs embed.FS
func ServeUI(app *http.ServeMux) {
distFs, _ := fs.Sub(embeddedFs, "dist")
fileServer := http.FileServer(http.FS(distFs))
app.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
_path := path.Clean(r.URL.Path)[1:]
// Rewrite non-existing paths to index.html
if _, err := fs.Stat(distFs, _path); err != nil {
index, _ := fs.ReadFile(distFs, "index.html")
w.Header().Add("Content-Type", "text/html")
w.WriteHeader(http.StatusOK)
w.Write(index)
return
}
fileServer.ServeHTTP(w, r)
}))
}

12
ui/vite.config.ts Normal file
View File

@ -0,0 +1,12 @@
import { defineConfig } from 'vite'
import react from '@vitejs/plugin-react-swc'
// https://vitejs.dev/config/
export default defineConfig({
plugins: [react()],
server: {
proxy: {
'/api': 'http://localhost:8080'
}
}
})

10
utils/utils.go Normal file
View File

@ -0,0 +1,10 @@
package utils
import "os"
func GetEnv(key string, fallback string) string {
if value, ok := os.LookupEnv(key); ok {
return value
}
return fallback
}