Initial QSfera import

Server/services/postprocessing/pkg/command/postprocessing.go

@@ -0,0 +1,88 @@
+package command
+
+import (
+	"context"
+
+	"github.com/qsfera/server/pkg/config/configlog"
+	"github.com/qsfera/server/pkg/generators"
+	"github.com/qsfera/server/services/postprocessing/pkg/config"
+	"github.com/qsfera/server/services/postprocessing/pkg/config/parser"
+	"github.com/opencloud-eu/reva/v2/pkg/events"
+	"github.com/opencloud-eu/reva/v2/pkg/events/stream"
+	"github.com/opencloud-eu/reva/v2/pkg/utils"
+
+	"github.com/spf13/cobra"
+)
+
+// RestartPostprocessing cli command to restart postprocessing
+func RestartPostprocessing(cfg *config.Config) *cobra.Command {
+	restartPostprocessingCmd := &cobra.Command{
+		Use:     "resume",
+		Aliases: []string{"restart"},
+		Short:   "resume postprocessing for an uploadID",
+
+		PreRunE: func(cmd *cobra.Command, args []string) error {
+			return configlog.ReturnFatal(parser.ParseConfig(cfg))
+		},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			connName := generators.GenerateConnectionName(cfg.Service.Name, generators.NTypeBus)
+			stream, err := stream.NatsFromConfig(connName, false, stream.NatsConfig{
+				Endpoint:             cfg.Postprocessing.Events.Endpoint,
+				Cluster:              cfg.Postprocessing.Events.Cluster,
+				EnableTLS:            cfg.Postprocessing.Events.EnableTLS,
+				TLSInsecure:          cfg.Postprocessing.Events.TLSInsecure,
+				TLSRootCACertificate: cfg.Postprocessing.Events.TLSRootCACertificate,
+				AuthUsername:         cfg.Postprocessing.Events.AuthUsername,
+				AuthPassword:         cfg.Postprocessing.Events.AuthPassword,
+			})
+			if err != nil {
+				return err
+			}
+
+			uid, _ := cmd.Flags().GetString("upload-id")
+			step := ""
+			if uid == "" {
+				step, _ = cmd.Flags().GetString("step")
+			}
+
+			restart, _ := cmd.Flags().GetBool("restart")
+			var ev events.Unmarshaller
+			switch {
+			case restart:
+				ev = events.RestartPostprocessing{
+					UploadID:  uid,
+					Timestamp: utils.TSNow(),
+				}
+			default:
+				ev = events.ResumePostprocessing{
+					UploadID:  uid,
+					Step:      events.Postprocessingstep(step),
+					Timestamp: utils.TSNow(),
+				}
+			}
+
+			return events.Publish(context.Background(), stream, ev)
+		},
+	}
+
+	restartPostprocessingCmd.Flags().StringP(
+		"upload-id",
+		"u",
+		"",
+		"the uploadid to resume. Ignored if unset.",
+	)
+	restartPostprocessingCmd.Flags().StringP(
+		"step",
+		"s",
+		"finished",
+		"resume all uploads in the given postprocessing step. Ignored if upload-id is set.",
+	)
+	restartPostprocessingCmd.Flags().BoolP(
+		"restart",
+		"r",
+		false,
+		"restart postprocessing for the given uploadID. Ignores the step flag.",
+	)
+
+	return restartPostprocessingCmd
+}