Skip to content

Commit

Permalink
chore: use exec
Browse files Browse the repository at this point in the history
  • Loading branch information
@acid-chicken
acid-chicken committed May 22, 2024
1 parent c99b0dd commit f03cfe1
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 774 deletions.
114 changes: 15 additions & 99 deletions hariko/cmd/root.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,35 +2,22 @@ package cmd

import (
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
"os"
"path/filepath"
"strconv"
"os/exec"
"strings"

"github.com/bwmarrin/discordgo"
"github.com/go-playground/webhooks/v6/github"
"github.com/spf13/cobra"
"helm.sh/helm/v3/pkg/action"
"helm.sh/helm/v3/pkg/chart/loader"
"helm.sh/helm/v3/pkg/cli"
"helm.sh/helm/v3/pkg/getter"
"helm.sh/helm/v3/pkg/release"
"helm.sh/helm/v3/pkg/repo"
"helm.sh/helm/v3/pkg/storage/driver"
"sigs.k8s.io/yaml"
)

const (
ColorSuccess = 0x3fb950
ColorFailure = 0xf85149
)

var settings = cli.New()

func newCmd() *cobra.Command {
var discordWebhookIDToken string
var githubJobName string
Expand All @@ -46,7 +33,6 @@ func newCmd() *cobra.Command {
Long: "Hariko watches the GitHub repository and automatically deploys the application to the server.",
Args: cobra.NoArgs,
RunE: func(cmd *cobra.Command, args []string) error {
settings.SetNamespace(namespace)
hook, err := github.New(github.Options.Secret(githubWebhookSecret))
if err != nil {
return err
Expand Down Expand Up @@ -118,7 +104,7 @@ func newCmd() *cobra.Command {
},
}, nil)
b := new(bytes.Buffer)
release, err := deploy(packageName, repositoryName, repositoryURL, b)
err := deploy(namespace, packageName, repositoryName, repositoryURL, b)
if err != nil {
discord(&discordgo.WebhookParams{
Embeds: []*discordgo.MessageEmbed{
Expand All @@ -137,23 +123,6 @@ func newCmd() *cobra.Command {
Title: "Deployment succeeded",
Color: ColorSuccess,
Description: "```\n" + b.String() + "```",
Fields: []*discordgo.MessageEmbedField{
{
Name: "Name",
Value: release.Name,
Inline: true,
},
{
Name: "Namespace",
Value: release.Namespace,
Inline: true,
},
{
Name: "Revision",
Value: strconv.Itoa(release.Version),
Inline: true,
},
},
},
},
}, st)
Expand Down Expand Up @@ -188,73 +157,20 @@ func Execute() {
}
}

func deploy(packageName string, repositoryName string, repositoryURL string, log io.Writer) (*release.Release, error) {
p := getter.All(settings)
c := repo.Entry{
Name: repositoryName,
URL: repositoryURL,
}
r, err := repo.NewChartRepository(&c, p)
if err != nil {
return nil, err
}
if settings.RepositoryCache != "" {
r.CachePath = settings.RepositoryCache
}
index, err := r.DownloadIndexFile()
if err != nil {
return nil, err
}
b, _ := os.ReadFile(settings.RepositoryConfig)
re := repo.NewFile()
if b != nil {
yaml.Unmarshal(b, &re)
}
re.Update(&c)
err = os.MkdirAll(filepath.Dir(settings.RepositoryConfig), os.ModePerm)
if err != nil && !os.IsExist(err) {
return nil, err
func deploy(namespace string, packageName string, repositoryName string, repositoryURL string, log io.Writer) error {
if err := run(exec.Command("helm", "repo", "add", "-n", namespace, repositoryName, repositoryURL), log); err != nil {
return err
}
if err := re.WriteFile(settings.RepositoryConfig, 0600); err != nil {
return nil, err
if err := run(exec.Command("helm", "repo", "update"), log); err != nil {
return err
}
if _, err := repo.LoadIndexFile(index); err != nil {
return nil, err
if err := run(exec.Command("helm", "upgrade", "-n", namespace, packageName, repositoryName+"/"+packageName), log); err != nil {
return err
}
actionConfig := new(action.Configuration)
actionConfig.Init(settings.RESTClientGetter(), settings.Namespace(), "configmap", func(format string, v ...interface{}) {
fmt.Fprintf(log, format+"\n", v...)
})
if _, ok := actionConfig.Releases.Driver.(*driver.Memory); ok {
return nil, fmt.Errorf("storage driver could not be initialized")
} else {
releases, err := actionConfig.Releases.Driver.List(func(_ *release.Release) bool { return true })
if err != nil {
return nil, err
}
if releases == nil {
log.Write([]byte("no releases found\n"))
}
for _, release := range releases {
data, err := json.Marshal(release)
if err != nil {
log.Write([]byte(fmt.Sprintf("failed to marshal release: %s\n", err.Error())))
} else {
log.Write([]byte("detected release: "))
log.Write(data)
log.Write([]byte("\n"))
}
}
}
client := action.NewUpgrade(actionConfig)
client.Namespace = settings.Namespace()
chartPath, err := client.LocateChart(repositoryName+"/"+packageName, settings)
if err != nil {
return nil, err
}
ch, err := loader.Load(chartPath)
if err != nil {
return nil, err
}
return client.Run(repositoryName, ch, map[string]interface{}{})
}

func run(cmd *exec.Cmd, log io.Writer) error {
cmd.Stdout = log
cmd.Stderr = log
return cmd.Run()
}
130 changes: 1 addition & 129 deletions hariko/go.mod
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,142 +6,14 @@ require (
github.com/bwmarrin/discordgo v0.28.1
github.com/go-playground/webhooks/v6 v6.3.0
github.com/spf13/cobra v1.8.0
helm.sh/helm/v3 v3.15.0
sigs.k8s.io/yaml v1.4.0
)

require (
github.com/AdaLogics/go-fuzz-headers v0.0.0-20230811130428-ced1acdcaa24 // indirect
github.com/Azure/go-ansiterm v0.0.0-20210617225240-d185dfc1b5a1 // indirect
github.com/BurntSushi/toml v1.3.2 // indirect
github.com/MakeNowJust/heredoc v1.0.0 // indirect
github.com/Masterminds/goutils v1.1.1 // indirect
github.com/Masterminds/semver/v3 v3.2.1 // indirect
github.com/Masterminds/sprig/v3 v3.2.3 // indirect
github.com/Masterminds/squirrel v1.5.4 // indirect
github.com/Microsoft/hcsshim v0.11.4 // indirect
github.com/asaskevich/govalidator v0.0.0-20200428143746-21a406dcc535 // indirect
github.com/beorn7/perks v1.0.1 // indirect
github.com/cespare/xxhash/v2 v2.2.0 // indirect
github.com/chai2010/gettext-go v1.0.2 // indirect
github.com/containerd/containerd v1.7.12 // indirect
github.com/containerd/log v0.1.0 // indirect
github.com/cyphar/filepath-securejoin v0.2.4 // indirect
github.com/davecgh/go-spew v1.1.1 // indirect
github.com/docker/cli v24.0.6+incompatible // indirect
github.com/docker/distribution v2.8.2+incompatible // indirect
github.com/docker/docker v24.0.9+incompatible // indirect
github.com/docker/docker-credential-helpers v0.7.0 // indirect
github.com/docker/go-connections v0.4.0 // indirect
github.com/docker/go-metrics v0.0.1 // indirect
github.com/docker/go-units v0.5.0 // indirect
github.com/emicklei/go-restful/v3 v3.11.0 // indirect
github.com/evanphx/json-patch v5.7.0+incompatible // indirect
github.com/exponent-io/jsonpath v0.0.0-20151013193312-d6023ce2651d // indirect
github.com/fatih/color v1.13.0 // indirect
github.com/felixge/httpsnoop v1.0.3 // indirect
github.com/go-errors/errors v1.4.2 // indirect
github.com/go-gorp/gorp/v3 v3.1.0 // indirect
github.com/go-logr/logr v1.4.1 // indirect
github.com/go-logr/stdr v1.2.2 // indirect
github.com/go-openapi/jsonpointer v0.19.6 // indirect
github.com/go-openapi/jsonreference v0.20.2 // indirect
github.com/go-openapi/swag v0.22.3 // indirect
github.com/gobwas/glob v0.2.3 // indirect
github.com/gogo/protobuf v1.3.2 // indirect
github.com/golang/protobuf v1.5.4 // indirect
github.com/google/btree v1.0.1 // indirect
github.com/google/gnostic-models v0.6.8 // indirect
github.com/google/go-cmp v0.6.0 // indirect
github.com/google/gofuzz v1.2.0 // indirect
github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510 // indirect
github.com/google/uuid v1.3.0 // indirect
github.com/gorilla/mux v1.8.0 // indirect
github.com/gorilla/websocket v1.5.1 // indirect
github.com/gosuri/uitable v0.0.4 // indirect
github.com/gregjones/httpcache v0.0.0-20180305231024-9cad4c3443a7 // indirect
github.com/hashicorp/errwrap v1.1.0 // indirect
github.com/hashicorp/go-multierror v1.1.1 // indirect
github.com/huandu/xstrings v1.4.0 // indirect
github.com/imdario/mergo v0.3.13 // indirect
github.com/inconshreveable/mousetrap v1.1.0 // indirect
github.com/jmoiron/sqlx v1.3.5 // indirect
github.com/josharian/intern v1.0.0 // indirect
github.com/json-iterator/go v1.1.12 // indirect
github.com/klauspost/compress v1.16.0 // indirect
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 // indirect
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 // indirect
github.com/lib/pq v1.10.9 // indirect
github.com/liggitt/tabwriter v0.0.0-20181228230101-89fcab3d43de // indirect
github.com/mailru/easyjson v0.7.7 // indirect
github.com/mattn/go-colorable v0.1.13 // indirect
github.com/mattn/go-isatty v0.0.17 // indirect
github.com/mattn/go-runewidth v0.0.9 // indirect
github.com/matttproud/golang_protobuf_extensions v1.0.4 // indirect
github.com/mitchellh/copystructure v1.2.0 // indirect
github.com/mitchellh/go-wordwrap v1.0.1 // indirect
github.com/mitchellh/reflectwalk v1.0.2 // indirect
github.com/moby/locker v1.0.1 // indirect
github.com/moby/spdystream v0.2.0 // indirect
github.com/moby/term v0.5.0 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // indirect
github.com/monochromegane/go-gitignore v0.0.0-20200626010858-205db1a8cc00 // indirect
github.com/morikuni/aec v1.0.0 // indirect
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
github.com/mxk/go-flowrate v0.0.0-20140419014527-cca7078d478f // indirect
github.com/opencontainers/go-digest v1.0.0 // indirect
github.com/opencontainers/image-spec v1.1.0-rc5 // indirect
github.com/peterbourgon/diskv v2.0.1+incompatible // indirect
github.com/pkg/errors v0.9.1 // indirect
github.com/prometheus/client_golang v1.16.0 // indirect
github.com/prometheus/client_model v0.4.0 // indirect
github.com/prometheus/common v0.44.0 // indirect
github.com/prometheus/procfs v0.10.1 // indirect
github.com/rubenv/sql-migrate v1.5.2 // indirect
github.com/russross/blackfriday/v2 v2.1.0 // indirect
github.com/shopspring/decimal v1.3.1 // indirect
github.com/sirupsen/logrus v1.9.3 // indirect
github.com/spf13/cast v1.5.0 // indirect
github.com/spf13/pflag v1.0.5 // indirect
github.com/xeipuuv/gojsonpointer v0.0.0-20190905194746-02993c407bfb // indirect
github.com/xeipuuv/gojsonreference v0.0.0-20180127040603-bd5ef7bd5415 // indirect
github.com/xeipuuv/gojsonschema v1.2.0 // indirect
github.com/xlab/treeprint v1.2.0 // indirect
go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.45.0 // indirect
go.opentelemetry.io/otel v1.19.0 // indirect
go.opentelemetry.io/otel/metric v1.19.0 // indirect
go.opentelemetry.io/otel/trace v1.19.0 // indirect
go.starlark.net v0.0.0-20230525235612-a134d8f9ddca // indirect
github.com/stretchr/testify v1.8.4 // indirect
golang.org/x/crypto v0.23.0 // indirect
golang.org/x/net v0.25.0 // indirect
golang.org/x/oauth2 v0.10.0 // indirect
golang.org/x/sync v0.6.0 // indirect
golang.org/x/sys v0.20.0 // indirect
golang.org/x/term v0.20.0 // indirect
golang.org/x/text v0.15.0 // indirect
golang.org/x/time v0.3.0 // indirect
google.golang.org/appengine v1.6.7 // indirect
google.golang.org/genproto/googleapis/rpc v0.0.0-20230822172742-b8732ec3820d // indirect
google.golang.org/grpc v1.58.3 // indirect
google.golang.org/protobuf v1.33.0 // indirect
gopkg.in/inf.v0 v0.9.1 // indirect
gopkg.in/yaml.v2 v2.4.0 // indirect
gopkg.in/yaml.v3 v3.0.1 // indirect
k8s.io/api v0.30.0 // indirect
k8s.io/apiextensions-apiserver v0.30.0 // indirect
k8s.io/apimachinery v0.30.0 // indirect
k8s.io/apiserver v0.30.0 // indirect
k8s.io/cli-runtime v0.30.0 // indirect
k8s.io/client-go v0.30.0 // indirect
k8s.io/component-base v0.30.0 // indirect
k8s.io/klog/v2 v2.120.1 // indirect
k8s.io/kube-openapi v0.0.0-20240228011516-70dd3763d340 // indirect
k8s.io/kubectl v0.30.0 // indirect
k8s.io/utils v0.0.0-20230726121419-3b25d923346b // indirect
oras.land/oras-go v1.2.4 // indirect
sigs.k8s.io/json v0.0.0-20221116044647-bc3834ca7abd // indirect
sigs.k8s.io/kustomize/api v0.13.5-0.20230601165947-6ce0bf390ce3 // indirect
sigs.k8s.io/kustomize/kyaml v0.14.3-0.20230601165947-6ce0bf390ce3 // indirect
sigs.k8s.io/structured-merge-diff/v4 v4.4.1 // indirect
)
Loading

0 comments on commit f03cfe1

Please sign in to comment.