feat: Allow setting project type during creation

Closes #1961
DSD-DBS · Nov 7, 2024 · e66ef12 · e66ef12
1 parent 3c5c1fd
commit e66ef12
Show file tree

Hide file tree

Showing 7 changed files with 46 additions and 15 deletions.
diff --git a/backend/capellacollab/projects/models.py b/backend/capellacollab/projects/models.py
@@ -107,6 +107,7 @@ class PostProjectRequest(core_pydantic.BaseModel):
     name: str
     description: str | None = None
     visibility: Visibility = Visibility.PRIVATE
+    type: ProjectType = ProjectType.GENERAL
 
 
 class DatabaseProject(database.Base):

diff --git a/backend/capellacollab/projects/routes.py b/backend/capellacollab/projects/routes.py
@@ -141,6 +141,7 @@ def create_project(
         post_project.name,
         post_project.description or "",
         post_project.visibility,
+        post_project.type,
     )
 
     if user.role != users_models.Role.ADMIN:

diff --git a/frontend/src/app/openapi/model/post-project-request.ts b/frontend/src/app/openapi/model/post-project-request.ts
diff --git a/frontend/src/app/projects/create-project/create-project.component.html b/frontend/src/app/projects/create-project/create-project.component.html
@@ -53,16 +53,33 @@ <h3 class="flex justify-center sm:hidden">Create new project</h3>
                   formControlName="visibility"
                   class="flex flex-col"
                 >
-                  <mat-radio-button
-                    *ngFor="
-                      let visibility of projectService.getAvailableVisibilities()
-                    "
-                    [value]="visibility"
-                  >
-                    {{
-                      projectService.getProjectVisibilityDescription(visibility)
-                    }}
-                  </mat-radio-button>
+                  @for (
+                    visibility of projectService.getAvailableVisibilities();
+                    track visibility
+                  ) {
+                    <mat-radio-button [value]="visibility">
+                      {{
+                        projectService.getProjectVisibilityDescription(
+                          visibility
+                        )
+                      }}
+                    </mat-radio-button>
+                  }
+                </mat-radio-group>
+              </div>
+            </fieldset>
+            <fieldset>
+              <div class="flex flex-col">
+                <legend>Project type</legend>
+                <mat-radio-group formControlName="type" class="flex flex-col">
+                  @for (
+                    type of projectService.getAvailableProjectTypes();
+                    track type
+                  ) {
+                    <mat-radio-button [value]="type">
+                      {{ projectService.getProjectTypeDescription(type) }}
+                    </mat-radio-button>
+                  }
                 </mat-radio-group>
               </div>
             </fieldset>

diff --git a/frontend/src/app/projects/create-project/create-project.component.ts b/frontend/src/app/projects/create-project/create-project.component.ts
@@ -2,7 +2,7 @@
  * SPDX-FileCopyrightText: Copyright DB InfraGO AG and contributors
  * SPDX-License-Identifier: Apache-2.0
  */
-import { NgFor, AsyncPipe } from '@angular/common';
+import { AsyncPipe } from '@angular/common';
 import { Component, OnDestroy, OnInit, ViewChild } from '@angular/core';
 import {
   FormControl,
@@ -18,7 +18,7 @@ import { MatInput } from '@angular/material/input';
 import { MatRadioGroup, MatRadioButton } from '@angular/material/radio';
 import { MatStepper, MatStep, MatStepLabel } from '@angular/material/stepper';
 import { RouterLink } from '@angular/router';
-import { Visibility } from 'src/app/openapi';
+import { ProjectType, Visibility } from 'src/app/openapi';
 import {
   CreateModelComponent,
   CreateModelStep,
@@ -44,7 +44,6 @@ import { ProjectWrapperService } from '../service/project.service';
     MatInput,
     MatError,
     MatRadioGroup,
-    NgFor,
     MatRadioButton,
     MatButton,
     MatIcon,
@@ -73,6 +72,7 @@ export class CreateProjectComponent implements OnInit, OnDestroy {
     }),
     description: new FormControl(''),
     visibility: new FormControl('private'),
+    type: new FormControl('general'),
   });
 
   ngOnInit(): void {
@@ -90,6 +90,7 @@ export class CreateProjectComponent implements OnInit, OnDestroy {
           name: this.form.value.name!,
           description: this.form.value.description!,
           visibility: this.form.value.visibility! as Visibility,
+          type: this.form.value.type! as ProjectType,
         })
         .subscribe((project) => {
           this.toastService.showSuccess(

diff --git a/frontend/src/app/projects/service/project.service.ts b/frontend/src/app/projects/service/project.service.ts
@@ -132,6 +132,6 @@ export const ProjectVisibilityDescriptions = {
 };
 
 export const ProjectTypeDescriptions = {
-  general: 'General (a project that contains multiple related models)',
+  general: 'General (a project that contains related models)',
   training: 'Training (special project containing training material)',
 };
diff --git a/frontend/src/storybook/project.ts b/frontend/src/storybook/project.ts
@@ -4,8 +4,9 @@
  */
 import { AsyncValidatorFn, ValidationErrors } from '@angular/forms';
 import { BehaviorSubject, Observable, of } from 'rxjs';
-import { Project, Visibility } from 'src/app/openapi';
+import { Project, ProjectType, Visibility } from 'src/app/openapi';
 import {
+  ProjectTypeDescriptions,
   ProjectVisibilityDescriptions,
   ProjectWrapperService,
 } from 'src/app/projects/service/project.service';
@@ -51,6 +52,14 @@ export class MockProjectWrapperService
   getAvailableVisibilities(): Visibility[] {
     return Object.keys(ProjectVisibilityDescriptions) as Visibility[];
   }
+
+  getProjectTypeDescription(type: ProjectType): string {
+    return ProjectTypeDescriptions[type];
+  }
+
+  getAvailableProjectTypes(): ProjectType[] {
+    return Object.keys(ProjectTypeDescriptions) as ProjectType[];
+  }
   createProject(project: Project): Observable<Project> {
     return of(project);
   }