tnypxl/rollup
1
0
Fork 0
mirror of https://github.com/tnypxl/rollup.git synced 2025-12-13 06:23:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: enhance logging in runWeb function for better debugging

Browse Source
This commit is contained in:
Arik Jones (aider)
2024-09-21 10:57:51 -05:00
parent 6f39dd6726
commit 8f824d8990
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
cmd/web.go
View File

@@ -39,10 +39,12 @@ func init() {
}
func runWeb(cmd *cobra.Command, args []string) error {
log.Printf("Starting web scraping process with verbose mode: %v", verbose)
scraperConfig.Verbose = verbose
var siteConfigs []scraper.SiteConfig
if len(cfg.Scrape.Sites) > 0 {
log.Printf("Using configuration from rollup.yml for %d sites", len(cfg.Scrape.Sites))
siteConfigs = make([]scraper.SiteConfig, len(cfg.Scrape.Sites))
for i, site := range cfg.Scrape.Sites {
siteConfigs[i] = scraper.SiteConfig{
@@ -55,9 +57,11 @@ func runWeb(cmd *cobra.Command, args []string) error {
OutputAlias: site.OutputAlias,
PathOverrides: convertPathOverrides(site.PathOverrides),
}
log.Printf("Site %d configuration: BaseURL=%s, CSSLocator=%s, MaxDepth=%d, AllowedPaths=%v",
i+1, site.BaseURL, site.CSSLocator, site.MaxDepth, site.AllowedPaths)
}
} else {
// Fallback to URL-based configuration if no sites are defined
log.Printf("No sites defined in rollup.yml, falling back to URL-based configuration")
siteConfigs = make([]scraper.SiteConfig, len(urls))
for i, u := range urls {
siteConfigs[i] = scraper.SiteConfig{
@@ -66,10 +70,13 @@ func runWeb(cmd *cobra.Command, args []string) error {
ExcludeSelectors: excludeSelectors,
MaxDepth: depth,
}
log.Printf("URL %d configuration: BaseURL=%s, CSSLocator=%s, MaxDepth=%d",
i+1, u, includeSelector, depth)
}
}
if len(siteConfigs) == 0 {
log.Println("Error: No sites or URLs provided")
return fmt.Errorf("no sites or URLs provided. Use --urls flag with comma-separated URLs or set 'scrape.sites' in the rollup.yml file")
}
@@ -82,15 +89,22 @@ func runWeb(cmd *cobra.Command, args []string) error {
BurstLimit: cfg.Scrape.BurstLimit,
},
}
log.Printf("Scraper configuration: OutputType=%s, RequestsPerSecond=%f, BurstLimit=%d",
outputType, cfg.Scrape.RequestsPerSecond, cfg.Scrape.BurstLimit)
log.Println("Starting scraping process")
scrapedContent, err := scraper.ScrapeSites(scraperConfig)
if err != nil {
log.Printf("Error occurred during scraping: %v", err)
return fmt.Errorf("error scraping content: %v", err)
}
log.Printf("Scraping completed. Total content scraped: %d", len(scrapedContent))
if outputType == "single" {
log.Println("Writing content to a single file")
return writeSingleFile(scrapedContent)
} else {
log.Println("Writing content to multiple files")
return writeMultipleFiles(scrapedContent)
}
}