mirror of
https://github.com/appleboy/drone-discord.git
synced 2026-06-04 18:33:47 +08:00
Bo-Yi Wu
284 lines
7.2 KiB
Go
284 lines
7.2 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
_ "github.com/joho/godotenv/autoload"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
// Version set at compile-time
|
|
var Version string
|
|
|
|
func main() {
|
|
year := fmt.Sprintf("%v", time.Now().Year())
|
|
app := cli.NewApp()
|
|
app.Name = "Drone Discord"
|
|
app.Usage = "Sending message to Discord channel using Webhook"
|
|
app.Copyright = "Copyright (c) " + year + " Bo-Yi Wu"
|
|
app.Authors = []cli.Author{
|
|
{
|
|
Name: "Bo-Yi Wu",
|
|
Email: "appleboy.tw@gmail.com",
|
|
},
|
|
}
|
|
app.Action = run
|
|
app.Version = Version
|
|
app.Flags = []cli.Flag{
|
|
cli.StringFlag{
|
|
Name: "webhook-id",
|
|
Usage: "discord webhook id",
|
|
EnvVar: "PLUGIN_WEBHOOK_ID,WEBHOOK_ID,DISCORD_WEBHOOK_ID,INPUT_WEBHOOK_ID",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "webhook-token",
|
|
Usage: "discord webhook token",
|
|
EnvVar: "PLUGIN_WEBHOOK_TOKEN,WEBHOOK_TOKEN,DISCORD_WEBHOOK_TOKEN,INPUT_WEBHOOK_TOKEN",
|
|
},
|
|
cli.StringSliceFlag{
|
|
Name: "message",
|
|
Usage: "the message contents (up to 2000 characters)",
|
|
EnvVar: "PLUGIN_MESSAGE,DISCORD_MESSAGE,MESSAGE,INPUT_MESSAGE",
|
|
},
|
|
cli.StringSliceFlag{
|
|
Name: "file",
|
|
Usage: "the contents of the file being sent",
|
|
EnvVar: "PLUGIN_FILE,DISCORD_FILE,FILE,INPUT_FILE",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "color",
|
|
Usage: "color code of the embed",
|
|
EnvVar: "PLUGIN_COLOR,COLOR,INPUT_COLOR",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "wait",
|
|
Usage: "waits for server confirmation of message send before response, and returns the created message body",
|
|
EnvVar: "PLUGIN_WAIT,WAIT,INPUT_WAIT",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "tts",
|
|
Usage: "true if this is a TTS message",
|
|
EnvVar: "PLUGIN_TTS,TTS,INPUT_TTS",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "username",
|
|
Usage: "override the default username of the webhook",
|
|
EnvVar: "PLUGIN_USERNAME,USERNAME,INPUT_USERNAME",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "avatar-url",
|
|
Usage: "override the default avatar of the webhook",
|
|
EnvVar: "PLUGIN_AVATAR_URL,AVATAR_URL,INPUT_AVATAR_URL",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "drone",
|
|
Usage: "environment is drone",
|
|
EnvVar: "DRONE",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo",
|
|
Usage: "repository owner and repository name",
|
|
EnvVar: "DRONE_REPO,GITHUB_REPOSITORY",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.namespace",
|
|
Usage: "repository namespace",
|
|
EnvVar: "DRONE_REPO_OWNER,DRONE_REPO_NAMESPACE,GITHUB_ACTOR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "repo.name",
|
|
Usage: "repository name",
|
|
EnvVar: "DRONE_REPO_NAME",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.sha",
|
|
Usage: "git commit sha",
|
|
EnvVar: "DRONE_COMMIT_SHA,GITHUB_SHA",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.ref",
|
|
Usage: "git commit ref",
|
|
EnvVar: "DRONE_COMMIT_REF,GITHUB_REF",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.branch",
|
|
Value: "master",
|
|
Usage: "git commit branch",
|
|
EnvVar: "DRONE_COMMIT_BRANCH",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.link",
|
|
Usage: "git commit link",
|
|
EnvVar: "DRONE_COMMIT_LINK",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author",
|
|
Usage: "git author name",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author.email",
|
|
Usage: "git author email",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR_EMAIL",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.author.avatar",
|
|
Usage: "git author avatar",
|
|
EnvVar: "DRONE_COMMIT_AUTHOR_AVATAR",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "commit.message",
|
|
Usage: "commit message",
|
|
EnvVar: "DRONE_COMMIT_MESSAGE",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.event",
|
|
Value: "push",
|
|
Usage: "build event",
|
|
EnvVar: "DRONE_BUILD_EVENT",
|
|
},
|
|
cli.IntFlag{
|
|
Name: "build.number",
|
|
Usage: "build number",
|
|
EnvVar: "DRONE_BUILD_NUMBER",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.status",
|
|
Usage: "build status",
|
|
Value: "success",
|
|
EnvVar: "DRONE_BUILD_STATUS",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.link",
|
|
Usage: "build link",
|
|
EnvVar: "DRONE_BUILD_LINK",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "build.tag",
|
|
Usage: "build tag",
|
|
EnvVar: "DRONE_TAG",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "pull.request",
|
|
Usage: "pull request",
|
|
EnvVar: "DRONE_PULL_REQUEST",
|
|
},
|
|
cli.Float64Flag{
|
|
Name: "job.started",
|
|
Usage: "job started",
|
|
EnvVar: "DRONE_BUILD_STARTED",
|
|
},
|
|
cli.Float64Flag{
|
|
Name: "job.finished",
|
|
Usage: "job finished",
|
|
EnvVar: "DRONE_BUILD_FINISHED",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "env-file",
|
|
Usage: "source env file",
|
|
},
|
|
cli.BoolFlag{
|
|
Name: "github",
|
|
Usage: "Boolean value, indicates the runtime environment is GitHub Action.",
|
|
EnvVar: "PLUGIN_GITHUB,GITHUB",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "github.workflow",
|
|
Usage: "The name of the workflow.",
|
|
EnvVar: "GITHUB_WORKFLOW",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "github.action",
|
|
Usage: "The name of the action.",
|
|
EnvVar: "GITHUB_ACTION",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "github.event.name",
|
|
Usage: "The webhook name of the event that triggered the workflow.",
|
|
EnvVar: "GITHUB_EVENT_NAME",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "github.event.path",
|
|
Usage: "The path to a file that contains the payload of the event that triggered the workflow. Value: /github/workflow/event.json.",
|
|
EnvVar: "GITHUB_EVENT_PATH",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "github.workspace",
|
|
Usage: "The GitHub workspace path. Value: /github/workspace.",
|
|
EnvVar: "GITHUB_WORKSPACE",
|
|
},
|
|
cli.StringFlag{
|
|
Name: "deploy.to",
|
|
Usage: "Provides the target deployment environment for the running build. This value is only available to promotion and rollback pipelines.",
|
|
EnvVar: "DRONE_DEPLOY_TO",
|
|
},
|
|
}
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func run(c *cli.Context) error {
|
|
if c.String("env-file") != "" {
|
|
_ = godotenv.Load(c.String("env-file"))
|
|
}
|
|
|
|
plugin := Plugin{
|
|
GitHub: GitHub{
|
|
Workflow: c.String("github.workflow"),
|
|
Workspace: c.String("github.workspace"),
|
|
Action: c.String("github.action"),
|
|
EventName: c.String("github.event.name"),
|
|
EventPath: c.String("github.event.path"),
|
|
},
|
|
Repo: Repo{
|
|
FullName: c.String("repo"),
|
|
Namespace: c.String("repo.namespace"),
|
|
Name: c.String("repo.name"),
|
|
},
|
|
Commit: Commit{
|
|
Sha: c.String("commit.sha"),
|
|
Ref: c.String("commit.ref"),
|
|
Branch: c.String("commit.branch"),
|
|
Link: c.String("commit.link"),
|
|
Author: c.String("commit.author"),
|
|
Email: c.String("commit.author.email"),
|
|
Avatar: c.String("commit.author.avatar"),
|
|
Message: c.String("commit.message"),
|
|
},
|
|
Build: Build{
|
|
Tag: c.String("build.tag"),
|
|
Number: c.Int("build.number"),
|
|
Event: c.String("build.event"),
|
|
Status: c.String("build.status"),
|
|
Link: c.String("build.link"),
|
|
Started: c.Float64("job.started"),
|
|
Finished: c.Float64("job.finished"),
|
|
PR: c.String("pull.request"),
|
|
DeployTo: c.String("deploy.to"),
|
|
},
|
|
Config: Config{
|
|
WebhookID: c.String("webhook-id"),
|
|
WebhookToken: c.String("webhook-token"),
|
|
Message: c.StringSlice("message"),
|
|
File: c.StringSlice("file"),
|
|
Color: c.String("color"),
|
|
Drone: c.Bool("drone"),
|
|
GitHub: c.Bool("github"),
|
|
},
|
|
Payload: Payload{
|
|
Wait: c.Bool("wait"),
|
|
Username: c.String("username"),
|
|
AvatarURL: c.String("avatar-url"),
|
|
TTS: c.Bool("tts"),
|
|
},
|
|
}
|
|
|
|
return plugin.Exec()
|
|
}
|