Commit 4d5d5138 authored by Yannik Dällenbach's avatar Yannik Dällenbach 🤼 Committed by Yannik Dällenbach
Browse files

Configure API client on `Init`

parent 005bc200

go.mod

+2 −0
Original line number Diff line number Diff line
@@ -3,11 +3,13 @@ module cloudscale-fleeting-plugin 
go 1.24.3



require (

	github.com/cloudscale-ch/cloudscale-go-sdk/v6 v6.0.0

	github.com/hashicorp/go-hclog v1.6.3

	gitlab.com/gitlab-org/fleeting/fleeting v0.0.0-20250425145049-7f673e7c5598

)



require (

	github.com/cenkalti/backoff/v5 v5.0.2 // indirect

	github.com/fatih/color v1.17.0 // indirect

	github.com/golang/protobuf v1.5.4 // indirect

	github.com/hashicorp/go-plugin v1.6.1 // indirect

go.sum

+4 −0
Original line number Diff line number Diff line
@@ -8,6 +8,10 @@ github.com/bodgit/windows v1.0.1 h1:tF7K6KOluPYygXa3Z2594zxlkbKPAOvqr97etrGNIz4= 
github.com/bodgit/windows v1.0.1/go.mod h1:a6JLwrB4KrTR5hBpp8FI9/9W9jJfeQ2h4XDXU74ZCdM=

github.com/bufbuild/protocompile v0.4.0 h1:LbFKd2XowZvQ/kajzguUp2DC9UEIQhIq77fZZlaQsNA=

github.com/bufbuild/protocompile v0.4.0/go.mod h1:3v93+mbWn/v3xzN+31nwkJfrEpAUwp+BagBSZWx+TP8=

github.com/cenkalti/backoff/v5 v5.0.2 h1:rIfFVxEf1QsI7E1ZHfp/B4DF/6QBAUhmgkxc0H7Zss8=

github.com/cenkalti/backoff/v5 v5.0.2/go.mod h1:rkhZdG3JZukswDf7f0cwqPNk4K0sa+F97BxZthm/crw=

github.com/cloudscale-ch/cloudscale-go-sdk/v6 v6.0.0 h1:lIVkmacVa4GogQ17dtrTEh/ph+k8gH2bsQcfJu/Tk0s=

github.com/cloudscale-ch/cloudscale-go-sdk/v6 v6.0.0/go.mod h1:agOnbZIZJUfW4V/4s5wYX7IoXoixGxXzQhW30+fpGPU=

github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=

github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=

github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=

main.go

+22 −3
Original line number Diff line number Diff line
@@ -2,24 +2,43 @@ package main 


import (

	"context"

	"fmt"

	"math"

	"net/http"



	"github.com/cloudscale-ch/cloudscale-go-sdk/v6"

	hclog "github.com/hashicorp/go-hclog"

	"gitlab.com/gitlab-org/fleeting/fleeting/plugin"

	"gitlab.com/gitlab-org/fleeting/fleeting/provider"

)



type instanceGroup struct{}

type instanceGroup struct {

	ApiToken string `json:"api_token"` // Holds the cloudscale API token configured in `plugin_config`.

	client   *cloudscale.Client



	settings provider.Settings

}



// Check if interface `InstanceGroup` is implemented.

var _ provider.InstanceGroup = (*instanceGroup)(nil)



// Init implements provider.InstanceGroup.

func (g *instanceGroup) Init(ctx context.Context, logger hclog.Logger, settings provider.Settings) (provider.ProviderInfo, error) {

	return provider.ProviderInfo{

	info := provider.ProviderInfo{

		ID:      "cloudscale",

		MaxSize: math.MaxInt,

	}, nil

	}



	g.settings = settings



	g.client = cloudscale.NewClient(http.DefaultClient)

	g.client.AuthToken = g.ApiToken



	if _, err := g.client.Servers.List(ctx); err != nil {

		return info, fmt.Errorf("failed to create cloudscale API client: %w", err)

	}



	return info, nil

}



// Increase implements provider.InstanceGroup.