YouTube.js/src/core/PlaylistManager.ts

import Playlist from '../parser/youtube/Playlist';
import Actions from './Actions';
import Feed from './Feed';

import { InnertubeError, throwIfMissing } from '../utils/Utils';

class PlaylistManager {
  #actions;

  constructor(actions: Actions) {
    this.#actions = actions;
  }

  /**
   * Creates a playlist.
   */
  async create(title: string, video_ids: string[]) {
    throwIfMissing({ title, video_ids });

    const response = await this.#actions.execute('/playlist/create', { title, ids: video_ids, parse: false });

    return {
      success: response.success,
      status_code: response.status_code,
      playlist_id: response.data.playlistId,
      data: response.data
    };
  }

  /**
   * Deletes a given playlist.
   */
  async delete(playlist_id: string) {
    throwIfMissing({ playlist_id });

    const response = await this.#actions.execute('playlist/delete', { playlistId: playlist_id });

    return {
      playlist_id,
      success: response.success,
      status_code: response.status_code,
      data: response.data
    };
  }

  /**
   * Adds videos to a given playlist.
   */
  async addVideos(playlist_id: string, video_ids: string[]) {
    throwIfMissing({ playlist_id, video_ids });

    const response = await this.#actions.execute('/browse/edit_playlist', {
      playlistId: playlist_id,
      actions: video_ids.map((id) => ({
        action: 'ACTION_ADD_VIDEO',
        addedVideoId: id
      })),
      parse: false
    });

    return {
      playlist_id,
      action_result: response.data.actions // TODO: implement actions in the parser
    };
  }

  /**
   * Removes videos from a given playlist.
   */
  async removeVideos(playlist_id: string, video_ids: string[]) {
    throwIfMissing({ playlist_id, video_ids });

    const info = await this.#actions.execute('/browse', { browseId: `VL${playlist_id}`, parse: true });
    const playlist = new Playlist(this.#actions, info, true);

    if (!playlist.info.is_editable)
      throw new InnertubeError('This playlist cannot be edited.', playlist_id);

    const payload = {
      playlistId: playlist_id,
      actions: [] as {
        action: string;
        setVideoId: string;
      }[]
    };

    const getSetVideoIds = async (pl: Feed): Promise<void> => {
      const videos = pl.videos.filter((video) => video_ids.includes(video.key('id').string()));

      videos.forEach((video) =>
        payload.actions.push({
          action: 'ACTION_REMOVE_VIDEO',
          setVideoId: video.key('set_video_id').string()
        })
      );

      if (payload.actions.length < video_ids.length) {
        const next = await pl.getContinuation();
        return getSetVideoIds(next);
      }
    };

    await getSetVideoIds(playlist);

    if (!payload.actions.length)
      throw new InnertubeError('Given video ids were not found in this playlist.', video_ids);

    const response = await this.#actions.execute('/browse/edit_playlist', { ...payload, parse: false });

    return {
      playlist_id,
      action_result: response.data.actions // TODO: implement actions in the parser
    };
  }

  /**
   * Moves a video to a new position within a given playlist.
   */
  async moveVideo(playlist_id: string, moved_video_id: string, predecessor_video_id: string) {
    throwIfMissing({ playlist_id, moved_video_id, predecessor_video_id });

    const info = await this.#actions.execute('/browse', { browseId: `VL${playlist_id}`, parse: true });
    const playlist = new Playlist(this.#actions, info, true);

    if (!playlist.info.is_editable)
      throw new InnertubeError('This playlist cannot be edited.', playlist_id);

    const payload = {
      playlistId: playlist_id,
      actions: [] as {
        action: string,
        setVideoId?: string,
        movedSetVideoIdPredecessor?: string
      }[]
    };

    let set_video_id_0: string | undefined, set_video_id_1: string | undefined;

    const getSetVideoIds = async (pl: Feed): Promise<void> => {
      const video_0 = pl.videos.find((video) => moved_video_id === video.key('id').string());
      const video_1 = pl.videos.find((video) => predecessor_video_id === video.key('id').string());

      set_video_id_0 = set_video_id_0 || video_0?.key('set_video_id').string();
      set_video_id_1 = set_video_id_1 || video_1?.key('set_video_id').string();

      if (!set_video_id_0 || !set_video_id_1) {
        const next = await pl.getContinuation();
        return getSetVideoIds(next);
      }
    };

    await getSetVideoIds(playlist);

    payload.actions.push({
      action: 'ACTION_MOVE_VIDEO_AFTER',
      setVideoId: set_video_id_0,
      movedSetVideoIdPredecessor: set_video_id_1
    });

    const response = await this.#actions.execute('/browse/edit_playlist', { ...payload, parse: false });

    return {
      playlist_id,
      action_result: response.data.actions // TODO: implement actions in the parser
    };
  }
}

export default PlaylistManager;