discord-jellyfin-bot/src/commands/play.comands.ts

import { TransformPipe } from '@discord-nestjs/common';

import {
  Command,
  DiscordTransformedCommand,
  On,
  Payload,
  TransformedCommandExecutionContext,
  UsePipes,
} from '@discord-nestjs/core';
import { Logger } from '@nestjs/common/services';
import {
  ComponentType,
  Events,
  GuildMember,
  Interaction,
  InteractionReplyOptions,
} from 'discord.js';
import { JellyfinSearchService } from '../clients/jellyfin/jellyfin.search.service';
import { TrackRequestDto } from '../models/track-request.dto';

import { DiscordMessageService } from '../clients/discord/discord.message.service';

import { formatDuration, intervalToDuration } from 'date-fns';
import { DiscordVoiceService } from '../clients/discord/discord.voice.service';
import { JellyfinStreamBuilderService } from '../clients/jellyfin/jellyfin.stream.builder.service';
import { PlaybackService } from '../playback/playback.service';
import { Constants } from '../utils/constants';
import { trimStringToFixedLength } from '../utils/stringUtils';

@Command({
  name: 'play',
  description: 'Search for an item on your Jellyfin instance',
})
@UsePipes(TransformPipe)
export class PlayItemCommand
  implements DiscordTransformedCommand<TrackRequestDto>
{
  private readonly logger: Logger = new Logger(PlayItemCommand.name);

  constructor(
    private readonly jellyfinSearchService: JellyfinSearchService,
    private readonly discordMessageService: DiscordMessageService,
    private readonly discordVoiceService: DiscordVoiceService,
    private readonly playbackService: PlaybackService,
    private readonly jellyfinStreamBuilder: JellyfinStreamBuilderService,
  ) {}

  async handler(
    @Payload() dto: TrackRequestDto,
    executionContext: TransformedCommandExecutionContext<any>,
  ): Promise<InteractionReplyOptions | string> {
    const items = await this.jellyfinSearchService.search(dto.search);

    if (items.length < 1) {
      return {
        embeds: [
          this.discordMessageService.buildErrorMessage({
            title: 'No results for your search query found',
            description: `I was not able to find any matches for your query \`\`${dto.search}\`\`. Please check that I have access to the desired libraries and that your query is not misspelled`,
          }),
        ],
      };
    }

    const firstItems = items.slice(0, 10);

    const lines: string[] = firstItems.map(
      (item) =>
        `:white_small_square: ${trimStringToFixedLength(
          this.markSearchTermOverlap(item.Name, dto.search),
          30,
        )} [${item.Artists.join(', ')}] *(${item.Type})*`,
    );

    let description =
      'I have found **' +
      items.length +
      '** results for your search ``' +
      dto.search +
      '``.';

    if (items.length > 10) {
      description +=
        '\nSince the results exceed 10 items, I truncated them for better readability.';
    }

    description += '\n\n' + lines.join('\n');

    const emojiForType = (type: string) => {
      switch (type) {
        case 'Audio':
          return '🎵';
        case 'Playlist':
          return '📚';
        default:
          return undefined;
      }
    };

    const selectOptions: { label: string; value: string; emoji?: string }[] =
      firstItems.map((item) => ({
        label: `${item.Name} [${item.Artists.join(', ')}]`,
        value: item.Id,
        emoji: emojiForType(item.Type),
      }));

    return {
      embeds: [
        this.discordMessageService.buildMessage({
          title: 'a',
          description: description,
          mixin(embedBuilder) {
            return embedBuilder.setAuthor({
              name: 'Jellyfin Search Results',
              iconURL: Constants.Design.Icons.JellyfinLogo,
            });
          },
        }),
      ],
      components: [
        {
          type: ComponentType.ActionRow,
          components: [
            {
              type: ComponentType.StringSelect,
              customId: 'searchItemSelect',
              options: selectOptions,
            },
          ],
        },
      ],
    };
  }

  @On(Events.InteractionCreate)
  async onStringSelect(interaction: Interaction) {
    if (!interaction.isStringSelectMenu()) return;

    if (interaction.customId !== 'searchItemSelect') {
      return;
    }

    if (interaction.values.length !== 1) {
      this.logger.warn(
        `Failed to process interaction select with values [${interaction.values.length}]`,
      );
      return;
    }

    const item = await this.jellyfinSearchService.getById(
      interaction.values[0],
    );

    const guildMember = interaction.member as GuildMember;

    const tryResult =
      this.discordVoiceService.tryJoinChannelAndEstablishVoiceConnection(
        guildMember,
      );

    if (!tryResult.success) {
      this.logger.warn(
        `Unable to process select result because the member was not in a voice channcel`,
      );
      const replyOptions = tryResult.reply as InteractionReplyOptions;
      await interaction.update({
        embeds: replyOptions.embeds,
        content: undefined,
        components: [],
      });
      return;
    }

    const bitrate = guildMember.voice.channel.bitrate;

    const stream = await this.jellyfinStreamBuilder.buildStreamUrl(
      item.Id,
      bitrate,
    );

    const milliseconds = item.RunTimeTicks / 10000;

    const duration = formatDuration(
      intervalToDuration({
        start: milliseconds,
        end: 0,
      }),
    );

    const addedIndex = this.playbackService.eneuqueTrack({
      jellyfinId: item.Id,
      name: item.Name,
      durationInMilliseconds: milliseconds,
      streamUrl: stream,
    });

    const artists = item.Artists.join(', ');

    await interaction.update({
      embeds: [
        this.discordMessageService.buildMessage({
          title: 'Jellyfin Search',
          description: `**Duration**: ${duration}\n**Artists**: ${artists}\n\nTrack was added to the queue at position ${addedIndex}`,
        }),
      ],
      components: [],
    });
  }

  private markSearchTermOverlap(value: string, searchTerm: string) {
    const startIndex = value.indexOf(searchTerm);
    const actualValue = value.substring(
      startIndex,
      startIndex + 1 + searchTerm.length,
    );
    return `${value.substring(
      0,
      startIndex,
    )}**${actualValue}**${value.substring(
      startIndex + 1 + actualValue.length,
    )}`;
  }
}