init from gitlab

internal/server/github/app.go

@@ -0,0 +1,86 @@
+package github
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/bradleyfalzon/ghinstallation/v2"
+	gh "github.com/google/go-github/v48/github"
+	"github.com/rs/zerolog/log"
+
+	"gitlab.com/texm/shokku/internal/env"
+	"gitlab.com/texm/shokku/internal/models"
+)
+
+type AppClient struct {
+	*gh.Client
+	appsTransport *ghinstallation.AppsTransport
+}
+
+func GetAppClient(e *env.Env) (*AppClient, error) {
+	var dbApp models.GithubApp
+	if err := e.DB.Find(&dbApp).Error; err != nil {
+		return nil, err
+	}
+	transport, err := ghinstallation.NewAppsTransport(
+		http.DefaultTransport, dbApp.AppId, []byte(dbApp.PEM))
+	if err != nil {
+		log.Debug().Err(err).Msg("failed to create transport")
+		return nil, err
+	}
+	appClient := &AppClient{
+		Client:        gh.NewClient(&http.Client{Transport: transport}),
+		appsTransport: transport,
+	}
+	return appClient, nil
+}
+
+type AppInstallationClient struct {
+	*gh.Client
+}
+
+func (c *AppClient) GetInstallationClient(id int64) *AppInstallationClient {
+	transport := ghinstallation.NewFromAppsTransport(c.appsTransport, id)
+	client := gh.NewClient(&http.Client{Transport: transport})
+	return &AppInstallationClient{Client: client}
+}
+
+func CompleteAppManifest(e *env.Env, code string) (*gh.AppConfig, error) {
+	ctx := context.Background()
+	client := gh.NewClient(nil)
+	cfg, _, ghErr := client.Apps.CompleteAppManifest(ctx, code)
+	if ghErr != nil {
+		return nil, ghErr
+	}
+
+	appId := cfg.GetID()
+	dbApp := models.GithubApp{AppId: appId}
+	if dbErr := e.DB.FirstOrCreate(&dbApp).Error; dbErr != nil {
+		log.Error().Err(dbErr).Msg("failed db lookup")
+		return nil, dbErr
+	}
+
+	dbApp.AppId = appId
+	dbApp.ClientId = cfg.GetClientID()
+	dbApp.NodeId = cfg.GetNodeID()
+	dbApp.Slug = cfg.GetSlug()
+	dbApp.PEM = cfg.GetPEM()
+	dbApp.ClientSecret = cfg.GetClientSecret()
+	dbApp.WebhookSecret = cfg.GetWebhookSecret()
+
+	// saveRes := e.DB.Where(&models.GithubApp{AppId: appId}).Save(&dbApp)
+	if err := e.DB.Save(&dbApp).Error; err != nil {
+		log.Error().Err(err).Msg("failed to save db app")
+		return nil, err
+	}
+
+	return cfg, nil
+}
+
+func (c *AppClient) GetApp(ctx context.Context) (*gh.App, error) {
+	app, _, err := c.Apps.Get(ctx, "")
+	if err != nil {
+		return nil, err
+	}
+	return app, nil
+}