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 { throw new Error("Method not implemented."); } - public async createUserAccountForDiscordUser(discordUser: GuildMember, guildId: string, requestId: string): Promise { + + private generatePasswordForUser(user: GuildMember): string { + return user.nickname + user.user.discriminator + } + + public async createUserAccountForDiscordUser(discordUser: GuildMember, guildId?: string, requestId?: string): Promise { 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 { - 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 { + 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 { + 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 { 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> { + 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 { + 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) + } + + } }