Add ability to choose what policy we have for NSFW videos

[github/Chocobozzz/PeerTube.git] / client / src / app / shared / users / user.model.ts

diff --git a/client/src/app/shared/users/user.model.ts b/client/src/app/shared/users/user.model.ts
index 52d89e0049f53991d6686fcc99d68fe3537ab802..2bdc48a1dc81a381ba0cda7915e7f0069aca296d 100644 (file)
--- a/client/src/app/shared/users/user.model.ts
+++ b/client/src/app/shared/users/user.model.ts
@@ -1,20 +1,70 @@
-export class User {
-  id: number;
-  username: string;
-  role: string;
-  createdAt: Date;
-
-  constructor(hash: { id: number, username: string, role: string, createdAt?: Date }) {
-    this.id = hash.id;
-    this.username = hash.username;
-    this.role = hash.role;
-
-    if (hash.createdAt) {
-      this.createdAt = hash.createdAt;
+import { hasUserRight, User as UserServerModel, UserRight, UserRole, VideoChannel } from '../../../../../shared'
+import { Account } from '../account/account.model'
+import { NSFWPolicyType } from '../../../../../shared/models/videos/nsfw-policy.type'
+
+export type UserConstructorHash = {
+  id: number,
+  username: string,
+  email: string,
+  role: UserRole,
+  videoQuota?: number,
+  nsfwPolicy?: NSFWPolicyType,
+  autoPlayVideo?: boolean,
+  createdAt?: Date,
+  account?: Account,
+  videoChannels?: VideoChannel[]
+}
+export class User implements UserServerModel {
+  id: number
+  username: string
+  email: string
+  role: UserRole
+  nsfwPolicy: NSFWPolicyType
+  autoPlayVideo: boolean
+  videoQuota: number
+  account: Account
+  videoChannels: VideoChannel[]
+  createdAt: Date
+
+  constructor (hash: UserConstructorHash) {
+    this.id = hash.id
+    this.username = hash.username
+    this.email = hash.email
+    this.role = hash.role
+    this.account = hash.account
+
+    if (hash.videoChannels !== undefined) {
+      this.videoChannels = hash.videoChannels
     }
+
+    if (hash.videoQuota !== undefined) {
+      this.videoQuota = hash.videoQuota
+    }
+
+    if (hash.nsfwPolicy !== undefined) {
+      this.nsfwPolicy = hash.nsfwPolicy
+    }
+
+    if (hash.autoPlayVideo !== undefined) {
+      this.autoPlayVideo = hash.autoPlayVideo
+    }
+
+    if (hash.createdAt !== undefined) {
+      this.createdAt = hash.createdAt
+    }
+  }
+
+  hasRight (right: UserRight) {
+    return hasUserRight(this.role, right)
+  }
+
+  getAvatarUrl () {
+    return Account.GET_ACCOUNT_AVATAR_URL(this.account)
   }
 
-  isAdmin() {
-    return this.role === 'admin';
+  patch (obj: UserServerModel) {
+    for (const key of Object.keys(obj)) {
+      this[key] = obj[key]
+    }
   }
 }