Add adminCourses to LoginUserResponse

Signed-off-by: loheagn <[email protected]>

Author: loheagn

openapi/cloudapi_v2.yaml

@@ -3449,13 +3449,18 @@ components:
           type: array
           items:
             type: string
+        adminCourses:
+          type: array
+          items:
+            $ref: '#/components/schemas/SimpleCourse'
       required:
         - userId
         - username
         - role
         - isAssistant
         - token
         - paasToken
+        - adminCourses
     ContainerServiceTemplate:
       title: ContainerServiceTemplate
       x-stoplight:
@@ -3560,6 +3565,20 @@ components:
         - bucket
         - key
         - region
+    SimpleCourse:
+      title: SimpleCourse
+      x-stoplight:
+        id: 6d3ivgkgbf3rv
+      type: object
+      properties:
+        id:
+          type: integer
+          format: int32
+        name:
+          type: string
+      required:
+        - id
+        - name
   securitySchemes:
     Authorization:
       type: apiKey

src/main/kotlin/cn/edu/buaa/scs/controller/models/ContainerRequest.kt

@@ -11,6 +11,7 @@
 */
 package cn.edu.buaa.scs.controller.models
 
+import cn.edu.buaa.scs.controller.models.SimpleCourse
 
 /**
  * 
@@ -20,6 +21,7 @@ package cn.edu.buaa.scs.controller.models
  * @param isAssistant 
  * @param token 
  * @param paasToken 
+ * @param adminCourses 
  * @param projects 
  */
 data class LoginUserResponse(
@@ -29,6 +31,7 @@ data class LoginUserResponse(
     val isAssistant: kotlin.Boolean,
     val token: kotlin.String,
     val paasToken: kotlin.String,
+    val adminCourses: kotlin.collections.List<SimpleCourse>,
     val projects: kotlin.collections.List<kotlin.String>? = null
 ) 
 

src/main/kotlin/cn/edu/buaa/scs/controller/models/ContainerServiceRequest.kt

@@ -14,13 +14,11 @@ package cn.edu.buaa.scs.controller.models
 
 /**
  * 
- * @param cpu m, 1000m = 1core
- * @param memory MB
+ * @param id 
+ * @param name 
  */
-data class Resource(
-    /* m, 1000m = 1core */
-    val cpu: kotlin.Int,
-    /* MB */
-    val memory: kotlin.Int
+data class SimpleCourse(
+    val id: kotlin.Int,
+    val name: kotlin.String
 ) 
 

src/main/kotlin/cn/edu/buaa/scs/controller/models/GetStatResourcePoolsResourcePoolIdUsed200Response.kt

@@ -91,6 +91,26 @@ interface User : Entity<User>, IEntity {
         return mysql.assistants.exists { (it.studentId eq this.id).and(it.courseId.inList(courseIdList)) }
     }
 
+    fun getAdminCourses(): List<Pair<Int, String>> {
+        return when (this.role) {
+            UserRole.STUDENT -> {
+                mysql.assistants
+                    .filter { it.studentId.eq(this.id) }.map { it.courseId }
+                    .toList()
+                    .map { it.toInt() }
+                    .let { courseIdList ->
+                        if (courseIdList.isEmpty()) emptyList()
+                        else mysql.courses.filter { it.id.inList(courseIdList) }.map { it.id to it.name }.toList()
+                    }
+            }
+
+            UserRole.SYS -> emptyList()
+            UserRole.TEACHER -> {
+                mysql.courses.filter { it.teacherId.eq(this.id) }.map { it.id to it.name }.toList()
+            }
+        }
+    }
+
     fun getAssistantCourseIdList(): List<Int> =
         mysql.assistants.filter { it.studentId eq this.id }.map { it.courseId.tryToInt() }.filterNotNull().distinct()
 
@@ -176,4 +196,4 @@ object Users : Table<User>("user") {
         mysql.users.find { Users.id eq id }
 }
 
-val Database.users get() = this.sequenceOf(Users)
+val Database.users get() = this.sequenceOf(Users)

src/main/kotlin/cn/edu/buaa/scs/controller/models/LoginUserResponse.kt

@@ -4,6 +4,7 @@ import cn.edu.buaa.scs.application
 import cn.edu.buaa.scs.auth.generateRSAToken
 import cn.edu.buaa.scs.cache.authRedis
 import cn.edu.buaa.scs.controller.models.LoginUserResponse
+import cn.edu.buaa.scs.controller.models.SimpleCourse
 import cn.edu.buaa.scs.model.*
 import cn.edu.buaa.scs.storage.mysql
 import cn.edu.buaa.scs.utils.*
@@ -119,6 +120,7 @@ class AuthService(val call: ApplicationCall) : IService {
             role = if (user.isStudent()) "student" else if (user.isTeacher()) "teacher" else "superAdmin",
             paasToken = user.paasToken,
             isAssistant = mysql.assistants.exists { it.studentId.eq(user.id) },
+            adminCourses = user.getAdminCourses().map { SimpleCourse(it.first, it.second) }
         )
     }