领取 v2ex 每日奖励。命令行参数1=用户名,命令行参数2=密码。 %v2ex_dialy username pwd

v2ex_daily.go

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"net/url"
	"os"
	"regexp"
	"strings"
)

var loginUrl = "http://v2ex.com/signin"
var missionUrl = "http://v2ex.com/mission/daily"

var username = os.Args[1]
var password = os.Args[2]

type myCookieJar struct {
	cookies []*http.Cookie
}

func (c *myCookieJar) SetCookies(u *url.URL, cookies []*http.Cookie) {
	if c.cookies == nil {
		c.cookies = make([]*http.Cookie, 0)
	}

	for _, it := range cookies {
		c.cookies = append(c.cookies, it)
	}
}

func (c *myCookieJar) Cookies(u *url.URL) []*http.Cookie {
	return c.cookies
}

func main() {
	cookieJar := &myCookieJar{}
	client := http.Client{Jar: cookieJar}
	// login
	loginreq, _ := http.NewRequest("GET", loginUrl, nil)
	loginresp, _ := client.Do(loginreq)
	loginHtml, _ := ioutil.ReadAll(loginresp.Body)
	re := regexp.MustCompile(`<input type="hidden" value="(\d+)" name="once" />`)
	matched := re.FindStringSubmatch(string(loginHtml))
	params := url.Values{"next": {"/", "/"}, "u": {username}, "p": {password}, "once": {matched[1]}}
	req, _ := http.NewRequest("POST", loginUrl, strings.NewReader(params.Encode()))
	req.Header.Add("Host", "v2ex.com")
	req.Header.Add("Origin", "http://v2ex.com")
	req.Header.Add("Referer", loginUrl)
	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
	for _, c := range cookieJar.cookies {
		req.AddCookie(c)
	}
	client.Do(req)

	// try finish mission
	missionReq, _ := http.NewRequest("GET", missionUrl, nil)
	missionResp, _ := client.Do(missionReq)
	missionHtml, _ := ioutil.ReadAll(missionResp.Body)
	missionHtmlStr := string(missionHtml)
	if strings.Index(missionHtmlStr, "每日登录奖励已领取") > 0 {
		fmt.Println("每日登录奖励已领取")
	} else {
		re = regexp.MustCompile(`location.href = '(.*)'`)
		matched = re.FindStringSubmatch(missionHtmlStr)
		req, _ = http.NewRequest("GET", "http://www.v2ex.com"+matched[1], nil)
		client.Do(req)
	}
}