add update status methods

internal/app/server.go

@@ -282,6 +282,46 @@ func (s *server) UpdateScenario(ctx context.Context, req *proto.UpdateScenarioRe
 	return &proto.UpdateScenarioRsp{}, nil
 }
 
+func (s *server) PublicScenario(ctx context.Context, req *proto.PublicScenarioReq) (*proto.PublicScenarioRsp, error) {
+	claims := ctx.Value("claims").(*processor_jwt.JWTClaims)
+	if !roles.HasRole(claims, roles.Author) {
+		return nil, status.Errorf(codes.PermissionDenied, "permission denied")
+	}
+
+	err := s.scenarioService.UpdateScenarioStatusByID(
+		ctx,
+		int(req.Id),
+		"public",
+	)
+	if err != nil {
+		return &proto.PublicScenarioRsp{
+			Error: err.Error(),
+		}, nil
+	}
+
+	return &proto.PublicScenarioRsp{}, nil
+}
+
+func (s *server) DraftScenario(ctx context.Context, req *proto.DraftScenarioReq) (*proto.DraftScenarioRsp, error) {
+	claims := ctx.Value("claims").(*processor_jwt.JWTClaims)
+	if !roles.HasRole(claims, roles.Author) {
+		return nil, status.Errorf(codes.PermissionDenied, "permission denied")
+	}
+
+	err := s.scenarioService.UpdateScenarioStatusByID(
+		ctx,
+		int(req.Id),
+		"draft",
+	)
+	if err != nil {
+		return &proto.DraftScenarioRsp{
+			Error: err.Error(),
+		}, nil
+	}
+
+	return &proto.DraftScenarioRsp{}, nil
+}
+
 func (s *server) DeleteScenario(ctx context.Context, req *proto.DeleteScenarioReq) (*proto.DeleteScenarioRsp, error) {
 	claims := ctx.Value("claims").(*processor_jwt.JWTClaims)
 	if !roles.HasRole(claims, roles.Author) {