Migrate models to TypeScript

Co-authored-by: David Mehren <dmehren1@gmail.com> Co-authored-by: Yannick Bungers <git@innay.de> Co-authored-by: Philipp Hochkamp <me@phochkamp.de> Co-authored-by: nzbr <mail@nzbr.de> Signed-off-by: David Mehren <dmehren1@gmail.com>
2025-05-18 17:25:16 -04:00 · 2020-04-10 22:06:12 +02:00 · 2020-04-10 22:06:12 +02:00 · 1d4107fe90
commit 1d4107fe90
parent 54cd556f2f
13 changed files with 565 additions and 539 deletions
--- a/lib/models/user.ts
+++ b/lib/models/user.ts
@ -0,0 +1,146 @@
+import { Note } from './note';
+import { Table, BeforeCreate, BeforeUpdate, HasMany, Unique, IsUUID, IsEmail, Column, DataType, PrimaryKey, Model, Default } from 'sequelize-typescript';
+import scrypt from 'scrypt-kdf';
+import { generateAvatarURL } from '../letter-avatars';
+import logger from '../logger';
+
+var Sequelize = require('sequelize')
+// core
+
+@Table
+export class User extends Model<User> {
+
+  @PrimaryKey
+  @Default(Sequelize.UUIDV4)
+  @Column(DataType.UUID)
+  id: string;
+
+  @Unique
+  @Column(DataType.STRING)
+  profileid: string;
+
+  @Column(DataType.TEXT)
+  profile: string;
+
+  @Column(DataType.TEXT)
+  histroy: string;
+
+  @Column(DataType.TEXT)
+  accessToken: string;
+
+  @Column(DataType.TEXT)
+  refreshToken: string;
+
+  @IsUUID(4)
+  @Column
+  deleteToken: string;
+
+  @IsEmail
+  @Column(DataType.TEXT)
+  email: string;
+
+  @Column(DataType.TEXT)
+  password: string;
+
+  verifyPassword(attempt: string) {
+    return scrypt.verify(Buffer.from(this.password, 'hex'), attempt);
+  }
+
+  @HasMany(() => Note, { foreignKey: 'lastchangeuserId', constraints: false })
+  @HasMany(() => Note, { foreignKey: 'ownerId', constraints: false })
+  
+  static parsePhotoByProfile(profile: any, bigger: boolean) {
+    let photo: string;
+    switch (profile.provider) {
+      case 'facebook':
+        photo = 'https://graph.facebook.com/' + profile.id + '/picture'
+        if (bigger) photo += '?width=400'
+        else photo += '?width=96'
+        break
+      case 'twitter':
+        photo = 'https://twitter.com/' + profile.username + '/profile_image'
+        if (bigger) photo += '?size=original'
+        else photo += '?size=bigger'
+        break
+      case 'github':
+        photo = 'https://avatars.githubusercontent.com/u/' + profile.id
+        if (bigger) photo += '?s=400'
+        else photo += '?s=96'
+        break
+      case 'gitlab':
+        photo = profile.avatarUrl
+        if (photo) {
+          if (bigger) photo = photo.replace(/(\?s=)\d*$/i, '$1400')
+          else photo = photo.replace(/(\?s=)\d*$/i, '$196')
+        } else {
+          photo = generateAvatarURL(profile.username)
+        }
+        break
+      case 'dropbox':
+        photo = generateAvatarURL('', profile.emails[0].value, bigger)
+        break
+      case 'google':
+        photo = profile.photos[0].value
+        if (bigger) photo = photo.replace(/(\?sz=)\d*$/i, '$1400')
+        else photo = photo.replace(/(\?sz=)\d*$/i, '$196')
+        break
+      case 'ldap':
+        photo = generateAvatarURL(profile.username, profile.emails[0], bigger)
+        break
+      case 'saml':
+        photo = generateAvatarURL(profile.username, profile.emails[0], bigger)
+        break
+      default:
+        photo = generateAvatarURL(profile.username)
+        break
+    }
+    return photo
+  }
+  static parseProfileByEmail(email: string) {
+    return {
+      name: email.substring(0, email.lastIndexOf('@')),
+      photo: generateAvatarURL('', email, false),
+      biggerphoto: generateAvatarURL('', email, true)
+    }
+  }
+
+  @BeforeUpdate
+  @BeforeCreate
+  static async updatePasswordHashHook(user: User) {
+    // suggested way to hash passwords to be able to do this asynchronously:
+    // @see https://github.com/sequelize/sequelize/issues/1821#issuecomment-44265819
+
+    if (!user.changed('password')) {
+      return Promise.resolve();
+    }
+
+    return scrypt.kdf(user.getDataValue('password'), { logN: 15, r: 8, p: 1 }).then(keyBuf => {
+      user.setDataValue('password', keyBuf.toString('hex'));
+    })
+  }
+
+  static getProfile(user: User) {
+    if (!user) {
+      return null
+    }
+    return user.profile ? user.parseProfile(user.profile) : (user.email ? User.parseProfileByEmail(user.email) : null)
+  }
+
+  parseProfile(profile: any) {
+    try {
+      profile = JSON.parse(profile)
+    } catch (err) {
+      logger.error(err)
+      profile = null
+    }
+    if (profile) {
+      profile = {
+        name: profile.displayName || profile.username,
+        photo: User.parsePhotoByProfile(profile, false),
+        biggerphoto: User.parsePhotoByProfile(profile, true)
+      }
+    }
+    return profile
+  }
+
+}