go结合Slash Commands使用

潇洒

发布于 2023-10-23 15:01:38

1410

发布于 2023-10-23 15:01:38

文章被收录于专栏：石头岛石头岛

需求

通过 Slack 命令模式，调用操作远程服务器。服务端需要跟一个http服务来解析slash调过来的命令。

准备

准备以下步骤：

创建Slash conmand
开发服务端应用

添加 Slash Conmand

添加App

先到官网地址：https://api.slack.com/ 创建一个App

选Slash Conmand

输出一个自定义命令

到这就创建完成了，接下来开发服务端。

实现

创建 go 项目slacktool，添加：

main.go 文件
go.mod 文件
environment.env 文件

项目结构

slacktool
|---main.go
|---go.mod
|---environment.env

添加 go.mod 依赖

添加两个依赖： godotenv 和 slack。

module github.com/forfreeday/slacktool

go 1.17

require (
    github.com/joho/godotenv v1.4.0
    github.com/nlopes/slack v0.6.0
)

添加后执行:

go mod tidy

编辑 main.go

下面的功能包括：

运行一个http服务端
runConmand 运行逻辑
exec 执行shell命令

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"os/exec"

	"github.com/joho/godotenv"
	"github.com/nlopes/slack"
)

func main() {
    // 读取环境变量配置
	err := godotenv.Load("environment.env")
	if err != nil {
		log.Fatal("Error loading .env file")
	}
    // 具体处理逻辑
	http.HandleFunc("/deploy-test", slashCommandHandler)
	fmt.Println("[INFO] Server listening")
    // 启动 http
	http.ListenAndServe(":10001", nil)
}

func slashCommandHandler(w http.ResponseWriter, r *http.Request) {
	s, err := slack.SlashCommandParse(r)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
	fmt.Println("for test: invoke msg")
	if !s.ValidateToken(os.Getenv("SLACK_VERIFICATION_TOKEN")) {
		w.WriteHeader(http.StatusUnauthorized)
		return
	}

	switch s.Command {
	case "/deploy-test":
		params := &slack.Msg{Text: s.Text}
		response := fmt.Sprintf("Command params : %v", params.Text)
        // 开启协程，否则如果执行太久，slack端会返回 timeout
		go runConmand(params, w)
		w.Write([]byte(response))
	default:
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
}

// 执行命令
func runConmand(param *slack.Msg, w http.ResponseWriter) {
	switch param.Text {
	case "--restartAll":
		fmt.Println("restart All node")
		cmd := exec.Command("deploy-test", param.Text)
		_, err := cmd.CombinedOutput()
		if err != nil {
			log.Fatalf("cmd.Run() failed with %s\n", err)
		}
	case "--develop":
		fmt.Println("develop code")
	default:
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
}

服务端行一下，服务端可以多加一些日志，查看效果。