feat: ()dl command: download given video with yt-dlp, then upload it to imagehost and post the link in response

cmd/nourybot/catbox.go

@@ -1,199 +0,0 @@
-package main
-
-// This whole file has been pretty much straight up copied
-// from: https://github.com/wabarc/go-catbox/blob/main/catbox.go
-// since I couldn't otherwise use litterbox instead.
-
-import (
-	"bytes"
-	"errors"
-	"fmt"
-	"io"
-	"io/ioutil"
-	"mime/multipart"
-	"net/http"
-	"net/url"
-	"os"
-	"path/filepath"
-	"time"
-
-	"github.com/wabarc/helper"
-)
-
-const (
-	//ENDPOINT = "https://catbox.moe/user/api.php"
-	ENDPOINT = "https://litterbox.catbox.moe/resources/internals/api.php"
-)
-
-type catbox struct {
-	Client   *http.Client
-	Userhash string
-}
-
-func New(client *http.Client) *catbox {
-	if client == nil {
-		client = &http.Client{
-			Timeout: 300 * time.Second,
-		}
-	}
-
-	return &catbox{
-		Client: client,
-	}
-}
-
-// Upload file or URI to the Catbox. It returns an URL string and error.
-func (cat *catbox) Upload(v ...interface{}) (string, error) {
-	if len(v) == 0 {
-		return "", fmt.Errorf(`must specify file path or byte slice`)
-	}
-
-	switch t := v[0].(type) {
-	case string:
-		path := t
-		parse := func(s string, _ error) (string, error) {
-			uri, err := url.Parse(s)
-			if err != nil {
-				return "", err
-			}
-			return uri.String(), nil
-		}
-		switch {
-		case helper.IsURL(path):
-			return parse(cat.urlUpload(path))
-		case helper.Exists(path):
-			return parse(cat.fileUpload(path))
-		default:
-			return "", errors.New(`path invalid`)
-		}
-	case []byte:
-		if len(v) != 2 {
-			return "", fmt.Errorf(`must specify file name`)
-		}
-		return cat.rawUpload(t, v[1].(string))
-	}
-	return "", fmt.Errorf(`unhandled`)
-}
-
-func (cat *catbox) rawUpload(b []byte, name string) (string, error) {
-	r, w := io.Pipe()
-	m := multipart.NewWriter(w)
-
-	go func() {
-		defer w.Close()
-		defer m.Close()
-
-		m.WriteField("reqtype", "fileupload")
-		m.WriteField("userhash", cat.Userhash)
-		part, err := m.CreateFormFile("fileToUpload", filepath.Base(name))
-		if err != nil {
-			return
-		}
-		if _, err = io.Copy(part, bytes.NewBuffer(b)); err != nil {
-			return
-		}
-	}()
-	req, _ := http.NewRequest(http.MethodPost, ENDPOINT, r)
-	req.Header.Add("Content-Type", m.FormDataContentType())
-
-	resp, err := cat.Client.Do(req)
-	if err != nil {
-		return "", err
-	}
-	defer resp.Body.Close()
-
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return "", err
-	}
-
-	return string(body), nil
-}
-
-func (cat *catbox) fileUpload(path string) (string, error) {
-	file, err := os.Open(path)
-	if err != nil {
-		return "", err
-	}
-	defer file.Close()
-
-	// if size := helper.FileSize(path); size > 209715200 {
-	// 	return "", fmt.Errorf("file too large, size: %d MB", size/1024/1024)
-	// }
-
-	r, w := io.Pipe()
-	m := multipart.NewWriter(w)
-
-	go func() {
-		defer w.Close()
-		defer m.Close()
-
-		m.WriteField("reqtype", "fileupload")
-		//m.WriteField("userhash", cat.Userhash)
-		m.WriteField("time", "24h")
-		part, err := m.CreateFormFile("fileToUpload", filepath.Base(file.Name()))
-		if err != nil {
-			return
-		}
-
-		if _, err = io.Copy(part, file); err != nil {
-			return
-		}
-	}()
-
-	req, _ := http.NewRequest(http.MethodPost, ENDPOINT, r)
-	req.Header.Add("Content-Type", m.FormDataContentType())
-
-	resp, err := cat.Client.Do(req)
-	if err != nil {
-		return "", err
-	}
-	defer resp.Body.Close()
-
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return "", err
-	}
-
-	return string(body), nil
-}
-
-func (cat *catbox) urlUpload(uri string) (string, error) {
-	b := new(bytes.Buffer)
-	w := multipart.NewWriter(b)
-	w.WriteField("reqtype", "urlupload")
-	w.WriteField("userhash", cat.Userhash)
-	w.WriteField("url", uri)
-
-	req, _ := http.NewRequest(http.MethodPost, ENDPOINT, b)
-	req.Header.Add("Content-Type", w.FormDataContentType())
-
-	resp, err := cat.Client.Do(req)
-	if err != nil {
-		return "", err
-	}
-	defer resp.Body.Close()
-
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return "", err
-	}
-
-	return string(body), nil
-}
-
-func (cat *catbox) Delete(files ...string) error {
-	// TODO
-	return nil
-}
-
-func (app *application) UploadCatbox(target, path string) {
-	app.Send(target, "Uploading to catbox... dankCircle")
-
-	cat := New(nil)
-	if url, err := cat.fileUpload(path); err != nil {
-		app.Send(target, "Something went wrong FeelsBadMan")
-	} else {
-		app.Send(target, string(url))
-	}
-}

