login.go

package main

import (
	"bytes"
	"encoding/json"
	"encoding/xml"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"strconv"
	"strings"
	"time"

	"github.com/skratchdot/open-golang/open"
)

func getUUID() (uuid string, err error) {
	jsloginUrl := "https://login.weixin.qq.com/jslogin"
	params := url.Values{}
	params.Set("appid", "wx782c26e4c19acffb")
	params.Set("fun", "new")
	params.Set("lang", "zh_CN")
	params.Set("_", strconv.FormatInt(time.Now().Unix(), 10))

	resp, err := Client.PostForm(jsloginUrl, params)
	if err != nil {
		return
	}
	defer resp.Body.Close()

	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return
	}
	ds := string(data)

	if *Debug {
		log.Printf("[debug] uuid:[%s]\n", ds)
	}

	code, err := findData(ds, "window.QRLogin.code = ", ";")
	if err != nil {
		return
	}
	if code != "200" {
		err = fmt.Errorf("错误的状态码:[%s], data:[%s]", code, ds)
		return
	}

	uuid, err = findData(ds, `window.QRLogin.uuid = "`, `";`)
	if err != nil {
		return
	}

	return
}

func showQRImage(uuid string) (err error) {
	qrUrl := `https://login.weixin.qq.com/qrcode/` + uuid
	params := url.Values{}
	params.Set("t", "webwx")
	params.Set("_", strconv.FormatInt(time.Now().Unix(), 10))

	req, err := http.NewRequest("POST", qrUrl, strings.NewReader(params.Encode()))
	if err != nil {
		return
	}

	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	req.Header.Set("Cache-Control", "no-cache")
	resp, err := Client.Do(req)
	if err != nil {
		return
	}
	defer resp.Body.Close()

	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return
	}

	if err = createFile(QRImagePath, data, false); err != nil {
		return
	}

	return open.Start(QRImagePath)
}

func waitForLogin(uuid string, tip int) (redirectUri, code string, rt int, err error) {
	loginUrl, rt := fmt.Sprintf("https://login.weixin.qq.com/cgi-bin/mmwebwx-bin/login?tip=%d&uuid=%s&_=%s", tip, uuid, time.Now().Unix()), tip
	resp, err := Client.Get(loginUrl)
	if err != nil {
		return
	}
	defer resp.Body.Close()

	data, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return
	}
	ds := string(data)

	if *Debug {
		log.Printf("[debug] wait for login:[%s]\n", ds)
	}

	code, err = findData(ds, "window.code=", ";")
	if err != nil {
		return
	}

	rt = 0
	switch code {
	case "201":
		log.Println("成功扫描,请在手机上点击确认以登录")
	case "200":
		redirectUri, err = findData(ds, `window.redirect_uri="`, `";`)
		if err != nil {
			return
		}
		redirectUri += "&fun=new"
	case "408":
	case "0":
		err = fmt.Errorf("超时，请重新运行程序")
	default:
		err = fmt.Errorf("未知错误，请重试")
	}

	return
}

func login(redirectUri string) (bReq *BaseRequest, err error) {
	resp, err := Client.Get(redirectUri)
	if err != nil {
		return
	}
	defer resp.Body.Close()

	reader := resp.Body.(io.Reader)
	if *Debug {
		var data []byte
		data, err = ioutil.ReadAll(reader)
		if err != nil {
			return
		}
		log.Printf("[debug] login:[%s]\n", string(data))
		reader = bytes.NewReader(data)
	}

	bReq = new(BaseRequest)
	if err = xml.NewDecoder(reader).Decode(bReq); err != nil {
		return
	}

	if bReq.Ret != Success {
		err = fmt.Errorf("message:[%s]", bReq.Message)
		return
	}

	bReq.DeviceID = *DeviceId
	return
}

func webwxInit(baseUri string, bReq *BaseRequest) (err error) {
	br := Request{
		BaseRequest: bReq,
	}
	data, err := json.Marshal(br)
	if err != nil {
		return
	}

	name, resp := "webwxinit", new(InitResp)
	apiUri := fmt.Sprintf("%s/%s?pass_ticket=%s&skey=%s&r=%s", baseUri, name, bReq.PassTicket, bReq.Skey, time.Now().Unix())
	if err = send(apiUri, name, bytes.NewReader(data), resp); err != nil {
		return
	}

	Myself = resp.User.UserName
	return
}