package background import ( "fmt" "net/http" "regexp" "sort" "strconv" "strings" "time" "github.com/PuerkitoBio/goquery" "git.trj.tw/golang/mtfosbot/model" "git.trj.tw/golang/mtfosbot/module/apis/line" ) var idRegex = []*regexp.Regexp{ regexp.MustCompile(`[\?|&]id\=(\d+)`), regexp.MustCompile(`\/posts\/(\d+)`), regexp.MustCompile(`\/photos\/.+?\/(\d+)`), regexp.MustCompile(`\/videos\/(\d+)`), } // PageData - facebook fan page data type PageData struct { ID string Text string Time int32 Link string } type byTime []*PageData func (pd byTime) Len() int { return len(pd) } func (pd byTime) Swap(i, j int) { pd[i], pd[j] = pd[j], pd[i] } func (pd byTime) Less(i, j int) bool { return pd[i].Time < pd[j].Time } func readFacebookPage() { pages, err := model.GetAllFacebookPage() if err != nil { return } for _, v := range pages { go getPageHTML(v) } } func getPageHTML(page *model.FacebookPage) { err := page.GetGroups() if err != nil { return } resp, err := http.Get(fmt.Sprintf("https://www.facebook.com/%s", page.ID)) if err != nil { return } defer resp.Body.Close() doc, err := goquery.NewDocumentFromReader(resp.Body) if err != nil { return } var pageData []*PageData doc.Find("div.userContentWrapper").Each(func(idx int, s *goquery.Selection) { timeEl := s.Find("abbr") time, timeExists := timeEl.Attr("data-utime") if !timeExists { return } link, linkExists := timeEl.Parent().Attr("href") if !linkExists { return } postContent := s.Find("div.userContent") text := postContent.Text() postID, idExists := postContent.First().Attr("id") if !idExists { idFlag := false for _, v := range idRegex { if v.MatchString(link) { idFlag = true m := v.FindStringSubmatch(link) postID = m[1] } } if !idFlag { return } } // fmt.Printf("Time: %s / Text: %s / ID: %s \n", time, text, postID) timeInt, err := strconv.ParseInt(time, 10, 32) if err != nil { return } re := regexp.MustCompile(`^\/`) pageLink := fmt.Sprintf("https://www.facebook.com/%s", re.ReplaceAllString(link, "")) data := &PageData{ ID: postID, Text: text, Time: int32(timeInt), Link: pageLink, } pageData = append(pageData, data) }) if len(pageData) == 0 { return } sort.Sort(sort.Reverse(byTime(pageData))) lastData := pageData[0] t := int32(time.Now().Unix()) if (t-600) < lastData.Time && lastData.ID != page.LastPost { err = page.UpdatePost(lastData.ID) if err != nil { return } for _, v := range page.Groups { if v.Notify { tmpl := v.Tmpl if len(tmpl) > 0 { tmpl = strings.Replace(tmpl, "{link}", lastData.Link, -1) tmpl = strings.Replace(tmpl, "{txt}", lastData.Text, -1) } else { tmpl = fmt.Sprintf("%s\n%s", lastData.Text, lastData.Link) } msg := line.TextMessage{ Text: tmpl, } line.PushMessage(v.ID, msg) } } } }