jellyfin-discord-bot/server/jellyfin/handler.ts

import { GuildMember } from "discord.js";
import { Config } from "../configuration";
import { Maybe, PermissionLevel } from "../interfaces";
import { logger } from "../logger";
import { CreateUserByNameOperationRequest, DeleteUserRequest, GetItemsRequest, ItemsApi, SystemApi, UpdateUserPasswordOperationRequest, UpdateUserPolicyOperationRequest, UserApi } from "./apis";
import { BaseItemDto, UpdateUserPasswordRequest } from "./models";
import { UserDto } from "./models/UserDto";
import { Configuration, ConfigurationParameters } from "./runtime";


export class JellyfinHandler {

  private userApi: UserApi
  private systemApi: SystemApi
  private moviesApi: ItemsApi
  private token: string
  private authHeader: { headers: { 'X-Emby-Authorization': string } }
  private config: Config
  private serverName = "";

  public async ServerName(): Promise<string> {
    if (this.serverName === "") {
      const info = await this.systemApi.getSystemInfo(this.authHeader)
      this.serverName = info.serverName ?? this.config.bot.jellyfin_url
    }
    return this.serverName
  }
  constructor(_config: Config, _userApi?: UserApi, _systemApi?: SystemApi, _itemsApi?: ItemsApi) {
    this.config = _config
    this.token = this.config.bot.jellfin_token
    this.authHeader = {
      headers: {
        "X-Emby-Authorization": this.config.bot.workaround_token
      }
    }
    const userApiConfigurationParams: ConfigurationParameters = {
      basePath: this.config.bot.jellyfin_url,
      headers: this.authHeader.headers
    }
    const systemApiConfigurationParams: ConfigurationParameters = {
      basePath: this.config.bot.jellyfin_url,
      headers: this.authHeader.headers
    }
    const libraryApiConfigurationParams: ConfigurationParameters = {
      basePath: this.config.bot.jellyfin_url,
      headers: this.authHeader.headers
    }

    this.userApi = _userApi ?? new UserApi(new Configuration(userApiConfigurationParams))
    this.systemApi = _systemApi ?? new SystemApi(new Configuration(systemApiConfigurationParams))
    this.moviesApi = _itemsApi ?? new ItemsApi(new Configuration(libraryApiConfigurationParams))
    logger.info(`Initialized Jellyfin handler`, { requestId: 'Init' })
  }

  private generateJFUserName(discordUser: GuildMember, level: PermissionLevel): string {
    return `${discordUser.displayName}${level == "TEMPORARY" ? "_tmp" : ""}`
  }

  private generatePasswordForUser(): string {
    return (Math.random() * 10000 + 10000).toFixed(0)
  }

