Add restart allocation function

Author: DistroByte

Dockerfile

@@ -1,16 +1,17 @@
 FROM golang:1.23.6 as builder
 
-COPY . /go/src/github.com/DistroByte/molocule
+COPY vendor /go/src/github.com/DistroByte/molecule/vendor
+COPY . /go/src/github.com/DistroByte/molecule
 
-WORKDIR /go/src/github.com/DistroByte/molocule
+WORKDIR /go/src/github.com/DistroByte/molecule
 
-RUN CGO_ENABLED=0 go build -o bin/molocule .
+RUN CGO_ENABLED=0 go build -o bin/molecule .
 
 FROM gcr.io/distroless/static-debian12
 
-COPY --from=builder /go/src/github.com/DistroByte/molocule/bin/molocule /bin/molocule
-COPY --from=builder /go/src/github.com/DistroByte/molocule/web /web
+COPY --from=builder /go/src/github.com/DistroByte/molocule/bin/molecule /bin/molecule
+COPY --from=builder /go/src/github.com/DistroByte/molecule/web /web
 
 WORKDIR /
 
-ENTRYPOINT ["/bin/molocule"]
+ENTRYPOINT ["/bin/molecule"]

apispec/spec/index.yaml

@@ -23,6 +23,12 @@ paths:
     $ref: v1/urls/host-urls.yaml
   /v1/urls/traefik:
     $ref: v1/urls/traefik-urls.yaml
+  /v1/services/{service}:
+    $ref: v1/services/index.yaml
+  /v1/services/{service}/alloc-restart:
+    $ref: v1/services/alloc-restart.yaml
+    security:
+      - ApiKeyAuth: []
 
 components:
   securitySchemes:

apispec/spec/v1/services/alloc-restart.yaml

@@ -0,0 +1,30 @@
+parameters:
+  - name: service
+    in: path
+    required: true
+    schema:
+      $ref: schemas/service-name.json
+    description: The name of the service
+
+post:
+  summary: Restart all allocations of a service
+  operationId: restart_service_allocations
+  responses:
+    "200":
+      description: OK
+      content:
+        application/json:
+          schema:
+            $ref: schemas/generic-response.json
+    "400":
+      description: Invalid request
+      content:
+        application/json:
+          schema:
+            $ref: schemas/generic-response.json
+    "500":
+      description: Internal server error
+      content:
+        application/json:
+          schema:
+            $ref: schemas/generic-response.json

apispec/spec/v1/services/index.yaml

@@ -0,0 +1,24 @@
+parameters:
+  - name: service
+    in: path
+    required: true
+    schema:
+      $ref: schemas/service-name.json
+    description: The name of the service
+
+get:
+  summary: Get the status of a service
+  operationId: get_service_status
+  responses:
+    "200":
+      description: OK
+      content:
+        application/json:
+          schema:
+            $ref: schemas/service-status.json
+    "400":
+      description: Invalid request
+      content:
+        application/json:
+          schema:
+            $ref: schemas/generic-response.json

apispec/spec/v1/services/schemas/generic-response.json

@@ -0,0 +1,12 @@
+{
+    "type": "object",
+    "properties": {
+        "status": {
+            "type": "string",
+            "enum": ["success", "error"]
+        },
+        "message": {
+            "type": "string"
+        }
+    }
+}

apispec/spec/v1/services/schemas/service-name.json

@@ -0,0 +1,5 @@
+{
+    "type": "string",
+    "pattern": "^[a-z0-9-]+$",
+    "example": "molecule"
+}

apispec/spec/v1/services/schemas/service-status.json

@@ -0,0 +1,5 @@
+{
+    "type": "string",
+    "pattern": "^[a-z0-9-]+$",
+    "example": "running"
+}

internal/api/v1/nomad_service.go

