// Package main provides ... // Get article' comments count package main import ( "encoding/json" "errors" "fmt" "io/ioutil" "net/http" "strings" "time" "github.com/deepzz0/logd" "github.com/eiblog/eiblog/setting" ) var ErrDisqusConfig = errors.New("disqus config incorrect") type result struct { Code int Response []struct { Id string Posts int Identifiers []string } } // 定时获取所有文章评论数量 func PostsCount() error { if setting.Conf.Disqus.PostsCount == "" || setting.Conf.Disqus.PublicKey == "" || setting.Conf.Disqus.ShortName == "" { return ErrDisqusConfig } time.AfterFunc(time.Duration(setting.Conf.Disqus.Interval)*time.Hour, func() { err := PostsCount() if err != nil { logd.Error(err) } }) baseUrl := setting.Conf.Disqus.PostsCount + "?api_key=" + setting.Conf.Disqus.PublicKey + "&forum=" + setting.Conf.Disqus.ShortName + "&" var count, index int for index < len(Ei.Articles) { var threads []string for ; index < len(Ei.Articles) && count < 50; index++ { artc := Ei.Articles[index] threads = append(threads, fmt.Sprintf("thread:ident=post-%s", artc.Slug)) count++ } count = 0 url := baseUrl + strings.Join(threads, "&") resp, err := http.Get(url) if err != nil { return err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) if err != nil { return err } if resp.StatusCode != http.StatusOK { return errors.New(string(b)) } rst := result{} err = json.Unmarshal(b, &rst) if err != nil { return err } for _, v := range rst.Response { i := strings.Index(v.Identifiers[0], "-") artc := Ei.MapArticles[v.Identifiers[0][i+1:]] if artc != nil { artc.Count = v.Posts artc.Thread = v.Id } } } return nil } type postsList struct { Cursor struct { HasNext bool Next string } Code int Response []struct { Parent int Id string CreatedAt string Message string Author struct { Name string ProfileUrl string Avatar struct { Cache string } } Thread string } } // 获取文章评论列表 func PostsList(slug, cursor string) (*postsList, error) { if setting.Conf.Disqus.PostsList == "" || setting.Conf.Disqus.PublicKey == "" || setting.Conf.Disqus.ShortName == "" { return nil, ErrDisqusConfig } url := setting.Conf.Disqus.PostsList + "?limit=50&api_key=" + setting.Conf.Disqus.PublicKey + "&forum=" + setting.Conf.Disqus.ShortName + "&cursor=" + cursor + "&thread:ident=post-" + slug resp, err := http.Get(url) if err != nil { return nil, err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) if err != nil { return nil, err } if resp.StatusCode != http.StatusOK { return nil, errors.New(string(b)) } pl := &postsList{} err = json.Unmarshal(b, pl) if err != nil { return nil, err } return pl, nil } type PostCreate struct { Message string `json:"message"` Parent string `json:"parent"` Thread string `json:"thread"` AuthorEmail string `json:"author_email"` AuthorName string `json:"autor_name"` IpAddress string `json:"ip_address"` Identifier string `json:"identifier"` UserAgent string `json:"user_agent"` } type PostResponse struct { Code int `json:"code"` Response struct { Id string `json:"id"` } `json:"response"` } // 评论文章 func PostComment(pc *PostCreate) (string, error) { if setting.Conf.Disqus.PostsList == "" || setting.Conf.Disqus.PublicKey == "" || setting.Conf.Disqus.ShortName == "" { return "", ErrDisqusConfig } url := setting.Conf.Disqus.PostCreate + "?api_key=E8Uh5l5fHZ6gD8U3KycjAIAk46f68Zw7C6eW8WSjZvCLXebZ7p0r1yrYDrLilk2F" + "&message=" + pc.Message + "&parent=" + pc.Parent + "&thread=" + pc.Thread + "&author_email=" + pc.AuthorEmail + "&author_name=" + pc.AuthorName request, err := http.NewRequest("POST", url, nil) if err != nil { return "", err } request.Header.Set("Referer", "https://disqus.com") resp, err := http.DefaultClient.Do(request) if err != nil { return "", err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) if err != nil { return "", err } if resp.StatusCode != http.StatusOK { return "", errors.New(string(b)) } pr := &PostResponse{} err = json.Unmarshal(b, pr) if err != nil { return "", err } return pr.Response.Id, nil } type ApprovedResponse struct { Code int `json:"code"` Response []struct { Id string `json:"id"` } `json:"response"` } // 批准评论通过 func PostApprove(post string) error { if setting.Conf.Disqus.PostsList == "" || setting.Conf.Disqus.PublicKey == "" || setting.Conf.Disqus.ShortName == "" { return ErrDisqusConfig } url := setting.Conf.Disqus.PostApprove + "?api_key=" + setting.Conf.Disqus.PublicKey + "&access_token=" + setting.Conf.Disqus.AccessToken + "&post=" + post request, err := http.NewRequest("POST", url, nil) if err != nil { return err } request.Header.Set("Referer", "https://disqus.com") resp, err := http.DefaultClient.Do(request) if err != nil { return err } defer resp.Body.Close() b, err := ioutil.ReadAll(resp.Body) if err != nil { return err } if resp.StatusCode != http.StatusOK { return errors.New(string(b)) } ar := &ApprovedResponse{} err = json.Unmarshal(b, ar) if err != nil { return err } return nil }