学习笔记|用Go写一个简单的接口实现(任意聊天工具)主机上线通知

package main

import (
	"fmt"
	"github.com/go-vgo/robotgo"
	"log"
	"net/http"
	"time"
)

func sendmsg(msg string){
	robotgo.TypeString(msg)
	time.Sleep(time.Second)
	robotgo.KeyTap("enter")    //模拟按键
	time.Sleep(time.Second)
	robotgo.KeyTap("lctrl","enter")
}

func getmsg(w http.ResponseWriter, r *http.Request) {
	w.Header().Set("Access-Control-Allow-Origin", "*")             //允许访问所有域
	w.Header().Add("Access-Control-Allow-Headers", "Content-Type") //header的类
	r.ParseForm()
	var name string // 初始化定义变量
	var ip string // 初始化定义变量
	if r.Method == "GET" {
		name = r.FormValue("name")
		ip = r.FormValue("ip")
		//io.WriteString(w, "收到请求了！")
		if ip == ""{
			fmt.Fprintf(w, "请输入ip!")
		}else{
			if name != ""{
				fmt.Println("收到请求")
				fmt.Println("ip:"+ip+"\n主机名:"+name)
				sendmsg("有主机已上线！\n"+ "  ip:"+ip +"   主机名:"+name)
				fmt.Fprintf(w, "<h1>发送成功</h1>")
			}else {
				fmt.Fprintf(w, "请输入主机名!")
			}
		}
	}else {
		fmt.Fprintf(w, "Test")
	}

}

func main() {
	fmt.Println("接口服务启动成功！等待接收信息。")
	fmt.Println("Example:http://127.0.0.1:7777/?name=测试&ip=10.10.10.10")
	http.HandleFunc("/", getmsg)
	err := http.ListenAndServe(":7777", nil) //设置监听的端口
	if err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}

# 循环获取所有beacon
on beacon_initial {

	sub http_get {
		local('$output');
		$url = [new java.net.URL: $1];
		$stream = [$url openStream];
		$handle = [SleepUtils getIOHandle: $stream, $null];

		@content = readAll($handle);

		foreach $line (@content) {
			$output .= $line . "\r\n";
		}
		
		println($output);
	}
	#获取ip、计算机名、登录账号
	$internalIP = replace(beacon_info($1, "external"), " ", "_");
	$computerName = replace(beacon_info($1, "computer"), " ", "_");
	
	#get一下Server酱的链接
	$url = 'http://127.0.0.1:9878/?ip='.$internalIP.'&name='.$computerName;
	
	http_get($url);
	
}