cmd/nourybot/download.go

@@ -4,7 +4,10 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"mime/multipart"
+	"net/http"
 	"os"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/wader/goutubedl"
@@ -18,13 +21,12 @@ func (app *application) Download(target, link string) {
 	if err != nil {
 		app.Log.Errorln(err)
 	}
-	rExt := result.Info.Ext
+	rExt := "mp4"
 	downloadResult, err := result.Download(context.Background(), "best")
 	if err != nil {
 		app.Log.Errorln(err)
 	}
 	app.Send(target, "Downloaded..")
-	defer downloadResult.Close()
 	fn, err := uuid.NewUUID()
 	if err != nil {
 		app.Log.Errorln(err)
@@ -40,10 +42,80 @@ func (app *application) Download(target, link string) {
 		app.Log.Errorln(err)
 	}
 
-	app.UploadCatbox(target, fmt.Sprintf("%s.%s", fn, rExt))
+	downloadResult.Close()
+	f.Close()
+	duration := 5 * time.Second
+	app.Send(target, "ResidentSleeper ..")
+	time.Sleep(duration)
 
-	//os.Remove(fmt.Sprintf("%s.mp4", fn))
+	app.upload(target, fmt.Sprintf("%s.%s", fn, rExt))
-
-	//app.TwitchClient.Say(channel, b.String())
 
 }
+
+func (app *application) upload(target, path string) {
+	const URL = "https://i.yaf.ee/upload"
+	app.Send(target, "Uploading .. dankCircle")
+	pr, pw := io.Pipe()
+	form := multipart.NewWriter(pw)
+
+	go func() {
+		defer pw.Close()
+
+		err := form.WriteField("name", "xd")
+		if err != nil {
+			app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+			return
+		}
+
+		file, err := os.Open(path) // path to image file
+		if err != nil {
+			app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+			return
+		}
+
+		w, err := form.CreateFormFile("file", path)
+		if err != nil {
+			app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+			return
+		}
+
+		_, err = io.Copy(w, file)
+		if err != nil {
+			app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+			return
+		}
+
+		form.Close()
+	}()
+
+	req, err := http.NewRequest(http.MethodPost, URL, pr)
+	if err != nil {
+		app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+		return
+	}
+	req.Header.Set("Content-Type", form.FormDataContentType())
+
+	httpClient := http.Client{}
+	resp, err := httpClient.Do(req)
+	if err != nil {
+		app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+		app.Log.Errorln("Error while sending HTTP request:", err)
+
+		return
+	}
+	defer resp.Body.Close()
+	app.Send(target, "Uploaded .. PogChamp")
+
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		app.Send(target, fmt.Sprintf("Something went wrong FeelsBadMan: %q", err))
+		app.Log.Errorln("Error while reading response:", err)
+		return
+	}
+
+	var reply = string(body[:])
+
+	app.Send(target, fmt.Sprintf("Removing file: %s", path))
+	os.Remove(path)
+	app.Send(target, reply)
+}