Search/main.go

package main

import (
    "fmt"
    "log"
    "math/rand"
    "net/http"
    "net/url"
    "strconv"
	"strings"
    "time"
    "html/template"
    "github.com/PuerkitoBio/goquery"
)

// Define a struct to represent search result
type SearchResult struct {
    URL         string
    Header      string
    Description string
}

var funcs = template.FuncMap{
    "title": func(s string) string {
        return strings.Title(s)
    },
    "url_for": func(filename string) string {
        return "/" + filename
    },
}

var templates = template.Must(template.New("").Funcs(funcs).ParseFiles("templates/results.html"))

func main() {
    // Serve static files from the 'static' directory
    http.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))

    http.HandleFunc("/", handleSearch)
    http.HandleFunc("/search", handleSearch)
    fmt.Println("Server is listening on http://localhost:5000")
    log.Fatal(http.ListenAndServe(":5000", nil))
}

func handleSearch(w http.ResponseWriter, r *http.Request) {
    if r.Method == "GET" {
        http.ServeFile(w, r, "static/search.html")
        return
    }

    query := r.FormValue("q")
    if query == "" {
        http.ServeFile(w, r, "static/search.html")
        return
    }

    // Random user agent
    userAgents := []string{
        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36",
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/107.0.0.0 Safari/537.36",
        "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0",
        "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36",
        "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:106.0) Gecko/20100101 Firefox/106.0",
        "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.1 Safari/605.1.15",
    }
    randIndex := rand.Intn(len(userAgents))
    userAgent := userAgents[randIndex]

    // Time
    start := time.Now()

    // Search query
    searchURL := "https://www.google.com/search?q=" + url.QueryEscape(query)
    req, err := http.NewRequest("GET", searchURL, nil)
    if err != nil {
        http.Error(w, "Failed to create request", http.StatusInternalServerError)
        return
    }
    req.Header.Set("User-Agent", userAgent)
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        http.Error(w, "Failed to fetch search results", http.StatusInternalServerError)
        return
    }
    defer resp.Body.Close()

    // Parse HTML
    doc, err := goquery.NewDocumentFromReader(resp.Body)
    if err != nil {
        http.Error(w, "Failed to parse HTML", http.StatusInternalServerError)
        return
    }

    // Retrieve links, headers, and descriptions
    var results []SearchResult
    doc.Find(".yuRUbf").Each(func(i int, s *goquery.Selection) {
        link := s.Find("a")
        href, _ := link.Attr("href")
        // Extract header/title
        header := link.Find("h3").Text()
        // Remove the unwanted text from the header/title
        header = strings.TrimSpace(strings.TrimSuffix(header, "›"))

        // Retrieve corresponding description
        descSelection := doc.Find(".VwiC3b").Eq(i)
        description := ""
        if descSelection.Length() > 0 {
            description = descSelection.Text()
        }

        results = append(results, SearchResult{
            URL:         href,
            Header:      header,
            Description: description,
        })
    })

    // Retrieve kno-rdesc
    kno := ""
    knoLink := ""
    rdesc := doc.Find(".kno-rdesc")
    if rdesc.Length() > 0 {
        span := rdesc.Find("span")
        kno = span.Text()
        descLink := rdesc.Find("a")
        knoLink, _ = descLink.Attr("href")
    }

    // Retrieve featured snippet
    snip := ""
    snipSpan := doc.Find(".hgKElc")
    if snipSpan.Length() > 0 {
        snip = snipSpan.Text()
    }

    elapsed := time.Since(start)

    // Prepare data for rendering template
    data := struct {
        Results     []SearchResult
        Title       string
        Query       string
        Fetched     string
        Snippet     string
        KnoRdesc    string
        RdescLink   string
        ElapsedTime string
    }{
        Results:     results,
        Title:       query + " - TailsGo",
        Query:       query,
        Fetched:     fmt.Sprintf("Fetched the results in %.2f seconds", elapsed.Seconds()),
        Snippet:     snip,
        KnoRdesc:    kno,
        RdescLink:   knoLink,
        ElapsedTime: strconv.FormatFloat(elapsed.Seconds(), 'f', 2, 64),
    }

    // Render template
    err = templates.ExecuteTemplate(w, "results.html", data)
    if err != nil {
        http.Error(w, "Failed to render template", http.StatusInternalServerError)
        return
    }
}