Auto download yt-dlp if not found in path
This commit is contained in:
parent
540ddc2bad
commit
f1724719ec
10
main.go
10
main.go
@ -6,6 +6,7 @@ import (
|
|||||||
"net"
|
"net"
|
||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
|
"os/exec"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -13,6 +14,7 @@ import (
|
|||||||
"github.com/gotd/td/telegram/message"
|
"github.com/gotd/td/telegram/message"
|
||||||
"github.com/gotd/td/telegram/uploader"
|
"github.com/gotd/td/telegram/uploader"
|
||||||
"github.com/gotd/td/tg"
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/wader/goutubedl"
|
||||||
"golang.org/x/exp/slices"
|
"golang.org/x/exp/slices"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -160,6 +162,14 @@ func main() {
|
|||||||
telegramUploader = uploader.NewUploader(api).WithProgress(dlUploader)
|
telegramUploader = uploader.NewUploader(api).WithProgress(dlUploader)
|
||||||
telegramSender = message.NewSender(api).WithUploader(telegramUploader)
|
telegramSender = message.NewSender(api).WithUploader(telegramUploader)
|
||||||
|
|
||||||
|
goutubedl.Path, err = exec.LookPath(goutubedl.Path)
|
||||||
|
if err != nil {
|
||||||
|
goutubedl.Path, err = ytdlpDownloadLatest(ctx)
|
||||||
|
if err != nil {
|
||||||
|
panic(fmt.Sprint("error: ", err))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
dlQueue.Init(ctx)
|
dlQueue.Init(ctx)
|
||||||
|
|
||||||
dispatcher.OnNewMessage(handleMsg)
|
dispatcher.OnNewMessage(handleMsg)
|
||||||
|
|||||||
@ -4,7 +4,6 @@ import (
|
|||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@ -79,10 +78,6 @@ func (p *paramsType) Init() error {
|
|||||||
if goutubedl.Path == "" {
|
if goutubedl.Path == "" {
|
||||||
goutubedl.Path = "yt-dlp"
|
goutubedl.Path = "yt-dlp"
|
||||||
}
|
}
|
||||||
goutubedl.Path, err = exec.LookPath(goutubedl.Path)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("yt-dlp not found")
|
|
||||||
}
|
|
||||||
|
|
||||||
if allowedUserIDs == "" {
|
if allowedUserIDs == "" {
|
||||||
allowedUserIDs = os.Getenv("ALLOWED_USERIDS")
|
allowedUserIDs = os.Getenv("ALLOWED_USERIDS")
|
||||||
|
|||||||
1
run.sh
1
run.sh
@ -14,4 +14,5 @@ ALLOWED_USERIDS=$ALLOWED_USERIDS \
|
|||||||
ADMIN_USERIDS=$ADMIN_USERIDS \
|
ADMIN_USERIDS=$ADMIN_USERIDS \
|
||||||
ALLOWED_GROUPIDS=$ALLOWED_GROUPIDS \
|
ALLOWED_GROUPIDS=$ALLOWED_GROUPIDS \
|
||||||
MAX_SIZE=$MAX_SIZE \
|
MAX_SIZE=$MAX_SIZE \
|
||||||
|
YTDLP_PATH=$YTDLP_PATH \
|
||||||
$bin
|
$bin
|
||||||
|
|||||||
99
vercheck.go
99
vercheck.go
@ -2,7 +2,12 @@ package main
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/google/go-github/v53/github"
|
"github.com/google/go-github/v53/github"
|
||||||
@ -11,12 +16,100 @@ import (
|
|||||||
|
|
||||||
const ytdlpVersionCheckTimeout = time.Second * 10
|
const ytdlpVersionCheckTimeout = time.Second * 10
|
||||||
|
|
||||||
func ytdlpVersionCheck(ctx context.Context) (latestVersion, currentVersion string, err error) {
|
func ytdlpGetLatestRelease(ctx context.Context) (release *github.RepositoryRelease, err error) {
|
||||||
client := github.NewClient(nil)
|
client := github.NewClient(nil)
|
||||||
|
|
||||||
release, _, err := client.Repositories.GetLatestRelease(ctx, "yt-dlp", "yt-dlp")
|
release, _, err = client.Repositories.GetLatestRelease(ctx, "yt-dlp", "yt-dlp")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", "", fmt.Errorf("getting latest yt-dlp version: %w", err)
|
return nil, fmt.Errorf("getting latest yt-dlp version: %w", err)
|
||||||
|
}
|
||||||
|
return release, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type ytdlpGithubReleaseAsset struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
URL string `json:"browser_download_url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func ytdlpGetLatestReleaseURL(ctx context.Context) (url string, err error) {
|
||||||
|
release, err := ytdlpGetLatestRelease(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
assetsURL := release.GetAssetsURL()
|
||||||
|
if assetsURL == "" {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: no assets url")
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := http.Get(assetsURL)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
var assets []ytdlpGithubReleaseAsset
|
||||||
|
err = json.Unmarshal(body, &assets)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(assets) == 0 {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: no release assets")
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, asset := range assets {
|
||||||
|
if asset.Name == "yt-dlp" {
|
||||||
|
url = asset.URL
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if url == "" {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: no release asset url")
|
||||||
|
}
|
||||||
|
return url, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func ytdlpDownloadLatest(ctx context.Context) (path string, err error) {
|
||||||
|
url, err := ytdlpGetLatestReleaseURL(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := http.Get(url)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
file, err := os.Create(filepath.Join(os.TempDir(), "yt-dlp"))
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("downloading latest yt-dlp: %w", err)
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
_, err = io.Copy(file, resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.Chmod(file.Name(), 0755)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return file.Name(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func ytdlpVersionCheck(ctx context.Context) (latestVersion, currentVersion string, err error) {
|
||||||
|
release, err := ytdlpGetLatestRelease(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return "", "", err
|
||||||
}
|
}
|
||||||
latestVersion = release.GetTagName()
|
latestVersion = release.GetTagName()
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user