Add usagereport service

Create the service and register it when the content-service starts.

Author: Andrew Farries

components/content-service/cmd/run.go

@@ -57,6 +57,12 @@ var runCmd = &cobra.Command{
 		}
 		api.RegisterIDEPluginServiceServer(srv.GRPC(), idePluginService)
 
+		usageReportService, err := service.NewUsageReportService(cfg.Storage)
+		if err != nil {
+			log.WithError(err).Fatalf("Cannot create usage report service")
+		}
+		api.RegisterUsageReportServiceServer(srv.GRPC(), usageReportService)
+
 		err = srv.ListenAndServe()
 		if err != nil {
 			log.WithError(err).Fatal("Cannot start server")

components/content-service/pkg/service/usagereport-service.go

@@ -0,0 +1,63 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package service
+
+import (
+	"context"
+
+	"github.com/opentracing/opentracing-go"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+
+	"github.com/gitpod-io/gitpod/common-go/log"
+	"github.com/gitpod-io/gitpod/common-go/tracing"
+	"github.com/gitpod-io/gitpod/content-service/api"
+	"github.com/gitpod-io/gitpod/content-service/api/config"
+	"github.com/gitpod-io/gitpod/content-service/pkg/storage"
+)
+
+// UsageReportService implements UsageReportServiceServer
+type UsageReportService struct {
+	cfg config.StorageConfig
+	s   storage.PresignedAccess
+
+	api.UnimplementedUsageReportServiceServer
+}
+
+// NewUsageReportService create a new usagereport service
+func NewUsageReportService(cfg config.StorageConfig) (res *UsageReportService, err error) {
+	s, err := storage.NewPresignedAccess(&cfg)
+	if err != nil {
+		return nil, err
+	}
+	return &UsageReportService{cfg: cfg, s: s}, nil
+}
+
+// UploadURL provides a URL to which clients can upload the content via HTTP PUT.
+func (us *UsageReportService) UploadURL(ctx context.Context, req *api.UsageReportUploadURLRequest) (resp *api.UsageReportUploadURLResponse, err error) {
+	span, ctx := opentracing.StartSpanFromContext(ctx, "UsageReport.UploadURL")
+	span.SetTag("name", req.Name)
+	defer tracing.FinishSpan(span, &err)
+
+	err = us.s.EnsureExists(ctx, req.Bucket)
+	if err != nil {
+		return nil, status.Error(codes.NotFound, err.Error())
+	}
+
+	info, err := us.s.SignUpload(ctx, req.Bucket, req.Name, &storage.SignedURLOptions{
+		ContentType: "*/*",
+	})
+	if err != nil {
+		log.WithField("name", req.Name).
+			WithField("bucket", req.Bucket).
+			WithError(err).
+			Error("error getting UsageReport SignUpload URL")
+		if err == storage.ErrNotFound {
+			return nil, status.Error(codes.NotFound, err.Error())
+		}
+		return nil, status.Error(codes.Unknown, err.Error())
+	}
+	return &api.UsageReportUploadURLResponse{Url: info.URL}, nil
+}