Exemplo n.º 1
0
func main() {
	defer time.Sleep(1 * time.Second)
	config := new(cfg)
	_, err := toml.DecodeFile(`config.toml`, config)

	if err != nil {
		log.Fatalln("read config error:", err)
	}

	client := webqq.New(config.Username, config.Password, config.DailTimeout, config.LogChannelLen) //qq Number and Password,obtain a client object
	if config.FileLog.Use {
		client.SetLogger("file", fmt.Sprintf("{\"level\":%d,\"filename\":\"%s\"}", config.FileLog.Level, config.FileLog.Filename))
	}
	if config.ConsoleLog.Use {
		client.SetLogger("console", fmt.Sprintf("{\"level\":%d}", config.ConsoleLog.Level))
	}

L:
	ret, err := client.LoginStep1() //test login to know if a validation image is necessary
	if err != nil {
		client.Error("login step 1 fail: error: %v", err)
		client.Info("relogin after 1 minute")
		time.Sleep(1 * time.Minute)
		goto L
	}

	if ret.NeedPic() {
		re, err := client.Get(ret.PicAddr) //get pic addr
		if err != nil {
			client.Error("get login pic fail: error: %v", err)
		}

		f, err := os.Create(config.VfCodeFile) //write image to file
		if err != nil {
			client.Critical("create vfcode pic fail: error: %v", err)
			return
		}
		io.Copy(f, re.Body)
		f.Close()

		re.Body.Close()

		trigger := strings.Fields(config.VfCodeTrigger)

		cmd := exec.Command(trigger[0], trigger[1:]...)
		out, _ := cmd.StdoutPipe()
		cmd.Stdin = os.Stdin

		s := ``
		go func() { fmt.Fscan(out, &s) }()

		err = cmd.Run()
		if err != nil {
			client.Critical("cannot run vf trigger: error: %v", err)
			return
		}

		ret.SetVFCode(s) //set validation code
	}

	//if ret.NeedPic()==false ret.SetVFCode() is unnecessary

	err = client.LoginStep2(ret) //true login
	if err != nil {
		client.Error("login step2 fail: error: %v", err)
		client.Info("wait 10 second to relogin")
		time.Sleep(10 * time.Second)
		client.Warn(`relogin`)
		goto L
	}

	for {
		msgraw, err := client.RawPoll() //qq use poll to get message.
		if err != nil {
			client.Error("poll message fail: error: %v", err)
			client.Warn("relogin")
			goto L
		}
		msg, err := webqq.ParseRawPoll(msgraw)
		if err != nil {
			client.Error("parse raw poll fail: error: %v", err)
			continue
		}

		for _, v := range msg {
			if v.IsNothing() {
				continue
			}

			if m, ok := v.IsBuddyMessage(); ok {

				err := client.SendBuddyMsgEasy(m.FromUin, m.MessageId, "你好"+m.Content[0])
				if err != nil {
					fmt.Println(err)
				}
				gid, err := client.GetId(m.FromUin)
				client.Info("收到个人消息: %v ,id: %v", m.Content, gid)
				continue
			}
			if _, ok := v.IsBuddyStatusChange(); ok {
				continue
			}
			if m, ok := v.IsGroupMessage(); ok {
				err = client.SendGroupMsgEasy(m.FromUin, 55440000, "机器人重复说:"+m.Content[0])
				if err != nil {
					client.Error("reply group message fail: error: %v", err)
				}

				gid, err := client.GetId(m.GroupCode)
				if err != nil {
					client.Warn("get gid fail %v", err)
				}
				id, err := client.GetId(m.FromUin)
				if err != nil {
					client.Warn("get id fail %v", err)
				}
				client.Info("收到群消息: %v, 群号: %v 来自号码: %v", m.Content, gid, id)
				continue
			}
			if ptwebqq, ok := v.IsNewPtwebqq(); ok {
				client.SetPtWebqq(ptwebqq)
				continue
			}
			if ok := v.IsKick(); ok {
				client.Warn("你被踢下线了")
				if config.KickRelogin == true {
					goto L
				}
				return
			}
			if _, ok := v.IsSysGMessage(); ok {
				continue
			}
			if _, ok := v.IsTips(); ok {
				continue
			}
			if _, ok := v.IsGroupWebMessage(); ok {
				continue
			}
			if _, ok := v.IsSystemMessage(); ok {
				continue
			}
			if _, ok := v.IsBuddylistChange(); ok {
				continue
			}
			client.Warn("unexpected message: %v", v)
		}

	}

}
Exemplo n.º 2
0
func main() {

	cl := webqq.New(`173165159`, `R9:T4K6@`, 1000, 20000000) //qq Number and Password,obtain a client object
L:
	ret, err := cl.LoginStep1() //test login to know if a validation image is necessary
	if err != nil {
		log.Fatalln(err)

	}
	if ret.NeedPic() {
		re, err := cl.Get(ret.PicAddr) //get pic addr
		if err != nil {
			log.Fatalln(err)

		}

		f, err := os.Create(`cool.png`) //write image to file
		if err != nil {
			log.Fatalln(err)

		}
		io.Copy(f, re.Body)
		f.Close()

		re.Body.Close()

		exec.Command(`gwenview`, `cool.png`).Run() //show the image
		var s string
		fmt.Scanf("%s", &s) //recognize it
		ret.SetVFCode(s)    //set validation code
	}

	//if ret.NeedPic()==false ret.SetVFCode() is unnecessary

	err = cl.LoginStep2(ret) //true login
	if err != nil {
		time.Sleep(10 * time.Second)
		goto L

	}

	for {
		msgraw, err := cl.RawPoll() //qq use poll to get message.
		if err != nil {
			log.Fatalln(err)
		}

		msg, err := webqq.ParseRawPoll(msgraw)
		if err != nil {
			log.Fatalln(err)
		}
		for _, v := range msg {
			if v.IsNothing() {
				continue
			}

			if _, ok := v.IsBuddyMessage(); ok {
				// fmt.Printf("%+v\n", bmsg)
				// err := cl.SendBuddyMsgEasy(bmsg.FromUin, bmsg.MessageId+1, "got"+bmsg.Content[0])
				// if err != nil {
				// 	fmt.Println(err)
				// }
				continue
			}
			if _, ok := v.IsBuddyStatusChange(); ok {
				continue
			}
			if _, ok := v.IsGroupMessage(); ok {
				// err := cl.SendGroupMsgEasy(gmsg.FromUin, gmsg.MessageId, "fa")
				// if err != nil {
				// 	fmt.Println(err)
				// }
				// fmt.Printf("%+v\n", gmsg)
				continue
			}
			if ptwebqq, ok := v.IsNewPtwebqq(); ok {
				cl.SetPtWebqq(ptwebqq)
				continue
			}
			if ok := v.IsKick(); ok {
				return
			}
			if s, ok := v.IsSysGMessage(); ok {
				fmt.Printf("%+v\n", s)
				continue
			}
			if tips, ok := v.IsTips(); ok {
				fmt.Println("tips", tips)
				continue
			}
			if gwm, ok := v.IsGroupWebMessage(); ok {
				fmt.Println("gwm", gwm)
				continue
			}
			if gwm, ok := v.IsSystemMessage(); ok {
				fmt.Printf("%+v\n", gwm)
				continue
			}
			if gwm, ok := v.IsBuddylistChange(); ok {
				fmt.Println("blc", gwm)
				continue
			}
			fmt.Println(v)
		}

	}

}