plugin-drone-dingtalk/plugin.go

package main

import (
	"errors"
	"fmt"
	"log"
	"strings"
)

type (
	//  repo base info
	Repo struct {
		Owner      string //  providers the repository owner name
		Name       string //  providers the repository name
		Branch     string //  providers the default repository branch(e.g.master)
		Link       string //  providers the repository http link
		NameSpace  string //  providers the repository namespace(e.g.account owner)
		Private    bool   //  indicates the repository is public or private
		Visibility string //  providers the repository visibility level.Possible values are public,private and internal
		SCM        string //  providers the repository version control system
		FullName   string //  repository full name
	}
	//  build info
	Build struct {
		Action  string  //  document description not found
		Created float64 //  providers the date and time when the build was created in the system
		Event   string  //  providers the current build event
		Number  int     //  providers the current build number
		Started float64 //  providers the date and time when the build was started
		Status  string  //  providers the current build status
		Link    string  //  providers the current build link
	}
	//  commit info
	Commit struct {
		After   string //  providers the commit sha for the current build
		Author  string //  providers the author username for the current commit
		Before  string //  providers the parent commit sha for the current build
		Branch  string //  providers the branch for the current commit
		Link    string //  providers the http link to the current commit in the remote source code management system(e.g.GitHub)
		Message string //  providers the commit message for the current build
		Ref     string //  providers the reference for the current build
		Sha     string //  providers the commit sha for the current build
		//  repo author info
		Authors struct {
			Avatar string //  providers the author avatar for the current commit
			Email  string //  providers the author email for the current commit
			Name   string //  providers the author name for the current commit
		}
	}
	//  git url info
	Git struct {
		HttpUrl string //  providers the repository git+http url
		SSHUrl  string //  providers the repository git+ssh url
	}
	//  Drone runner info
	Runner struct {
		Host     string //  providers the Drone agent hostname
		Hostname string //  providers the Drone agent hostname
		Platform string //  providers the Drone agent os and architecture
		Label    string //  document description not found
	}
	//  Drone system info
	System struct {
		Host     string //  providers the Drone server hostname
		Hostname string //  providers the Drone server hostname
		Version  string //  providers the Drone server version
	}
	//  plugin private config
	Config struct {
		AccessToken    string
		Message        string
		Lang           string
		IsAtALL        bool
		Mobiles        string
		Username       string
		AvatarURL      string
		MsgType        string
		LinkUrls       string
		LinkTitles     string
		HideAvatar     bool
		BtnOrientation bool
		PicURL         string
		MsgURL         string
	}
	// plugin all config
	Plugin struct {
		Git     Git
		Runner  Runner
		System  System
		Commit  Commit
		Repo    Repo
		Build   Build
		Config  Config
		WebHook *WebHook
	}
)

func (p *Plugin) Exec() error {
	log.Println("start execute sending...")
	if 0 == len(p.Config.AccessToken) {
		msg := "missing dingtalk access token"
		log.Println(msg)
		return errors.New(msg)
	}
	log.Println("access token pass...")
	p.WebHook = NewWebHook(p.Config.AccessToken)
	mobiles := strings.Split(p.Config.Mobiles, ",")
	linkUrls := strings.Split(p.Config.LinkUrls, ",")
	linkTitles := strings.Split(p.Config.LinkTitles, ",")
	log.Println("sending message type: " + p.Config.MsgType)
	switch strings.ToLower(p.Config.MsgType) {
	case "markdown":
		err := p.WebHook.SendMarkdownMsg(
			"You have a new message...",
			p.baseTpl(),
			p.Config.IsAtALL,
			mobiles...
		)
		if nil != err {
			log.Println(err)
			return err
		}
	case "text":
		err := p.WebHook.SendTextMsg(p.baseTpl(), p.Config.IsAtALL, mobiles...)
		if nil != err {
			log.Println(err)
			return err
		}
	case "actioncard":
		err := p.WebHook.SendActionCardMsg(
			"A actionCard title",
			p.baseTpl(),
			linkUrls,
			linkTitles,
			p.Config.HideAvatar,
			p.Config.BtnOrientation,
		)
		if nil != err {
			log.Println(err)
			return err
		}
	case "link":
		err := p.WebHook.SendLinkMsg(p.Build.Status, p.baseTpl(), p.Commit.Authors.Avatar, p.Build.Link)
		if nil != err {
			log.Println(err)
			return err
		}
	default:
		msg := "not support message type"
		log.Println(msg)
		return errors.New(msg)
	}
	log.Println("send " + p.Config.MsgType + " message success!")
	return nil
}

func (p *Plugin) baseTpl() string {
	tpl := ""
	switch strings.ToLower(p.Config.MsgType) {
	case "markdown":
		tpl = fmt.Sprintf(`# **%s** 
### [%s](%s)
##### %s (%s)
##### @%s
##### %s(%s)
`,
			strings.Title(p.Build.Status),
			strings.TrimSpace(p.Commit.Message),
			p.Build.Link,
			p.Repo.FullName,
			p.Commit.Branch,
			p.Commit.Sha,
			p.Commit.Authors.Name,
			p.Commit.Authors.Email)
	case "text":
		tpl = fmt.Sprintf(`[%s] %s
%s (%s)
@%s
%s (%s)
`,
			p.Build.Status,
			strings.TrimSpace(p.Commit.Message),
			p.Repo.FullName,
			p.Commit.Branch,
			p.Commit.Sha,
			p.Commit.Authors.Name,
			p.Commit.Authors.Email)
	case "link":
		tpl = fmt.Sprintf(`%s(%s) @%s %s(%s)`,
			p.Repo.FullName,
			p.Commit.Branch,
			p.Commit.Sha[:6],
			p.Commit.Authors.Name,
			p.Commit.Authors.Email)

	}

	return tpl
}