package main

import (
    "os/exec"
    "context"
    "google.golang.org/api/option"
    "cloud.google.com/go/pubsub"
    "encoding/json"
    "fmt"
)

const ProjectName string = ""        // GCP プロジェクト名
const KeyPath string = ""            // 鍵ファイルパス
const SubscriptionName string = ""   // Pub/Subのサブスクリプション名
const RepositoryName string = ""     // ビルドを実行するリポジトリ名

// 成功時に実行するコマンド
const SuccessCommand string = "../fly.exe"
// 失敗時に実行するコマンド
const ErrorCommand string = "../panic.exe"

type BuildMessage struct {
    Id        string `json:"id"`
    ProjectId string `json:"projectId"`
    Status    string `json:"status"`
    Source    Source `json:"source"`
}
type Source struct {
    RepoSource RepoSource `json:"repoSource"`
}
type RepoSource struct {
    ProjectId  string `json:"projectId"`
    RepoName   string `json:"repoName"`
    BranchName string `json:"branchName"`
}

func main() {
    // pubsub clientの呼び出し
    ctx := context.Background()
    client, _ := pubsub.NewClient(
        ctx,
        ProjectName,
        option.WithCredentialsFile(KeyPath),
    )
    sub := client.Subscription(SubscriptionName)
    // メッセージの受信待ち受け処理
    sub.Receive(context.Background(), MessageHandler)
}

// メッセージ受信時の処理
func MessageHandler(ctx context.Context, m *pubsub.Message) {
    var message BuildMessage
    json.Unmarshal(m.Data, &message)

    // 受信確認
    m.Ack()

    // 意図したリポジトリかどうか
    if message.Source.RepoSource.RepoName != RepositoryName {
        fmt.Println("飛ばない")
        return
    }

    if message.Status == "SUCCESS" {
        fmt.Println("飛べ！！")
        exec.Command(SuccessCommand).Run()
    }

    if message.Status == "FAILURE" {
        // 失敗処理
        fmt.Println("パニック")
        exec.Command(ErrorCommand).Run()
    }
}