From 8e3f62c670be9eaff9e5d9a120c4ed7a556d8768 Mon Sep 17 00:00:00 2001
From: mightypanders <mighty.panders@posteo.de>
Date: Thu, 20 Apr 2023 20:54:20 +0200
Subject: [PATCH] WIP: update user, remove user

---
 jellyfin/handler.ts | 94 +++++++++++++++++++++++++++++++++++++--------
 1 file changed, 77 insertions(+), 17 deletions(-)

diff --git a/jellyfin/handler.ts b/jellyfin/handler.ts
index caa4c2b..416822d 100644
--- a/jellyfin/handler.ts
+++ b/jellyfin/handler.ts
@@ -4,6 +4,8 @@ import { Config } from "../server/configuration";
 import { UserApi } from "./api/userApi";
 import { logger } from "../server/logger";
 import { SystemApi } from "./api/systemApi";
+import { CreateUserByNameRequest, UpdateUserPolicyRequest, UpdateUserRequest } from "./api";
+import { Maybe } from "../server/interfaces";
 
 export class JellyfinHandler {
 
@@ -40,20 +42,46 @@ export class JellyfinHandler {
   public async addPermissionsToUserAccount(jfUserAccount: UserDto, guildId: string, requestId: string): Promise<UserDto> {
     throw new Error("Method not implemented.");
   }
-  public async createUserAccountForDiscordUser(discordUser: GuildMember, guildId: string, requestId: string): Promise<UserDto> {
+
+  private generatePasswordForUser(user: GuildMember): string {
+    return user.nickname + user.user.discriminator
+  }
+
+  public async createUserAccountForDiscordUser(discordUser: GuildMember, guildId?: string, requestId?: string): Promise<UserDto> {
     const newUserName = this.generateJFUserName(discordUser)
     logger.info(`New Username for ${discordUser.displayName}: ${newUserName}`, { guildId, requestId })
-    throw new Error("Method not implemented.");
+    const req: CreateUserByNameRequest = {
+      name: newUserName,
+      password: this.generatePasswordForUser(discordUser)
+    }
+    logger.debug(JSON.stringify(req))
+    const createResult = await this.userApi.createUserByName(req, this.authHeader)
+    if (createResult.body)
+      return createResult.body
+    else throw new Error('Could not create User in Jellyfin')
   }
-  public async isUserAlreadyPresent(discordUser: GuildMember, guildId: string, requestId?: string): Promise<boolean> {
-    const jfUsernameFromDiscordUsername = this.generateJFUserName(discordUser)
-    const jfUsers = await this.getCurrentUsers(guildId, requestId)
-    const currentJfUserNames = jfUsers.map(x => x.name)
-    logger.debug(`Current JF Usernames: ${JSON.stringify(currentJfUserNames)}`)
-    const presence = currentJfUserNames.find(x => x === jfUsernameFromDiscordUsername)
-    logger.debug(`Presence for DiscordUser ${discordUser.id}:${presence}`)
-    return presence !== undefined
+
+  public async disableJfUser(user: UserDto, guildId?: string, requestId?: string): Promise<void> {
+    if (user.id) {
+      logger.info(`Trying to disable user: ${user.name}|${user.id}`)
+      const r: UpdateUserPolicyRequest = {
+        ...user.policy,
+        isDisabled: true
+      }
+      await this.userApi.updateUserPolicy(user.id, r, this.authHeader)
+      logger.info(`Succeeded with disabling user: ${user.name}`)
+    }
+    else {
+      logger.error(`Can not disable user ${JSON.stringify(user)}, has no id?!`, { requestId, guildId })
+    }
   }
+
+  public async isUserAlreadyPresent(discordUser: GuildMember, requestId?: string): Promise<boolean> {
+    const jfuser = await this.getUser(discordUser, requestId)
+    logger.debug(`Presence for DiscordUser ${discordUser.id}:${jfuser !== undefined}`)
+    return jfuser !== undefined
+  }
+
   public async getCurrentUsers(guildId: string, requestId?: string): Promise<UserDto[]> {
     try {
       const result = await this.userApi.getUsers(undefined, undefined, this.authHeader)
@@ -64,15 +92,47 @@ export class JellyfinHandler {
     return []
   }
 
-  public async removeUser(newMember: GuildMember) {
-    logger.error(`Trying to remove user ${newMember.displayName}, but method is not implemented`)
+  public async getUser(discordUser: GuildMember, requestId?: string): Promise<Maybe<UserDto>> {
+    const jfUsernameFromDiscordUsername = this.generateJFUserName(discordUser)
+    const jfUsers = await this.getCurrentUsers(discordUser.guild.id, requestId)
+    const foundUser = jfUsers.find(x => x.name === jfUsernameFromDiscordUsername)
+    return foundUser
   }
-  public async upsertUser(newMember: GuildMember, level: string, requestId?: string) {
-    logger.error(`Trying to upsert user ${newMember.displayName}, with permissionLevel ${level} but method is not implemented`)
-    const presence = await this.isUserAlreadyPresent(newMember, newMember.guild.id, requestId)
-    if (presence) {
-      logger.info(`User with name ${newMember.displayName} is already present`)
+
+  public async removeUser(newMember: GuildMember, requestId?: string) {
+    logger.error(`Trying to remove user ${newMember.displayName}, but method is not implemented`)
+    const jfuser = await this.getUser(newMember, requestId)
+    if (jfuser) {
+      await this.disableJfUser(jfuser, newMember.guild.id, requestId)
     }
+  }
+
+  public async enableUser(user: UserDto, guildId?: string, requestId?: string): Promise<void> {
+    if (user.id) {
+      logger.info(`Trying to enable user: ${user.name}|${user.id}`)
+      const r: UpdateUserPolicyRequest = {
+        ...user.policy,
+        isDisabled: false
+      }
+      await this.userApi.updateUserPolicy(user.id, r, this.authHeader)
+      logger.info(`Succeeded with enabling user: ${user.name}`)
+    }
+    else {
+      logger.error(`Can not enable user ${JSON.stringify(user)}, has no id?!`, { requestId, guildId })
+    }
+  }
+
+  public async upsertUser(newMember: GuildMember, level: string, requestId?: string) {
+    logger.error(`Trying to upsert user ${newMember.displayName}, with permissionLevel ${level}`)
+    const jfuser = await this.getUser(newMember, requestId)
+    if (jfuser) {
+      logger.info(`User with name ${newMember.displayName} is already present`)
+      await this.enableUser(jfuser, requestId)
+    } else {
+      this.createUserAccountForDiscordUser(newMember, newMember.guild.id, requestId)
+    }
+
+
 
   }
 }