package main

import (
	"fmt"
	"log"
	"net/http"
	"net/url"
	"sync"

	"github.com/PuerkitoBio/goquery"
)

// News 結(jié)構(gòu)體用于存儲(chǔ)新聞信息
type News struct {
	Title   string // 新聞標(biāo)題
	Link    string // 新聞鏈接
	Summary string // 新聞?wù)?
	Image   string // 新聞圖片
}

// getProxy 函數(shù)用于從代理服務(wù)獲取代理IP地址和端口
func getProxy() (string, error) {
	// 使用億牛云爬蟲代理 設(shè)置代理IP的域名、端口、用戶名和密碼
	proxyHost := "www.16yun.cn"
	proxyPort := "8100"
	proxyUser := "16XXXX"
	proxyPass := "IPXXXX"

	// 構(gòu)造代理請(qǐng)求的URL
	requestURL := fmt.Sprintf("http://%s:%s/getip?num=1&type=1&pro=&city=0&yys=0&port=11&pack=%s&ts=0&ys=0&cs=0&lb=1&sb=0&pb=4&mr=1&regions=", proxyHost, proxyPort, proxyUser)

	// 發(fā)送請(qǐng)求到代理服務(wù)
	resp, err := http.Get(requestURL)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	// 從響應(yīng)中讀取代理IP地址和端口
	var proxy string
	fmt.Fscanf(resp.Body, "%s", &proxy)

	return proxy, nil
}

// getNews 函數(shù)用于抓取新聞信息
func getNews(proxy string, wg *sync.WaitGroup, result *[]News) {
	defer wg.Done()

	// 使用代理IP地址創(chuàng)建HTTP客戶端
	client := &http.Client{
		Transport: &http.Transport{
			Proxy: func(req *http.Request) (*url.URL, error) {
				return url.Parse("http://" + proxy)
			},
		},
	}

	// 發(fā)送請(qǐng)求到頭條新聞首頁
	resp, err := client.Get("https://www.toutiao.com/")
	if err != nil {
		log.Println(err)
		return
	}
	defer resp.Body.Close()

	// 解析HTML文檔
	doc, err := goquery.NewDocumentFromReader(resp.Body)
	if err != nil {
		log.Println(err)
		return
	}

	// 查找新聞信息節(jié)點(diǎn)并遍歷
	doc.Find(".single-mode-rbox").Each(func(i int, s *goquery.Selection) {
		var news News
		news.Title = s.Find(".title-box a").Text()
		news.Link, _ = s.Find(".title-box a").Attr("href")
		news.Summary = s.Find(".abstract").Text()
		news.Image, _ = s.Find(".img-wrap img").Attr("src")

		fmt.Println(news)
		*result = append(*result, news)
	})
}

func main() {
	var wg sync.WaitGroup
	var result []News
	var threadNum int = 10

	for i := 0; i < threadNum; i++ {
		wg.Add(1)
		proxy, err := getProxy()
		if err != nil {
			log.Println("無法獲取代理：", err)
		} else {
			go getNews(proxy, &wg, &result)
		}
	}

	wg.Wait()

	fmt.Println("抓取到的新聞：")
	for i, news := range result {
		fmt.Printf("新聞 %d:\n", i+1)
		fmt.Printf("標(biāo)題: %s\n", news.Title)
		fmt.Printf("鏈接: %s\n", news.Link)
		fmt.Printf("摘要: %s\n", news.Summary)
		fmt.Printf("圖片: %s\n", news.Image)
	}
}