feat: initial commit
51
.air.toml
Normal 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
@ -0,0 +1,5 @@
|
||||
out/
|
||||
tmp/
|
||||
main
|
||||
._.DS_Store
|
||||
.DS_Store
|
10
Makefile
Normal 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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -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
6
ui/postcss.config.js
Normal file
@ -0,0 +1,6 @@
|
||||
export default {
|
||||
plugins: {
|
||||
tailwindcss: {},
|
||||
autoprefixer: {},
|
||||
},
|
||||
}
|
BIN
ui/public/android-chrome-192x192.png
Executable file
After Width: | Height: | Size: 9.5 KiB |
BIN
ui/public/android-chrome-512x512.png
Executable file
After Width: | Height: | Size: 22 KiB |
BIN
ui/public/apple-touch-icon.png
Executable file
After Width: | Height: | Size: 8.6 KiB |
BIN
ui/public/favicon-16x16.png
Executable file
After Width: | Height: | Size: 483 B |
BIN
ui/public/favicon-32x32.png
Executable file
After Width: | Height: | Size: 1.1 KiB |
BIN
ui/public/favicon.ico
Executable file
After Width: | Height: | Size: 15 KiB |
1
ui/public/site.webmanifest
Executable 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
@ -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
@ -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
@ -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
@ -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
@ -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
@ -0,0 +1,3 @@
|
||||
@tailwind base;
|
||||
@tailwind components;
|
||||
@tailwind utilities;
|
10
ui/src/main.tsx
Normal 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
@ -0,0 +1 @@
|
||||
/// <reference types="vite/client" />
|
15
ui/tailwind.config.js
Normal 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
@ -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
@ -0,0 +1,7 @@
|
||||
{
|
||||
"files": [],
|
||||
"references": [
|
||||
{ "path": "./tsconfig.app.json" },
|
||||
{ "path": "./tsconfig.node.json" }
|
||||
]
|
||||
}
|
22
ui/tsconfig.node.json
Normal 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
@ -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
@ -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
@ -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
@ -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
|
||||
}
|