  public async createUserAccountForDiscordUser(discordUser: GuildMember, level: PermissionLevel, guildId?: string, requestId?: string): Promise<UserDto> {
    const newUserName = this.generateJFUserName(discordUser, level)
    logger.info(`New Username for ${discordUser.displayName}: ${newUserName}`, { guildId, requestId })
    const req: CreateUserByNameOperationRequest = {
      createUserByNameRequest: {
        name: newUserName,
        password: this.generatePasswordForUser(),
      }
    }
    logger.debug(JSON.stringify(req), { requestId, guildId })
    const createResult = await this.userApi.createUserByName(req)
    if (createResult) {
      (await discordUser.createDM()).send(`Ich hab dir mal nen Account angelegt :)\nDein Username ist ${createResult.name}, dein Password ist "${req.createUserByNameRequest.password}"!`)
      return createResult
    }
    else throw new Error('Could not create User in Jellyfin')
  }

  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}`, { guildId: discordUser.guild.id, requestId })
    return jfuser !== undefined
  }

  public async getCurrentUsers(guildId: string, requestId?: string): Promise<UserDto[]> {
    try {
      logger.info(`Fetching current users from Jellyfin`, { requestId, guildId })
      const result = await this.userApi.getUsers(undefined, this.authHeader)
      return result
    } catch (error) {
      logger.error(`Could not fetch current users from jellyfin`, { guildId, requestId })
    }
    return []
  }

  public async getUser(discordUser: GuildMember, requestId?: string): Promise<Maybe<UserDto>> {
    logger.info(`Getting user for discord member ${discordUser.displayName}`, { requestId, guildId: discordUser.guild.id })
    const jfUsers = await this.getCurrentUsers(discordUser.guild.id, requestId)
    const foundUser = jfUsers.find(x => x.name?.includes(discordUser.displayName))
    return foundUser
  }

  public async removeUser(newMember: GuildMember, level: PermissionLevel, requestId?: string) {
    logger.info(`${level == "TEMPORARY" ? "Deleting" : "Disabling"} user ${newMember.displayName}, but method is not implemented`, { requestId, guildId: newMember.guild.id })
    const jfuser = await this.getUser(newMember, requestId)
    if (jfuser && jfuser.id) {
      if (level === "TEMPORARY") {
        const r: DeleteUserRequest = {
          userId: jfuser.id
        }
        this.userApi.deleteUser(r)
      }
      else
        await this.disableUser(jfuser, newMember.guild.id, requestId)
    }
  }

  public async purge(guildId: string, requestId?: string) {
    logger.info("Deleting tmp users", { requestId, guildId })
    const users = (await this.userApi.getUsers()).filter(user => user.name?.endsWith("_tmp"))

    users.forEach(user => {
      if (user.id) {
        const r: DeleteUserRequest = {
          userId: user.id
        }
        this.userApi.deleteUser(r)
      }
    })
  }

  public async resetUserPasswort(member: GuildMember, requestId?: string) {
    logger.info(`Resetting password for user ${member.displayName}`, { requestId, guildId: member.guild.id })
    const jfUser = await this.getUser(member, requestId)
    if (jfUser && jfUser.id) {

      // const reset: UpdateUserPasswordRequest = {
      //   resetPassword: true
      // }

      // const shit: UpdateUserPasswordOperationRequest = {
      //   updateUserPasswordRequest: reset,
      //   userId: jfUser.id
      // }

      // logger.info(JSON.stringify(jfUser.policy, null, 2))

      // logger.info("Resetting password", {requestId})
      // await this.userApi.updateUserPassword(shit);

      const password = this.generatePasswordForUser()
      const passwordRequest: UpdateUserPasswordRequest = {
        // resetPassword: true,
        currentPw: "",
        newPw: password
      }

      const passwordOperationRequest: UpdateUserPasswordOperationRequest = {
        updateUserPasswordRequest: passwordRequest,
        userId: jfUser.id
      }

      logger.info("Setting new password", { requestId, guildId: member.guild.id })
      await this.userApi.updateUserPassword(passwordOperationRequest);


      (await member.createDM()).send(`Hier ist dein neues Passwort: ${password}`)
    } else {
      (await member.createDM()).send("Ich konnte leider keinen User von dir auf Jellyfin finden. Bitte melde dich bei Markus oder Samantha!")
    }


  }

  public async disableUser(user: UserDto, guildId?: string, requestId?: string): Promise<void> {
    if (user.id) {
      const jfUser = await this.getUser(<GuildMember>{ displayName: user.name, guild: { id: guildId } }, requestId)
      logger.info(`Trying to disable user: ${user.name}|${user.id}|${JSON.stringify(jfUser, null, 2)}`, { guildId, requestId })
      const r: UpdateUserPolicyOperationRequest = {
        userId: user.id ?? "",
        updateUserPolicyRequest: {
          ...jfUser?.policy,
          isDisabled: true,
        }
      }
      await this.userApi.updateUserPolicy(r)
      logger.info(`Succeeded with disabling user: ${user.name}`, { guildId, requestId })
    }
    else {
      logger.error(`Can not disable user ${JSON.stringify(user)}, has no id?!`, { requestId, guildId })
    }
  }

  public async enableUser(user: UserDto, guildId: string, requestId?: string): Promise<void> {
    if (user.id) {
      const jfUser = await this.getUser(<GuildMember>{ displayName: user.name, guild: { id: guildId } }, requestId)
      logger.info(`Trying to enable user: ${user.name}|${user.id}|${JSON.stringify(jfUser, null, 2)}`, { guildId, requestId })
      const r: UpdateUserPolicyOperationRequest = {
        userId: user.id ?? "",
        updateUserPolicyRequest: {
          ...jfUser?.policy,
          isDisabled: false,
        }
      }
      await this.userApi.updateUserPolicy(r)
      logger.info(`Succeeded with enabling user: ${user.name}`, { guildId, requestId })
    }
    else {
      logger.error(`Can not enable user ${JSON.stringify(user)}, has no id?!`, { requestId, guildId })
    }
  }

  public async upsertUser(newMember: GuildMember, level: PermissionLevel, requestId?: string): Promise<UserUpsertResult> {
    logger.info(`Trying to upsert user ${newMember.displayName}, with permissionLevel ${level}`, { guildId: newMember.guild.id, requestId })
    const jfuser = await this.getUser(newMember, requestId)
    if (jfuser && !jfuser.policy?.isDisabled) {
      logger.info(`User with name ${newMember.displayName} is already present`, { guildId: newMember.guild.id, requestId })
      await this.enableUser(jfuser, newMember.guild.id, requestId)
      return UserUpsertResult.enabled
    } else {
      this.createUserAccountForDiscordUser(newMember, level, newMember.guild.id, requestId)
      return UserUpsertResult.created
    }
  }

  public async getAllMovies(guildId: string, requestId: string): Promise<BaseItemDto[]> {
    logger.info("requesting all movies from jellyfin", { guildId, requestId })

    const liloJfUser = await this.getUser(<GuildMember>{ guild: { id: guildId }, displayName: "lilo" }, requestId)

    const searchParams: GetItemsRequest = {
      userId: liloJfUser?.id,
      parentId: this.config.bot.jf_collection_id // collection ID for all movies
    }
    const movies = (await (this.moviesApi.getItems(searchParams))).items?.filter(item => !item.isFolder)
    // logger.debug(JSON.stringify(movies, null, 2), { guildId: guildId, requestId })
    logger.info(`Found ${movies?.length} movies in total`, { guildId, requestId })
    return movies ?? []
  }

  public async getRandomMovies(count: number, guildId: string, requestId: string): Promise<BaseItemDto[]> {
    logger.info(`${count} random movies requested.`, { guildId, requestId })
    const allMovies = await this.getAllMovies(guildId, requestId)
    if (count >= allMovies.length) {
      logger.info(`${count} random movies requested but found only ${allMovies.length}. Returning all Movies.`, { guildId, requestId })
      return allMovies
    }
    const movies: BaseItemDto[] = []
    for (let i = 0; i < count; i++) {
      const index = Math.floor(Math.random() * allMovies.length)
      movies.push(...allMovies.splice(index, 1)) // maybe out of bounds? ? 
    }

    return movies
  }


}
export enum UserUpsertResult { enabled, created }