Initial commit for open source release 🚀

Co-authored-by: Francesco Renzi <rentziass@github.com> Co-authored-by: Nikola Jokic <jokicnikola07@gmail.com>
2026-02-03 16:41:15 +01:00
commit e4a017ce06
44 changed files with 7340 additions and 0 deletions
@@ -0,0 +1,149 @@
+package scaleset
+
+import (
+	"time"
+
+	"github.com/google/uuid"
+)
+
+const DefaultRunnerGroup = "default"
+
+type MessageType string
+
+// message types
+const (
+	MessageTypeJobAssigned  MessageType = "JobAssigned"
+	MessageTypeJobStarted   MessageType = "JobStarted"
+	MessageTypeJobCompleted MessageType = "JobCompleted"
+)
+
+type JobAssigned struct {
+	JobMessageBase
+}
+
+type JobStarted struct {
+	RunnerID   int    `json:"runnerId"`
+	RunnerName string `json:"runnerName"`
+	JobMessageBase
+}
+
+type JobCompleted struct {
+	Result     string `json:"result"`
+	RunnerID   int    `json:"runnerId"`
+	RunnerName string `json:"runnerName"`
+	JobMessageBase
+}
+
+type JobMessageType struct {
+	MessageType MessageType `json:"messageType"`
+}
+
+type JobMessageBase struct {
+	JobMessageType
+	RunnerRequestID    int64     `json:"runnerRequestId"`
+	RepositoryName     string    `json:"repositoryName"`
+	OwnerName          string    `json:"ownerName"`
+	JobID              string    `json:"jobId"`
+	JobWorkflowRef     string    `json:"jobWorkflowRef"`
+	JobDisplayName     string    `json:"jobDisplayName"`
+	WorkflowRunID      int64     `json:"workflowRunId"`
+	EventName          string    `json:"eventName"`
+	RequestLabels      []string  `json:"requestLabels"`
+	QueueTime          time.Time `json:"queueTime"`
+	ScaleSetAssignTime time.Time `json:"scaleSetAssignTime"`
+	RunnerAssignTime   time.Time `json:"runnerAssignTime"`
+	FinishTime         time.Time `json:"finishTime"`
+}
+
+type Label struct {
+	Type string `json:"type"`
+	Name string `json:"name"`
+}
+
+type RunnerGroup struct {
+	ID        int    `json:"id"`
+	Name      string `json:"name"`
+	Size      int    `json:"size"`
+	IsDefault bool   `json:"isDefaultGroup"`
+}
+
+type RunnerGroupList struct {
+	Count        int           `json:"count"`
+	RunnerGroups []RunnerGroup `json:"value"`
+}
+
+type RunnerScaleSet struct {
+	ID                 int                      `json:"id,omitempty"`
+	Name               string                   `json:"name,omitempty"`
+	RunnerGroupID      int                      `json:"runnerGroupId,omitempty"`
+	RunnerGroupName    string                   `json:"runnerGroupName,omitempty"`
+	Labels             []Label                  `json:"labels,omitempty"`
+	RunnerSetting      RunnerSetting            `json:"RunnerSetting,omitempty"`
+	CreatedOn          time.Time                `json:"createdOn,omitempty"`
+	RunnerJitConfigURL string                   `json:"runnerJitConfigUrl,omitempty"`
+	Statistics         *RunnerScaleSetStatistic `json:"statistics,omitempty"`
+}
+
+type RunnerScaleSetJitRunnerSetting struct {
+	Name       string `json:"name"`
+	WorkFolder string `json:"workFolder"`
+}
+
+type runnerScaleSetMessageResponse struct {
+	MessageID   int                      `json:"messageId"`
+	MessageType string                   `json:"messageType"`
+	Body        string                   `json:"body"`
+	Statistics  *RunnerScaleSetStatistic `json:"statistics"`
+}
+
+type RunnerScaleSetMessage struct {
+	MessageID            int
+	Statistics           *RunnerScaleSetStatistic
+	JobAssignedMessages  []*JobAssigned
+	JobStartedMessages   []*JobStarted
+	JobCompletedMessages []*JobCompleted
+}
+
+type runnerScaleSetsResponse struct {
+	Count           int              `json:"count"`
+	RunnerScaleSets []RunnerScaleSet `json:"value"`
+}
+
+type RunnerScaleSetSession struct {
+	SessionID               uuid.UUID                `json:"sessionId,omitempty"`
+	OwnerName               string                   `json:"ownerName,omitempty"`
+	RunnerScaleSet          *RunnerScaleSet          `json:"runnerScaleSet,omitempty"`
+	MessageQueueURL         string                   `json:"messageQueueUrl,omitempty"`
+	MessageQueueAccessToken string                   `json:"messageQueueAccessToken,omitempty"`
+	Statistics              *RunnerScaleSetStatistic `json:"statistics,omitempty"`
+}
+
+type RunnerScaleSetStatistic struct {
+	TotalAvailableJobs     int `json:"totalAvailableJobs"`
+	TotalAcquiredJobs      int `json:"totalAcquiredJobs"`
+	TotalAssignedJobs      int `json:"totalAssignedJobs"`
+	TotalRunningJobs       int `json:"totalRunningJobs"`
+	TotalRegisteredRunners int `json:"totalRegisteredRunners"`
+	TotalBusyRunners       int `json:"totalBusyRunners"`
+	TotalIdleRunners       int `json:"totalIdleRunners"`
+}
+
+type RunnerSetting struct {
+	DisableUpdate bool `json:"disableUpdate,omitempty"`
+}
+
+type RunnerReferenceList struct {
+	Count            int               `json:"count"`
+	RunnerReferences []RunnerReference `json:"value"`
+}
+
+type RunnerReference struct {
+	ID               int    `json:"id"`
+	Name             string `json:"name"`
+	RunnerScaleSetID int    `json:"runnerScaleSetId"`
+}
+
+type RunnerScaleSetJitRunnerConfig struct {
+	Runner           *RunnerReference `json:"runner"`
+	EncodedJITConfig string           `json:"encodedJITConfig"`
+}