feat: add download commands (yt, ig, tt, fb) and middleware support
- Add core Middleware support via `core.Default.Use()` - Implement download command handlers for YouTube, Instagram, TikTok, and Facebook - Add regex-based URL validation for all download commands - Add API client for fetching download links - Add helpers for uploading and sending media to WhatsApp Co-Authored-By: Claude Opus 4.7 <[email protected]>
This commit is contained in:
Claude Opus 4.7
+119
@@ -0,0 +1,119 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/gabriel-vasile/mimetype"
|
||||
)
|
||||
|
||||
// Client is the shared HTTP client for the Mono REST API.
|
||||
type Client struct {
|
||||
BaseURL string
|
||||
APIKey string
|
||||
}
|
||||
|
||||
const MaxDownloadBytes int64 = 256 * 1024 * 1024
|
||||
|
||||
var (
|
||||
defaultClient *Client
|
||||
defaultOnce sync.Once
|
||||
httpClient = &http.Client{Timeout: 2 * time.Minute}
|
||||
)
|
||||
|
||||
func initDefault() {
|
||||
baseURL := os.Getenv("MONO_BASEURL")
|
||||
if baseURL == "" {
|
||||
baseURL = "https://mono.fdvky.me/api/v1"
|
||||
}
|
||||
defaultClient = &Client{
|
||||
BaseURL: strings.TrimRight(baseURL, "/"),
|
||||
APIKey: os.Getenv("MONO_API_KEY"),
|
||||
}
|
||||
}
|
||||
|
||||
// Default returns the shared API client, initialized lazily so env vars
|
||||
// loaded via godotenv in main.init() are already available.
|
||||
func Default() *Client {
|
||||
defaultOnce.Do(initDefault)
|
||||
return defaultClient
|
||||
}
|
||||
|
||||
// Get calls the API and decodes the JSON response into target.
|
||||
func (c *Client) Get(path string, params map[string]string, target any) error {
|
||||
u, err := url.Parse(c.BaseURL + path)
|
||||
if err != nil {
|
||||
return fmt.Errorf("api url: %w", err)
|
||||
}
|
||||
q := u.Query()
|
||||
for k, v := range params {
|
||||
q.Set(k, v)
|
||||
}
|
||||
u.RawQuery = q.Encode()
|
||||
|
||||
req, err := http.NewRequest("GET", u.String(), nil)
|
||||
if err != nil {
|
||||
return fmt.Errorf("api request: %w", err)
|
||||
}
|
||||
if c.APIKey != "" {
|
||||
req.Header.Set("X-API-Key", c.APIKey)
|
||||
}
|
||||
|
||||
resp, err := httpClient.Do(req)
|
||||
if err != nil {
|
||||
return fmt.Errorf("api: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
body, _ := io.ReadAll(resp.Body)
|
||||
return fmt.Errorf("api %d: %s", resp.StatusCode, strings.TrimSpace(string(body)))
|
||||
}
|
||||
|
||||
return json.NewDecoder(resp.Body).Decode(target)
|
||||
}
|
||||
|
||||
// Download fetches a URL and returns the raw bytes and MIME type.
|
||||
func Download(rawURL string) ([]byte, string, error) {
|
||||
req, err := http.NewRequest("GET", rawURL, nil)
|
||||
if err != nil {
|
||||
return nil, "", fmt.Errorf("download request: %w", err)
|
||||
}
|
||||
|
||||
resp, err := httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, "", fmt.Errorf("download: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||
body, _ := io.ReadAll(io.LimitReader(resp.Body, 4096))
|
||||
return nil, "", fmt.Errorf("download %d: %s", resp.StatusCode, strings.TrimSpace(string(body)))
|
||||
}
|
||||
if resp.ContentLength > MaxDownloadBytes {
|
||||
return nil, "", fmt.Errorf("download too large: %d bytes", resp.ContentLength)
|
||||
}
|
||||
|
||||
mime := resp.Header.Get("Content-Type")
|
||||
|
||||
data, err := io.ReadAll(io.LimitReader(resp.Body, MaxDownloadBytes+1))
|
||||
if err != nil {
|
||||
return nil, "", fmt.Errorf("read: %w", err)
|
||||
}
|
||||
if int64(len(data)) > MaxDownloadBytes {
|
||||
return nil, "", fmt.Errorf("download too large: over %d bytes", MaxDownloadBytes)
|
||||
}
|
||||
|
||||
if mime == "" || mime == "application/octet-stream" {
|
||||
mime = mimetype.Detect(data).String()
|
||||
}
|
||||
|
||||
return data, mime, nil
|
||||
}
|
||||
+128
@@ -0,0 +1,128 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type YtFormat struct {
|
||||
Ext string `json:"ext"`
|
||||
Filesize int `json:"filesize"`
|
||||
FormatID string `json:"formatId"`
|
||||
HasAudio bool `json:"hasAudio"`
|
||||
Resolution string `json:"resolution"`
|
||||
URL string `json:"url"`
|
||||
}
|
||||
|
||||
type YtDownloadResult struct {
|
||||
AudioFormats []YtFormat `json:"audioFormats"`
|
||||
Thumbnail string `json:"thumbnail"`
|
||||
Title string `json:"title"`
|
||||
VideoFormats []YtFormat `json:"videoFormats"`
|
||||
VideoID string `json:"videoId"`
|
||||
}
|
||||
|
||||
const ytCacheTTL = 10 * time.Minute
|
||||
|
||||
type ytCacheEntry struct {
|
||||
result *YtDownloadResult
|
||||
expiresAt time.Time
|
||||
}
|
||||
|
||||
var (
|
||||
ytMu sync.Mutex
|
||||
ytCache = map[string]ytCacheEntry{}
|
||||
)
|
||||
|
||||
func GetYtCache(messageID string) (*YtDownloadResult, bool) {
|
||||
ytMu.Lock()
|
||||
defer ytMu.Unlock()
|
||||
|
||||
entry, ok := ytCache[messageID]
|
||||
if !ok {
|
||||
return nil, false
|
||||
}
|
||||
if time.Now().After(entry.expiresAt) {
|
||||
delete(ytCache, messageID)
|
||||
return nil, false
|
||||
}
|
||||
return entry.result, true
|
||||
}
|
||||
|
||||
func SetYtCache(messageID string, result *YtDownloadResult) {
|
||||
ytMu.Lock()
|
||||
defer ytMu.Unlock()
|
||||
|
||||
now := time.Now()
|
||||
for id, entry := range ytCache {
|
||||
if now.After(entry.expiresAt) {
|
||||
delete(ytCache, id)
|
||||
}
|
||||
}
|
||||
ytCache[messageID] = ytCacheEntry{
|
||||
result: result,
|
||||
expiresAt: now.Add(ytCacheTTL),
|
||||
}
|
||||
}
|
||||
|
||||
func HasFormatID(res *YtDownloadResult, formatID string) bool {
|
||||
for _, f := range res.VideoFormats {
|
||||
if f.FormatID == formatID && f.URL != "" {
|
||||
return true
|
||||
}
|
||||
}
|
||||
for _, f := range res.AudioFormats {
|
||||
if f.FormatID == formatID && f.URL != "" {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// func DownloadRemuxedMP4(rawURL string) ([]byte, error) {
|
||||
// if _, err := exec.LookPath("ffmpeg"); err != nil {
|
||||
// return nil, fmt.Errorf("ffmpeg not found: %w", err)
|
||||
// }
|
||||
|
||||
// data, mime, err := Download(rawURL)
|
||||
// if err != nil {
|
||||
// return nil, err
|
||||
// }
|
||||
|
||||
// if !strings.HasPrefix(mime, "video/mp4") {
|
||||
// return data, nil
|
||||
// }
|
||||
|
||||
// // TEST: set YT_NO_REMUX=1 to skip ffmpeg and use raw API output
|
||||
// if os.Getenv("YT_NO_REMUX") != "" {
|
||||
// return data, nil
|
||||
// }
|
||||
|
||||
// out, err := os.CreateTemp("", "yt-out-*.mp4")
|
||||
// if err != nil {
|
||||
// return nil, fmt.Errorf("temp file: %w", err)
|
||||
// }
|
||||
// outPath := out.Name()
|
||||
// out.Close()
|
||||
// defer os.Remove(outPath)
|
||||
|
||||
// cmdCtx, cancelCmd := context.WithTimeout(context.Background(), 5*time.Minute)
|
||||
// defer cancelCmd()
|
||||
|
||||
// cmd := exec.CommandContext(cmdCtx, "ffmpeg", "-y", "-hide_banner", "-loglevel", "error",
|
||||
// "-i", "pipe:0", "-c", "copy", "-movflags", "+faststart", "-f", "mp4", outPath)
|
||||
// cmd.Stdin = bytes.NewReader(data)
|
||||
// if output, err := cmd.CombinedOutput(); err != nil {
|
||||
// return nil, fmt.Errorf("ffmpeg remux: %w: %s", err, strings.TrimSpace(string(output)))
|
||||
// }
|
||||
|
||||
// info, err := os.Stat(outPath)
|
||||
// if err != nil {
|
||||
// return nil, fmt.Errorf("stat remuxed file: %w", err)
|
||||
// }
|
||||
// if info.Size() > MaxDownloadBytes {
|
||||
// return nil, fmt.Errorf("remuxed file too large: %d bytes", info.Size())
|
||||
// }
|
||||
|
||||
// return os.ReadFile(outPath)
|
||||
// }
|
||||
@@ -0,0 +1,63 @@
|
||||
package download
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
|
||||
"neo/api"
|
||||
hc "neo/context"
|
||||
"neo/core"
|
||||
"neo/helper"
|
||||
)
|
||||
|
||||
type facebookResult struct {
|
||||
HD string `json:"hd"`
|
||||
SD string `json:"sd"`
|
||||
Thumbnail string `json:"thumbnail"`
|
||||
}
|
||||
|
||||
func init() { core.Default.Register(Facebook) }
|
||||
|
||||
var Facebook = &core.Command{
|
||||
Name: "facebook",
|
||||
Aliases: []string{"fb"},
|
||||
Category: "download",
|
||||
Description: "Download Facebook video",
|
||||
Run: func(ctx *hc.Ctx) {
|
||||
url := ctx.Arg(0)
|
||||
if url == "" {
|
||||
ctx.Reply("Usage: !fb <url>")
|
||||
return
|
||||
}
|
||||
|
||||
fbRegex := regexp.MustCompile(`(?i)(?:facebook\.com|fb\.com|fb\.watch)\/(?:share\/[pr]\/|reel\/|watch\/?\?v=|.*\/videos\/)?([a-zA-Z0-9_\-]+)`)
|
||||
if !fbRegex.MatchString(url) {
|
||||
ctx.Reply("Invalid Facebook URL")
|
||||
return
|
||||
}
|
||||
|
||||
ctx.Send("Downloading...")
|
||||
var res facebookResult
|
||||
if err := api.Default().Get("/fb", map[string]string{"url": url}, &res); err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Failed to fetch: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
downloadURL := res.HD
|
||||
if downloadURL == "" {
|
||||
downloadURL = res.SD
|
||||
}
|
||||
if downloadURL == "" {
|
||||
ctx.Reply("No video found")
|
||||
return
|
||||
}
|
||||
|
||||
data, mime, err := api.Download(downloadURL)
|
||||
if err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Download failed: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
helper.SendMedia(ctx, data, mime, "")
|
||||
},
|
||||
}
|
||||
@@ -0,0 +1,69 @@
|
||||
package download
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
|
||||
"neo/api"
|
||||
hc "neo/context"
|
||||
"neo/core"
|
||||
"neo/helper"
|
||||
)
|
||||
|
||||
type instagramVideo struct {
|
||||
Src string `json:"src"`
|
||||
Thumbnail string `json:"thumbnail"`
|
||||
}
|
||||
|
||||
type instagramResult struct {
|
||||
Images []string `json:"images"`
|
||||
Videos []instagramVideo `json:"videos"`
|
||||
}
|
||||
|
||||
func init() { core.Default.Register(Instagram) }
|
||||
|
||||
var Instagram = &core.Command{
|
||||
Name: "instagram",
|
||||
Aliases: []string{"ig"},
|
||||
Category: "download",
|
||||
Description: "Download Instagram media",
|
||||
Run: func(ctx *hc.Ctx) {
|
||||
url := ctx.Arg(0)
|
||||
if url == "" {
|
||||
ctx.Reply("Usage: !ig <url>")
|
||||
return
|
||||
}
|
||||
|
||||
igRegex := regexp.MustCompile(`(?i)(?:instagram\.com|instagr\.am)\/(?:p|reel|tv)\/([a-zA-Z0-9_\-]+)`)
|
||||
if !igRegex.MatchString(url) {
|
||||
ctx.Reply("Invalid Instagram URL")
|
||||
return
|
||||
}
|
||||
|
||||
ctx.Send("Downloading...")
|
||||
var res instagramResult
|
||||
if err := api.Default().Get("/ig", map[string]string{"url": url}, &res); err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Failed to fetch: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
mediaURL := ""
|
||||
if len(res.Videos) > 0 {
|
||||
mediaURL = res.Videos[0].Src
|
||||
} else if len(res.Images) > 0 {
|
||||
mediaURL = res.Images[0]
|
||||
}
|
||||
if mediaURL == "" {
|
||||
ctx.Reply("No media found")
|
||||
return
|
||||
}
|
||||
|
||||
data, mime, err := api.Download(mediaURL)
|
||||
if err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Download failed: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
helper.SendMedia(ctx, data, mime, "")
|
||||
},
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
package download
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
|
||||
"neo/api"
|
||||
hc "neo/context"
|
||||
"neo/core"
|
||||
"neo/helper"
|
||||
)
|
||||
|
||||
type tiktokResult struct {
|
||||
Description string `json:"description"`
|
||||
ImageURL []string `json:"image_url"`
|
||||
Username string `json:"username"`
|
||||
VideoURL []string `json:"video_url"`
|
||||
}
|
||||
|
||||
func init() { core.Default.Register(TikTok) }
|
||||
|
||||
var TikTok = &core.Command{
|
||||
Name: "tiktok",
|
||||
Aliases: []string{"tt", "tk"},
|
||||
Category: "download",
|
||||
Description: "Download TikTok video",
|
||||
Run: func(ctx *hc.Ctx) {
|
||||
url := ctx.Arg(0)
|
||||
if url == "" {
|
||||
ctx.Reply("Usage: !tiktok <url>")
|
||||
return
|
||||
}
|
||||
|
||||
ttRegex := regexp.MustCompile(`(?i)(?:tiktok\.com)\/(?:@[\w.-]+\/video\/|v\/|t\/)([0-9]+)`)
|
||||
if !ttRegex.MatchString(url) {
|
||||
ctx.Reply("Invalid TikTok URL")
|
||||
return
|
||||
}
|
||||
|
||||
ctx.Send("Downloading...")
|
||||
var res tiktokResult
|
||||
if err := api.Default().Get("/tiktok", map[string]string{"url": url}, &res); err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Failed to fetch: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
mediaURL := ""
|
||||
if len(res.VideoURL) > 0 {
|
||||
mediaURL = res.VideoURL[0]
|
||||
} else if len(res.ImageURL) > 0 {
|
||||
mediaURL = res.ImageURL[0]
|
||||
}
|
||||
if mediaURL == "" {
|
||||
ctx.Reply("No media found")
|
||||
return
|
||||
}
|
||||
|
||||
data, mime, err := api.Download(mediaURL)
|
||||
if err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Download failed: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
caption := fmt.Sprintf("@%s\n%s", res.Username, res.Description)
|
||||
helper.SendMedia(ctx, data, mime, caption)
|
||||
},
|
||||
}
|
||||
@@ -0,0 +1,156 @@
|
||||
package download
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"neo/api"
|
||||
hc "neo/context"
|
||||
"neo/core"
|
||||
"neo/helper"
|
||||
|
||||
waE2E "go.mau.fi/whatsmeow/proto/waE2E"
|
||||
)
|
||||
|
||||
func init() {
|
||||
core.Default.Register(YouTube)
|
||||
core.Default.Use(handleYTReply)
|
||||
}
|
||||
|
||||
func handleYTReply(ctx *hc.Ctx) bool {
|
||||
if ctx.Cmd() != "" {
|
||||
return false
|
||||
}
|
||||
|
||||
quotedID := hc.GetQuotedStanzaID(ctx.Message())
|
||||
if quotedID == "" {
|
||||
return false
|
||||
}
|
||||
|
||||
result, ok := api.GetYtCache(quotedID)
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
formatID := strings.TrimSpace(hc.ParseMessageText(ctx.Event()))
|
||||
if formatID == "" || !api.HasFormatID(result, formatID) {
|
||||
return false
|
||||
}
|
||||
|
||||
ctx.Send("Downloading...")
|
||||
downloadByFormatID(ctx, result, formatID)
|
||||
return true
|
||||
}
|
||||
|
||||
var YouTube = &core.Command{
|
||||
Name: "youtube",
|
||||
Aliases: []string{"yt", "ytdl"},
|
||||
Category: "download",
|
||||
Description: "Download YouTube video",
|
||||
Run: func(ctx *hc.Ctx) {
|
||||
url := ctx.Arg(0)
|
||||
if url == "" {
|
||||
ctx.Reply("Usage: !yt <url>")
|
||||
return
|
||||
}
|
||||
|
||||
ytRegex := regexp.MustCompile(`(?i)(?:youtube\.com\/(?:[^\/]+\/.+\/|(?:v|e(?:mbed)?)\/|.*[?&]v=)|youtu\.be\/|youtube\.com\/shorts\/)([^"&?\/\s]{11})`)
|
||||
if !ytRegex.MatchString(url) {
|
||||
ctx.Reply("Invalid YouTube URL")
|
||||
return
|
||||
}
|
||||
|
||||
ctx.Send("Processing...")
|
||||
var res api.YtDownloadResult
|
||||
if err := api.Default().Get("/yt/query", map[string]string{"url": url}, &res); err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Error: %v", err))
|
||||
return
|
||||
}
|
||||
|
||||
showFormatPicker(ctx, &res)
|
||||
},
|
||||
}
|
||||
|
||||
func showFormatPicker(ctx *hc.Ctx, res *api.YtDownloadResult) {
|
||||
thumb, _, err := api.Download(res.Thumbnail)
|
||||
if err != nil {
|
||||
ctx.Send(formatCaption(res))
|
||||
return
|
||||
}
|
||||
|
||||
img, err := ctx.UploadImage(thumb, formatCaption(res))
|
||||
if err != nil {
|
||||
ctx.Reply("Failed to upload thumbnail: " + err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
resp, err := ctx.Client().SendMessage(context.Background(), ctx.ChatJID(), &waE2E.Message{ImageMessage: img})
|
||||
if err != nil {
|
||||
ctx.Reply("Failed to send: " + err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
api.SetYtCache(resp.ID, res)
|
||||
}
|
||||
|
||||
func formatCaption(res *api.YtDownloadResult) string {
|
||||
var b strings.Builder
|
||||
b.WriteString("Title: ")
|
||||
b.WriteString(res.Title)
|
||||
b.WriteString("\n\n── Video ──\n")
|
||||
for _, f := range res.VideoFormats {
|
||||
if f.URL == "" {
|
||||
continue
|
||||
}
|
||||
fmt.Fprintf(&b, "%s · %s · %s · %s\n", f.FormatID, f.Resolution, f.Ext, formatSize(f.Filesize))
|
||||
}
|
||||
b.WriteString("\n── Audio ──\n")
|
||||
for _, f := range res.AudioFormats {
|
||||
if f.URL == "" {
|
||||
continue
|
||||
}
|
||||
fmt.Fprintf(&b, "%s · %s · %s\n", f.FormatID, f.Ext, formatSize(f.Filesize))
|
||||
}
|
||||
b.WriteString("\nReply with formatId")
|
||||
return b.String()
|
||||
}
|
||||
|
||||
func formatSize(bytes int) string {
|
||||
if bytes <= 0 {
|
||||
return "?"
|
||||
}
|
||||
if bytes < 1024*1024 {
|
||||
return fmt.Sprintf("%dKB", bytes/1024)
|
||||
}
|
||||
return fmt.Sprintf("%.1fMB", float64(bytes)/(1024*1024))
|
||||
}
|
||||
|
||||
func downloadByFormatID(ctx *hc.Ctx, res *api.YtDownloadResult, fid string) {
|
||||
// Look for video
|
||||
for _, f := range res.VideoFormats {
|
||||
if f.FormatID == fid && f.URL != "" {
|
||||
data, mime, err := api.Download(f.URL)
|
||||
if err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Download failed: %v", err))
|
||||
return
|
||||
}
|
||||
helper.SendMedia(ctx, data, mime, res.Title) // using existing sendMedia handling
|
||||
return
|
||||
}
|
||||
}
|
||||
// Look for audio
|
||||
for _, f := range res.AudioFormats {
|
||||
if f.FormatID == fid && f.URL != "" {
|
||||
data, mime, err := api.Download(f.URL)
|
||||
if err != nil {
|
||||
ctx.Reply(fmt.Sprintf("Download failed: %v", err))
|
||||
return
|
||||
}
|
||||
helper.SendMedia(ctx, data, mime, res.Title)
|
||||
return
|
||||
}
|
||||
}
|
||||
ctx.Reply("Format " + fid + " not found")
|
||||
}
|
||||
@@ -1,5 +1,6 @@
|
||||
package cmds
|
||||
|
||||
import (
|
||||
_ "neo/cmds/download"
|
||||
_ "neo/cmds/misc"
|
||||
)
|
||||
|
||||
@@ -66,6 +66,29 @@ func WithReply(evt *events.Message) *waE2E.ContextInfo {
|
||||
}
|
||||
}
|
||||
|
||||
// GetQuotedStanzaID returns the stanza ID of the message being replied to, if any.
|
||||
func GetQuotedStanzaID(m *waE2E.Message) string {
|
||||
if m == nil {
|
||||
return ""
|
||||
}
|
||||
switch {
|
||||
case m.GetExtendedTextMessage().GetContextInfo() != nil:
|
||||
return m.GetExtendedTextMessage().GetContextInfo().GetStanzaID()
|
||||
case m.GetImageMessage().GetContextInfo() != nil:
|
||||
return m.GetImageMessage().GetContextInfo().GetStanzaID()
|
||||
case m.GetVideoMessage().GetContextInfo() != nil:
|
||||
return m.GetVideoMessage().GetContextInfo().GetStanzaID()
|
||||
case m.GetDocumentMessage().GetContextInfo() != nil:
|
||||
return m.GetDocumentMessage().GetContextInfo().GetStanzaID()
|
||||
case m.GetAudioMessage().GetContextInfo() != nil:
|
||||
return m.GetAudioMessage().GetContextInfo().GetStanzaID()
|
||||
case m.GetStickerMessage().GetContextInfo() != nil:
|
||||
return m.GetStickerMessage().GetContextInfo().GetStanzaID()
|
||||
default:
|
||||
return ""
|
||||
}
|
||||
}
|
||||
|
||||
// ParseJID parses a JID string, adding @s.whatsapp.net if no server is specified.
|
||||
func ParseJID(arg string) (waTypes.JID, bool) {
|
||||
if arg == "" {
|
||||
|
||||
+20
-1
@@ -26,10 +26,19 @@ type Command struct {
|
||||
// via init() — see neo/cmds for the registration pattern.
|
||||
var Default = New()
|
||||
|
||||
type Middleware func(ctx *hc.Ctx) bool
|
||||
|
||||
type Handler struct {
|
||||
mu sync.RWMutex
|
||||
commands map[string]*Command
|
||||
prefixes []string
|
||||
middlewares []Middleware
|
||||
}
|
||||
|
||||
func (h *Handler) Use(fn Middleware) {
|
||||
h.mu.Lock()
|
||||
defer h.mu.Unlock()
|
||||
h.middlewares = append(h.middlewares, fn)
|
||||
}
|
||||
|
||||
func New(prefixes ...string) *Handler {
|
||||
@@ -92,6 +101,17 @@ func (h *Handler) processMessage(client *whatsmeow.Client, evt *events.Message)
|
||||
|
||||
text := hc.ParseMessageText(evt)
|
||||
prefix, cmd, args, isCmd := parseCommand(text, h.prefixes)
|
||||
ctx := hc.NewCtx(client, evt, prefix, cmd, args)
|
||||
|
||||
h.mu.RLock()
|
||||
middlewares := append([]Middleware(nil), h.middlewares...)
|
||||
h.mu.RUnlock()
|
||||
for _, middleware := range middlewares {
|
||||
if middleware(ctx) {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
if !isCmd {
|
||||
return
|
||||
}
|
||||
@@ -108,7 +128,6 @@ func (h *Handler) processMessage(client *whatsmeow.Client, evt *events.Message)
|
||||
return
|
||||
}
|
||||
|
||||
ctx := hc.NewCtx(client, evt, prefix, cmd, args)
|
||||
fmt.Printf("[CMD] %s > %s%s\n", formatSenderInfo(evt), prefix, cmd)
|
||||
command.Run(ctx)
|
||||
}
|
||||
|
||||
@@ -3,9 +3,12 @@ module neo
|
||||
go 1.25.5
|
||||
|
||||
require (
|
||||
github.com/gabriel-vasile/mimetype v1.4.13
|
||||
github.com/joho/godotenv v1.5.1
|
||||
github.com/mattn/go-sqlite3 v1.14.47
|
||||
github.com/mdp/qrterminal/v3 v3.2.1
|
||||
go.mau.fi/whatsmeow v0.0.0-20260630180629-b572e5bcb92b
|
||||
google.golang.org/protobuf v1.36.11
|
||||
)
|
||||
|
||||
require (
|
||||
@@ -13,9 +16,7 @@ require (
|
||||
github.com/beeper/argo-go v1.1.2 // indirect
|
||||
github.com/coder/websocket v1.8.15 // indirect
|
||||
github.com/elliotchance/orderedmap/v3 v3.1.0 // indirect
|
||||
github.com/gabriel-vasile/mimetype v1.4.13 // indirect
|
||||
github.com/google/uuid v1.6.0 // indirect
|
||||
github.com/joho/godotenv v1.5.1 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/petermattis/goid v0.0.0-20260330135022-df67b199bc81 // indirect
|
||||
@@ -30,6 +31,5 @@ require (
|
||||
golang.org/x/sys v0.46.0 // indirect
|
||||
golang.org/x/term v0.44.0 // indirect
|
||||
golang.org/x/text v0.38.0 // indirect
|
||||
google.golang.org/protobuf v1.36.11 // indirect
|
||||
rsc.io/qr v0.2.0 // indirect
|
||||
)
|
||||
|
||||
@@ -0,0 +1,49 @@
|
||||
package helper
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
hc "neo/context"
|
||||
)
|
||||
|
||||
func SendMedia(ctx *hc.Ctx, data []byte, mime, caption string) {
|
||||
switch {
|
||||
case strings.HasPrefix(mime, "video/mp4"):
|
||||
ctx.SendVideo(data, caption)
|
||||
case strings.HasPrefix(mime, "image/"):
|
||||
ctx.SendImage(data, caption)
|
||||
case strings.HasPrefix(mime, "audio/"):
|
||||
ctx.SendAudio(data)
|
||||
default:
|
||||
ctx.SendDocument(data, "media", "media"+extByMIME(mime))
|
||||
}
|
||||
}
|
||||
|
||||
func extByMIME(mime string) string {
|
||||
switch {
|
||||
case strings.HasPrefix(mime, "video/mp4"):
|
||||
return ".mp4"
|
||||
case strings.HasPrefix(mime, "video/webm"):
|
||||
return ".webm"
|
||||
case strings.HasPrefix(mime, "video/quicktime"):
|
||||
return ".mov"
|
||||
case strings.HasPrefix(mime, "video/x-matroska"):
|
||||
return ".mkv"
|
||||
case strings.HasPrefix(mime, "video/"):
|
||||
return ".video"
|
||||
case strings.HasPrefix(mime, "image/jpeg"):
|
||||
return ".jpg"
|
||||
case strings.HasPrefix(mime, "image/png"):
|
||||
return ".png"
|
||||
case strings.HasPrefix(mime, "image/gif"):
|
||||
return ".gif"
|
||||
case strings.HasPrefix(mime, "image/webp"):
|
||||
return ".webp"
|
||||
case strings.HasPrefix(mime, "audio/mpeg"), strings.HasPrefix(mime, "audio/mp3"):
|
||||
return ".mp3"
|
||||
case strings.HasPrefix(mime, "audio/"):
|
||||
return ".m4a"
|
||||
default:
|
||||
return ".bin"
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user