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

feat: Separate include and exclude selectors in web scraper

Browse Source
This commit is contained in:
Arik Jones (aider)
2024-09-14 16:59:59 -05:00
parent 39e06ee9d5
commit 7cdd68d020
2 changed files with 15 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
cmd/web.go
View File

@@ -12,10 +12,11 @@ import (
)
var (
urls []string
outputFile string
depth int
cssSelector string
urls []string
outputFile string
depth int
includeSelector string
excludeSelectors []string
)
var scraperConfig scraper.Config
@@ -32,7 +33,8 @@ func init() {
webCmd.Flags().StringSliceVarP(&urls, "urls", "u", []string{}, "URLs of the webpages to scrape (comma-separated)")
webCmd.Flags().StringVarP(&outputFile, "output", "o", "", "Output Markdown file (default: rollup-web-<timestamp>.md)")
webCmd.Flags().IntVarP(&depth, "depth", "d", 0, "Depth of link traversal (default: 0, only scrape the given URLs)")
webCmd.Flags().StringVar(&cssSelector, "css", "", "CSS selector to extract specific content (use '-' to exclude elements, e.g., 'main - .ads - .navigation')")
webCmd.Flags().StringVar(&includeSelector, "include", "", "CSS selector to extract specific content")
webCmd.Flags().StringSliceVar(&excludeSelectors, "exclude", []string{}, "CSS selectors to exclude from the extracted content (comma-separated)")
}
func runWeb(cmd *cobra.Command, args []string) error {
@@ -125,8 +127,12 @@ func extractAndConvertContent(urlStr string) (string, error) {
return "", fmt.Errorf("error fetching webpage content: %v", err)
}
// The content is already extracted using the main element,
// so we don't need to use ExtractContentWithCSS or ExtractContentWithXPath here
if includeSelector != "" {
content, err = scraper.ExtractContentWithCSS(content, includeSelector, excludeSelectors)
if err != nil {
return "", fmt.Errorf("error extracting content with CSS: %v", err)
}
}
// Create a new converter
converter := md.NewConverter("", true, nil)

17
internal/scraper/scraper.go
View File

@@ -243,16 +243,14 @@ func ExtractLinks(urlStr string) ([]string, error) {
}
// ExtractContentWithCSS extracts content from HTML using a CSS selector
func ExtractContentWithCSS(content, selector string) (string, error) {
log.Printf("Extracting content with CSS selector: %s\n", selector)
func ExtractContentWithCSS(content, includeSelector string, excludeSelectors []string) (string, error) {
log.Printf("Extracting content with CSS selector: %s\n", includeSelector)
doc, err := goquery.NewDocumentFromReader(strings.NewReader(content))
if err != nil {
return "", fmt.Errorf("error parsing HTML: %v", err)
}
includeSelector, excludeSelectors := parseSelectors(selector)
selection := doc.Find(includeSelector)
if selection.Length() == 0 {
return "", fmt.Errorf("no content found with CSS selector: %s", includeSelector)
@@ -270,14 +268,3 @@ func ExtractContentWithCSS(content, selector string) (string, error) {
log.Printf("Extracted content length: %d\n", len(selectedContent))
return selectedContent, nil
}
// parseSelectors splits the CSS selector string into include and exclude parts
func parseSelectors(selector string) (string, []string) {
parts := strings.Split(selector, "-")
includeSelector := strings.TrimSpace(parts[0])
var excludeSelectors []string
for _, part := range parts[1:] {
excludeSelectors = append(excludeSelectors, strings.TrimPrefix(part, " "))
}
return includeSelector, excludeSelectors
}