@@ -24,6 +24,8 @@ type NomadServiceInterface interface {
 	ExtractURLs() (map[string]string, error)
 	ExtractHostPorts() (map[string]string, error)
 	ExtractServicePorts() (map[string]string, error)
+	GetServiceStatus(service string) (map[string]string, error)
+	RestartServiceAllocations(service string) (map[string]string, error)
 }
 
 var (
@@ -42,7 +44,7 @@ func NewNomadService(nomadClient *api.Client) NomadServiceInterface {
 	standardURLs["photos"] = "https://photos.dbyte.xyz"
 	standardURLs["drive"] = "https://drive.dbyte.xyz"
 	standardURLs["plex"] = "https://video.dbyte.xyz"
-	standardURLs["ghost"] = "https://admin-photo.james-hackett.ie"
+	standardURLs["ghost"] = "https://admin-photo.james-hackett.ie/ghost"
 
 	return &NomadService{nomadClient: nomadClient}
 }
@@ -147,6 +149,72 @@ func (s *NomadService) ExtractServicePorts() (map[string]string, error) {
 	return servicePorts, nil
 }
 
+func (s *NomadService) GetServiceStatus(serviceName string) (map[string]string, error) {
+	allocations, _, err := s.nomadClient.Allocations().List(nil)
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to list allocations")
+		return nil, err
+	}
+
+	serviceUrls = make(map[string]string)
+	hostReservedPorts = make(map[string]string)
+	servicePorts = make(map[string]string)
+
+	for _, allocation := range allocations {
+		s.processAllocation(allocation)
+	}
+
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	if val, ok := serviceUrls[serviceName]; ok {
+		return map[string]string{serviceName: val}, nil
+	}
+
+	if val, ok := hostReservedPorts[serviceName]; ok {
+		return map[string]string{serviceName: val}, nil
+	}
+
+	if val, ok := servicePorts[serviceName]; ok {
+		return map[string]string{serviceName: val}, nil
+	}
+
+	return nil, nil
+}
+
+// when called, this function will restart all allocations for a specific service
+// this is useful when a new version of a service is built and you want to restart all instances
+func (s *NomadService) RestartServiceAllocations(serviceName string) (map[string]string, error) {
+	allocations, _, err := s.nomadClient.Allocations().List(nil)
+	if err != nil {
+		log.Error().Err(err).Msg("Failed to list allocations")
+		return nil, err
+	}
+
+	for _, allocation := range allocations {
+		job, _, err := s.nomadClient.Jobs().Info(allocation.JobID, nil)
+		if err != nil {
+			log.Error().Err(err).Msg("Failed to get job info")
+			return nil, err
+		}
+
+		if *job.Name == serviceName {
+			allocationInfo, _, err := s.nomadClient.Allocations().Info(allocation.ID, nil)
+			if err != nil {
+				log.Error().Err(err).Msg("Failed to get allocation info")
+				return nil, err
+			}
+			err = s.nomadClient.Allocations().Restart(allocationInfo, "", nil)
+			if err != nil {
+				log.Error().Err(err).Msg("Failed to restart service allocations")
+				return nil, err
+			}
+		}
+	}
+
+	return nil, nil
+}
+
 func (s *NomadService) processAllocation(allocation *api.AllocationListStub) {
 	services := []*api.Service{}
 

internal/api/v1/nomad_service_mock.go

@@ -34,3 +34,13 @@ func (m *MockNomadService) ExtractServicePorts() (map[string]string, error) {
 	log.Debug().Msg("Mock: ExtractServicePorts called")
 	return urls, nil
 }
+
+func (m *MockNomadService) GetServiceStatus(service string) (map[string]string, error) {
+	log.Debug().Msg("Mock: GetServiceStatus called")
+	return urls, nil
+}
+
+func (m *MockNomadService) RestartServiceAllocations(service string) (map[string]string, error) {
+	log.Debug().Msg("Mock: RestartServiceAllocations called")
+	return urls, nil
+}

internal/api/v1/url_service.go

@@ -46,3 +46,23 @@ func (s *MoleculeAPIService) GetTraefikURLs(ctx context.Context) (openapi.ImplRe
 	// Return the response
 	return openapi.Response(http.StatusOK, urls), nil
 }
+
+func (s *MoleculeAPIService) GetServiceStatus(ctx context.Context, service string) (openapi.ImplResponse, error) {
+	status, err := s.nomadService.GetServiceStatus(service)
+	if err != nil {
+		return openapi.Response(http.StatusInternalServerError, err.Error()), nil
+	}
+
+	// Return the response
+	return openapi.Response(http.StatusOK, status), nil
+}
+
+func (s *MoleculeAPIService) RestartServiceAllocations(ctx context.Context, service string) (openapi.ImplResponse, error) {
+	_, err := s.nomadService.RestartServiceAllocations(service)
+	if err != nil {
+		return openapi.Response(http.StatusInternalServerError, err.Error()), nil
+	}
+
+	// Return the response
+	return openapi.Response(http.StatusOK, "OK"), nil
+}