kenobi
/
jellyfin-discord-bot
2
1
Fork 0
Code Issues 17 Pull Requests 1 Actions Packages Projects 1 Releases Wiki Activity

WIP: update user, remove user

This commit is contained in:
mightypanders 2023-04-20 20:54:20 +02:00
parent 3494a6ea14
commit 8e3f62c670
1 changed files with 77 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

88
jellyfin/handler.ts
View File

@ -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)
}
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
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 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) {
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 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} but method is not implemented`)
const presence = await this.isUserAlreadyPresent(newMember, newMember.guild.id, requestId)
if (presence) {
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)
}
}
}