mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-12 15:05:10 -05:00
6e2f00090c
### CHANGES - Add early return if registry is nil to prevent panics. - Introduce early return for non-chat tool operations. - Update error message to use original input on HTML readability failure. - Enhance error wrapping for playlist video fetching. - Modify temp patterns folder name with timestamp for uniqueness. - Improve error handling for patterns directory access.
87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/danielmiessler/fabric/internal/core"
|
|
"github.com/danielmiessler/fabric/internal/tools/youtube"
|
|
)
|
|
|
|
// handleToolProcessing handles YouTube and web scraping tool processing
|
|
func handleToolProcessing(currentFlags *Flags, registry *core.PluginRegistry) (messageTools string, err error) {
|
|
if currentFlags.YouTube != "" {
|
|
if !registry.YouTube.IsConfigured() {
|
|
err = fmt.Errorf("YouTube is not configured, please run the setup procedure")
|
|
return
|
|
}
|
|
|
|
var videoId string
|
|
var playlistId string
|
|
if videoId, playlistId, err = registry.YouTube.GetVideoOrPlaylistId(currentFlags.YouTube); err != nil {
|
|
return
|
|
} else if (videoId == "" || currentFlags.YouTubePlaylist) && playlistId != "" {
|
|
if currentFlags.Output != "" {
|
|
err = registry.YouTube.FetchAndSavePlaylist(playlistId, currentFlags.Output)
|
|
} else {
|
|
var videos []*youtube.VideoMeta
|
|
if videos, err = registry.YouTube.FetchPlaylistVideos(playlistId); err != nil {
|
|
err = fmt.Errorf("error fetching playlist videos: %w", err)
|
|
return
|
|
}
|
|
|
|
for _, video := range videos {
|
|
var message string
|
|
if message, err = processYoutubeVideo(currentFlags, registry, video.Id); err != nil {
|
|
return
|
|
}
|
|
|
|
if !currentFlags.IsChatRequest() {
|
|
if err = WriteOutput(message, fmt.Sprintf("%v.md", video.TitleNormalized)); err != nil {
|
|
return
|
|
}
|
|
} else {
|
|
messageTools = AppendMessage(messageTools, message)
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
if messageTools, err = processYoutubeVideo(currentFlags, registry, videoId); err != nil {
|
|
return
|
|
}
|
|
if !currentFlags.IsChatRequest() {
|
|
err = currentFlags.WriteOutput(messageTools)
|
|
return
|
|
}
|
|
}
|
|
|
|
if (currentFlags.ScrapeURL != "" || currentFlags.ScrapeQuestion != "") && registry.Jina.IsConfigured() {
|
|
// Check if the scrape_url flag is set and call ScrapeURL
|
|
if currentFlags.ScrapeURL != "" {
|
|
var website string
|
|
if website, err = registry.Jina.ScrapeURL(currentFlags.ScrapeURL); err != nil {
|
|
return
|
|
}
|
|
messageTools = AppendMessage(messageTools, website)
|
|
}
|
|
|
|
// Check if the scrape_question flag is set and call ScrapeQuestion
|
|
if currentFlags.ScrapeQuestion != "" {
|
|
var website string
|
|
if website, err = registry.Jina.ScrapeQuestion(currentFlags.ScrapeQuestion); err != nil {
|
|
return
|
|
}
|
|
|
|
messageTools = AppendMessage(messageTools, website)
|
|
}
|
|
|
|
if !currentFlags.IsChatRequest() {
|
|
err = currentFlags.WriteOutput(messageTools)
|
|
return
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|