package main

import (
	"log"
	"os"
	"strconv"
	"time"

	"github.com/joho/godotenv"
	"github.com/urfave/cli/v2"
	"github.com/yassinebenaid/godump"
)

// Version set at compile-time
var Version string

func main() {
	// Load env-file if it exists first
	if filename, found := os.LookupEnv("PLUGIN_ENV_FILE"); found {
		_ = godotenv.Load(filename)
	}

	if _, err := os.Stat("/run/drone/env"); err == nil {
		_ = godotenv.Overload("/run/drone/env")
	}

	app := cli.NewApp()
	app.Name = "Drone Discord"
	app.Usage = "Sending message to Discord channel using Webhook"
	app.Copyright = "Copyright (c) " + strconv.Itoa(time.Now().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-url",
			Usage:   "The Discord webhook URL to send messages to.",
			EnvVars: []string{"PLUGIN_WEBHOOK_URL", "WEBHOOK_URL", "DISCORD_WEBHOOK_URL", "INPUT_WEBHOOK_URL"},
		},
		&cli.StringFlag{
			Name:    "webhook-id",
			Usage:   "The Discord webhook ID.",
			EnvVars: []string{"PLUGIN_WEBHOOK_ID", "WEBHOOK_ID", "DISCORD_WEBHOOK_ID", "INPUT_WEBHOOK_ID"},
		},
		&cli.StringFlag{
			Name:    "webhook-token",
			Usage:   "The Discord webhook token.",
			EnvVars: []string{"PLUGIN_WEBHOOK_TOKEN", "WEBHOOK_TOKEN", "DISCORD_WEBHOOK_TOKEN", "INPUT_WEBHOOK_TOKEN"},
		},
		&cli.StringSliceFlag{
			Name:    "message",
			Usage:   "The message contents to send to the Discord channel (up to 2000 characters).",
			EnvVars: []string{"PLUGIN_MESSAGE", "DISCORD_MESSAGE", "MESSAGE", "INPUT_MESSAGE"},
		},
		&cli.StringSliceFlag{
			Name:    "file",
			Usage:   "The contents of the file being sent to the Discord channel.",
			EnvVars: []string{"PLUGIN_FILE", "DISCORD_FILE", "FILE", "INPUT_FILE"},
		},
		&cli.StringFlag{
			Name:    "color",
			Usage:   "The color code of the embed message.",
			EnvVars: []string{"PLUGIN_COLOR", "COLOR", "INPUT_COLOR"},
		},
		&cli.BoolFlag{
			Name:    "wait",
			Usage:   "Wait for server confirmation of message send before response, and return the created message body.",
			EnvVars: []string{"PLUGIN_WAIT", "WAIT", "INPUT_WAIT"},
		},
		&cli.BoolFlag{
			Name:    "tts",
			Usage:   "Set to true if this is a Text-to-Speech (TTS) message.",
			EnvVars: []string{"PLUGIN_TTS", "TTS", "INPUT_TTS"},
		},
		&cli.StringFlag{
			Name:    "username",
			Usage:   "Override the default username of the webhook.",
			EnvVars: []string{"PLUGIN_USERNAME", "USERNAME", "INPUT_USERNAME"},
		},
		&cli.StringFlag{
			Name:    "avatar-url",
			Usage:   "Override the default avatar of the webhook.",
			EnvVars: []string{"PLUGIN_AVATAR_URL", "AVATAR_URL", "INPUT_AVATAR_URL"},
		},
		&cli.BoolFlag{
			Name:    "drone",
			Usage:   "Indicate if the environment is Drone CI.",
			EnvVars: []string{"DRONE"},
		},
		&cli.StringFlag{
			Name:    "ci.environment",
			Usage:   "The name of the CI environment.",
			EnvVars: []string{"CI"},
		},
		&cli.StringFlag{
			Name:    "repo",
			Usage:   "The repository owner and repository name.",
			EnvVars: []string{"DRONE_REPO", "CI_REPO", "GITHUB_REPOSITORY"},
		},
		&cli.StringFlag{
			Name:    "repo.namespace",
			Usage:   "The repository namespace.",
			EnvVars: []string{"DRONE_REPO_OWNER", "DRONE_REPO_NAMESPACE", "CI_REPO_OWNER", "GITHUB_ACTOR"},
		},
		&cli.StringFlag{
			Name:    "repo.name",
			Usage:   "The repository name.",
			EnvVars: []string{"DRONE_REPO_NAME", "CI_REPO_NAME"},
		},
		&cli.StringFlag{
			Name:    "commit.sha",
			Usage:   "The Git commit SHA.",
			EnvVars: []string{"DRONE_COMMIT_SHA", "CI_COMMIT_SHA", "GITHUB_SHA"},
		},
		&cli.StringFlag{
			Name:    "commit.ref",
			Usage:   "The Git commit reference.",
			EnvVars: []string{"DRONE_COMMIT_REF", "CI_COMMIT_REF", "GITHUB_REF"},
		},
		&cli.StringFlag{
			Name:    "commit.branch",
			Value:   "master",
			Usage:   "The Git commit branch.",
			EnvVars: []string{"DRONE_COMMIT_BRANCH", "CI_COMMIT_BRANCH"},
		},
		&cli.StringFlag{
			Name:    "commit.link",
			Usage:   "The link to the Git commit.",
			EnvVars: []string{"DRONE_COMMIT_LINK", "CI_PIPELINE_FORGE_URL"},
		},
		&cli.StringFlag{
			Name:    "commit.author",
			Usage:   "The name of the Git commit author.",
			EnvVars: []string{"DRONE_COMMIT_AUTHOR", "CI_COMMIT_AUTHOR"},
		},
		&cli.StringFlag{
			Name:    "commit.author.email",
			Usage:   "The email of the Git commit author.",
			EnvVars: []string{"DRONE_COMMIT_AUTHOR_EMAIL", "CI_COMMIT_AUTHOR_EMAIL"},
		},
		&cli.StringFlag{
			Name:    "commit.author.avatar",
			Usage:   "The avatar URL of the Git commit author.",
			EnvVars: []string{"DRONE_COMMIT_AUTHOR_AVATAR", "CI_COMMIT_AUTHOR_AVATAR"},
		},
		&cli.StringFlag{
			Name:    "commit.message",
			Usage:   "The Git commit message.",
			EnvVars: []string{"DRONE_COMMIT_MESSAGE", "CI_COMMIT_MESSAGE"},
		},
		&cli.StringFlag{
			Name:    "source.branch",
			Value:   "develop",
			Usage:   "The Git source branch.",
			EnvVars: []string{"DRONE_SOURCE_BRANCH", "CI_COMMIT_SOURCE_BRANCH"},
		},
		&cli.StringFlag{
			Name:    "build.event",
			Value:   "push",
			Usage:   "The build event type.",
			EnvVars: []string{"DRONE_BUILD_EVENT", "CI_PIPELINE_EVENT"},
		},
		&cli.IntFlag{
			Name:    "build.number",
			Usage:   "The build number.",
			EnvVars: []string{"DRONE_BUILD_NUMBER", "CI_PIPELINE_NUMBER"},
		},
		&cli.StringFlag{
			Name:    "build.status",
			Usage:   "The build status.",
			Value:   "success",
			EnvVars: []string{"DRONE_BUILD_STATUS", "CI_PIPELINE_STATUS"},
		},
		&cli.StringFlag{
			Name:    "build.link",
			Usage:   "The link to the build.",
			EnvVars: []string{"DRONE_BUILD_LINK", "CI_PIPELINE_URL"},
		},
		&cli.StringFlag{
			Name:    "build.tag",
			Usage:   "The build tag.",
			EnvVars: []string{"DRONE_TAG", "CI_COMMIT_TAG"},
		},
		&cli.StringFlag{
			Name:    "pull.request",
			Usage:   "The pull request number.",
			EnvVars: []string{"DRONE_PULL_REQUEST", "CI_COMMIT_PULL_REQUEST"},
		},
		&cli.Int64Flag{
			Name:    "build.started",
			Usage:   "The timestamp when the build started.",
			EnvVars: []string{"DRONE_BUILD_STARTED", "CI_PIPELINE_STARTED"},
		},
		&cli.Int64Flag{
			Name:    "build.finished",
			Usage:   "The timestamp when the build finished.",
			EnvVars: []string{"DRONE_BUILD_FINISHED", "CI_PIPELINE_FINISHED"},
		},
		&cli.BoolFlag{
			Name:    "github",
			Usage:   "Indicate if the runtime environment is GitHub Actions.",
			EnvVars: []string{"PLUGIN_GITHUB", "GITHUB"},
		},
		&cli.StringFlag{
			Name:    "github.workflow",
			Usage:   "The name of the GitHub Actions workflow.",
			EnvVars: []string{"GITHUB_WORKFLOW"},
		},
		&cli.StringFlag{
			Name:    "github.action",
			Usage:   "The name of the GitHub Actions action.",
			EnvVars: []string{"GITHUB_ACTION"},
		},
		&cli.StringFlag{
			Name:    "github.event.name",
			Usage:   "The name of the GitHub event that triggered the workflow.",
			EnvVars: []string{"GITHUB_EVENT_NAME"},
		},
		&cli.StringFlag{
			Name:    "github.event.path",
			Usage:   "The path to the file containing the payload of the event that triggered the workflow. Default: /github/workflow/event.json",
			EnvVars: []string{"GITHUB_EVENT_PATH"},
		},
		&cli.StringFlag{
			Name:    "github.workspace",
			Usage:   "The GitHub workspace path. Default: /github/workspace",
			EnvVars: []string{"GITHUB_WORKSPACE"},
		},
		&cli.StringFlag{
			Name:    "deploy.to",
			Usage:   "The target deployment environment for the running build. This value is only available to promotion and rollback pipelines.",
			EnvVars: []string{"DRONE_DEPLOY_TO", "CI_PIPELINE_DEPLOY_TARGET"},
		},
		&cli.BoolFlag{
			Name:    "debug",
			Usage:   "Enable debug mode.",
			EnvVars: []string{"PLUGIN_DEBUG", "INPUT_DEBUG", "DEBUG"},
		},
	}

	if err := app.Run(os.Args); err != nil {
		log.Fatal(err)
	}
}

func run(c *cli.Context) error {
	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"),
		},
		Source: Source{
			Branch: c.String("source.branch"),
		},
		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.Int64("build.started"),
			Finished: c.Int64("build.finished"),
			PR:       c.String("pull.request"),
			DeployTo: c.String("deploy.to"),
		},
		Config: Config{
			webhookURL:   c.String("webhook-url"),
			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") || c.String("ci.environment") == "woodpecker",
			GitHub:       c.Bool("github"),
			Debug:        c.Bool("debug"),
		},
		Payload: Payload{
			Wait:      c.Bool("wait"),
			Username:  c.String("username"),
			AvatarURL: c.String("avatar-url"),
			TTS:       c.Bool("tts"),
		},
	}

	if plugin.Config.Debug {
		_ = godump.Dump(plugin)
	}

	return plugin.Exec(c.Context)
}