feat: add subscription service

api/v2/subscription_service.go

@@ -0,0 +1,98 @@
+package v2
+
+import (
+	"context"
+	"time"
+
+	"github.com/boojack/slash/plugin/license"
+	apiv2pb "github.com/boojack/slash/proto/gen/api/v2"
+	storepb "github.com/boojack/slash/proto/gen/store"
+	"github.com/boojack/slash/server/profile"
+	"github.com/boojack/slash/store"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/types/known/timestamppb"
+)
+
+type SubscriptionService struct {
+	apiv2pb.UnimplementedSubscriptionServiceServer
+
+	Profile *profile.Profile
+	Store   *store.Store
+}
+
+// NewSubscriptionService creates a new SubscriptionService.
+func NewSubscriptionService(profile *profile.Profile, store *store.Store) *SubscriptionService {
+	return &SubscriptionService{
+		Profile: profile,
+		Store:   store,
+	}
+}
+
+func (s *SubscriptionService) GetSubscription(ctx context.Context, _ *apiv2pb.GetSubscriptionRequest) (*apiv2pb.GetSubscriptionResponse, error) {
+	workspaceSetting, err := s.Store.GetWorkspaceSetting(ctx, &store.FindWorkspaceSetting{
+		Key: storepb.WorkspaceSettingKey_WORKSPACE_SETTING_LICENSE_KEY,
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to get workspace setting: %v", err)
+	}
+	subscription := &apiv2pb.Subscription{
+		Plan: apiv2pb.PlanType_FREE,
+	}
+	licenseKey := ""
+	if workspaceSetting != nil {
+		licenseKey = workspaceSetting.GetLicenseKey()
+	}
+	if licenseKey == "" {
+		return &apiv2pb.GetSubscriptionResponse{
+			Subscription: subscription,
+		}, nil
+	}
+
+	validateResponse, err := license.ValidateLicenseKey(licenseKey, "")
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to validate license key: %v", err)
+	}
+	if validateResponse.Valid {
+		subscription.Plan = apiv2pb.PlanType_PRO
+		if validateResponse.LicenseKey.ExpiresAt != nil && *validateResponse.LicenseKey.ExpiresAt != "" {
+			expiresTime, err := time.Parse("2006-01-02 15:04:05", *validateResponse.LicenseKey.ExpiresAt)
+			if err != nil {
+				return nil, status.Errorf(codes.Internal, "failed to parse license key expired time: %v", err)
+			}
+			subscription.ExpiresTime = timestamppb.New(expiresTime)
+		}
+		startedTime, err := time.Parse("2006-01-02 15:04:05", validateResponse.LicenseKey.CreatedAt)
+		if err != nil {
+			return nil, status.Errorf(codes.Internal, "failed to parse license key created time: %v", err)
+		}
+		subscription.StartedTime = timestamppb.New(startedTime)
+	}
+	return &apiv2pb.GetSubscriptionResponse{
+		Subscription: subscription,
+	}, nil
+}
+
+func (s *SubscriptionService) UpdateSubscription(ctx context.Context, request *apiv2pb.UpdateSubscriptionRequest) (*apiv2pb.UpdateSubscriptionResponse, error) {
+	licenseKey := request.LicenseKey
+	if licenseKey == "" {
+		return nil, status.Errorf(codes.InvalidArgument, "license key is required")
+	}
+	validateResponse, err := license.ValidateLicenseKey(licenseKey, "")
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to validate license key: %v", err)
+	}
+	if !validateResponse.Valid {
+		return nil, status.Errorf(codes.InvalidArgument, "invalid license key")
+	}
+	_, err = s.Store.UpsertWorkspaceSetting(ctx, &storepb.WorkspaceSetting{
+		Key: storepb.WorkspaceSettingKey_WORKSPACE_SETTING_LICENSE_KEY,
+		Value: &storepb.WorkspaceSetting_LicenseKey{
+			LicenseKey: licenseKey,
+		},
+	})
+	if err != nil {
+		return nil, status.Errorf(codes.Internal, "failed to update license key: %v", err)
+	}
+	return &apiv2pb.UpdateSubscriptionResponse{}, nil
+}

api/v2/v2.go

@@ -31,6 +31,7 @@ func NewAPIV2Service(secret string, profile *profile.Profile, store *store.Store
 			authProvider.AuthenticationInterceptor,
 		),
 	)
+	apiv2pb.RegisterSubscriptionServiceServer(grpcServer, NewSubscriptionService(profile, store))
 	apiv2pb.RegisterWorkspaceServiceServer(grpcServer, NewWorkspaceService(profile, store))
 	apiv2pb.RegisterUserServiceServer(grpcServer, NewUserService(secret, store))
 	apiv2pb.RegisterUserSettingServiceServer(grpcServer, NewUserSettingService(store))
@@ -64,6 +65,9 @@ func (s *APIV2Service) RegisterGateway(ctx context.Context, e *echo.Echo) error
 	}
 
 	gwMux := grpcRuntime.NewServeMux()
+	if err := apiv2pb.RegisterSubscriptionServiceHandler(context.Background(), gwMux, conn); err != nil {
+		return err
+	}
 	if err := apiv2pb.RegisterWorkspaceServiceHandler(context.Background(), gwMux, conn); err != nil {
 		return err
 	}

api/v2/workspace_service.go

@@ -71,7 +71,7 @@ func (s *WorkspaceService) GetWorkspaceSetting(ctx context.Context, _ *apiv2pb.G
 			workspaceSetting.CustomScript = v.GetCustomScript()
 		} else if isAdmin {
 			// For some settings, only admin can get the value.
-			if v.Key == storepb.WorkspaceSettingKey_WORKSPACE_LICENSE_KEY {
+			if v.Key == storepb.WorkspaceSettingKey_WORKSPACE_SETTING_LICENSE_KEY {
 				workspaceSetting.LicenseKey = v.GetLicenseKey()
 			} else if v.Key == storepb.WorkspaceSettingKey_WORKSPACE_SETTING_RESOURCE_RELATIVE_PATH {
 				workspaceSetting.ResourceRelativePath = v.GetResourceRelativePath()
@@ -91,7 +91,7 @@ func (s *WorkspaceService) UpdateWorkspaceSetting(ctx context.Context, request *
 	for _, path := range request.UpdateMask {
 		if path == "license_key" {
 			if _, err := s.Store.UpsertWorkspaceSetting(ctx, &storepb.WorkspaceSetting{
-				Key: storepb.WorkspaceSettingKey_WORKSPACE_LICENSE_KEY,
+				Key: storepb.WorkspaceSettingKey_WORKSPACE_SETTING_LICENSE_KEY,
 				Value: &storepb.WorkspaceSetting_LicenseKey{
 					LicenseKey: request.Setting.LicenseKey,
 				},