Chocobozzz/PeerTube

import { omit, pick } from '@peertube/peertube-core-utils'
import {
  BooleanBothQuery,
  HttpStatusCode,
  ResultList,
  VideoExistInPlaylist,
  VideoPlaylist,
  VideoPlaylistCreate,
  VideoPlaylistCreateResult,
  VideoPlaylistElement,
  VideoPlaylistElementCreate,
  VideoPlaylistElementCreateResult,
  VideoPlaylistElementUpdate,
  VideoPlaylistPrivacy,
  VideoPlaylistPrivacyType,
  VideoPlaylistReorder,
  VideoPlaylistType,
  VideoPlaylistType_Type,
  VideoPlaylistUpdate
} from '@peertube/peertube-models'
import { unwrapBody } from '../requests/index.js'
import { AbstractCommand, OverrideCommandOptions } from '../shared/index.js'

export class PlaylistsCommand extends AbstractCommand {

  list (options: OverrideCommandOptions & {
    start?: number
    count?: number
    sort?: string
    playlistType?: VideoPlaylistType_Type
  } = {}) {
    const path = '/api/v1/video-playlists'
    const query = pick(options, [ 'start', 'count', 'sort', 'playlistType' ])

    return this.getRequestBody<ResultList<VideoPlaylist>>({
      ...options,

      path,
      query,
      implicitToken: false,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  listByChannel (options: OverrideCommandOptions & {
    handle: string
    start?: number
    count?: number
    sort?: string
    playlistType?: VideoPlaylistType_Type
  }) {
    const path = '/api/v1/video-channels/' + options.handle + '/video-playlists'
    const query = pick(options, [ 'start', 'count', 'sort', 'playlistType' ])

    return this.getRequestBody<ResultList<VideoPlaylist>>({
      ...options,

      path,
      query,
      implicitToken: false,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  listByAccount (options: OverrideCommandOptions & {
    handle: string
    start?: number
    count?: number
    sort?: string
    search?: string
    playlistType?: VideoPlaylistType_Type
  }) {
    const path = '/api/v1/accounts/' + options.handle + '/video-playlists'
    const query = pick(options, [ 'start', 'count', 'sort', 'search', 'playlistType' ])

    return this.getRequestBody<ResultList<VideoPlaylist>>({
      ...options,

      path,
      query,
      implicitToken: false,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  // ---------------------------------------------------------------------------

  get (options: OverrideCommandOptions & {
    playlistId: number | string
  }) {
    const { playlistId } = options
    const path = '/api/v1/video-playlists/' + playlistId

    return this.getRequestBody<VideoPlaylist>({
      ...options,

      path,
      implicitToken: false,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  async getWatchLater (options: OverrideCommandOptions & {
    handle: string
  }) {
    const { data: playlists } = await this.listByAccount({
      ...options,

      playlistType: VideoPlaylistType.WATCH_LATER
    })

    return playlists[0]
  }

  // ---------------------------------------------------------------------------

  listVideos (options: OverrideCommandOptions & {
    playlistId: number | string
    start?: number
    count?: number
    query?: { nsfw?: BooleanBothQuery }
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId + '/videos'
    const query = options.query ?? {}

    return this.getRequestBody<ResultList<VideoPlaylistElement>>({
      ...options,

      path,
      query: {
        ...query,
        start: options.start,
        count: options.count
      },
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  delete (options: OverrideCommandOptions & {
    playlistId: number | string
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId

    return this.deleteRequest({
      ...options,

      path,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
    })
  }

  async create (options: OverrideCommandOptions & {
    attributes: VideoPlaylistCreate
  }) {
    const path = '/api/v1/video-playlists'

    const fields = omit(options.attributes, [ 'thumbnailfile' ])

    const attaches = options.attributes.thumbnailfile
      ? { thumbnailfile: options.attributes.thumbnailfile }
      : {}

    const body = await unwrapBody<{ videoPlaylist: VideoPlaylistCreateResult }>(this.postUploadRequest({
      ...options,

      path,
      fields,
      attaches,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.OK_200
    }))

    return body.videoPlaylist
  }

  async quickCreate (options: OverrideCommandOptions & {
    displayName: string
    privacy?: VideoPlaylistPrivacyType
  }) {
    const { displayName, privacy = VideoPlaylistPrivacy.PUBLIC } = options

    const { videoChannels } = await this.server.users.getMyInfo({ token: options.token })

    return this.create({
      ...options,

      attributes: {
        displayName,
        privacy,
        videoChannelId: privacy === VideoPlaylistPrivacy.PUBLIC
          ? videoChannels[0].id
          : undefined
      }
    })
  }

  update (options: OverrideCommandOptions & {
    attributes: VideoPlaylistUpdate
    playlistId: number | string
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId

    const fields = omit(options.attributes, [ 'thumbnailfile' ])

    const attaches = options.attributes.thumbnailfile
      ? { thumbnailfile: options.attributes.thumbnailfile }
      : {}

    return this.putUploadRequest({
      ...options,

      path,
      fields,
      attaches,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
    })
  }

  async addElement (options: OverrideCommandOptions & {
    playlistId: number | string
    attributes: VideoPlaylistElementCreate | { videoId: string }
  }) {
    const attributes = {
      ...options.attributes,

      videoId: await this.server.videos.getId({ ...options, uuid: options.attributes.videoId })
    }

    const path = '/api/v1/video-playlists/' + options.playlistId + '/videos'

    const body = await unwrapBody<{ videoPlaylistElement: VideoPlaylistElementCreateResult }>(this.postBodyRequest({
      ...options,

      path,
      fields: attributes,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.OK_200
    }))

    return body.videoPlaylistElement
  }

  updateElement (options: OverrideCommandOptions & {
    playlistId: number | string
    elementId: number | string
    attributes: VideoPlaylistElementUpdate
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId + '/videos/' + options.elementId

    return this.putBodyRequest({
      ...options,

      path,
      fields: options.attributes,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
    })
  }

  removeElement (options: OverrideCommandOptions & {
    playlistId: number | string
    elementId: number
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId + '/videos/' + options.elementId

    return this.deleteRequest({
      ...options,

      path,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
    })
  }

  reorderElements (options: OverrideCommandOptions & {
    playlistId: number | string
    attributes: VideoPlaylistReorder
  }) {
    const path = '/api/v1/video-playlists/' + options.playlistId + '/videos/reorder'

    return this.postBodyRequest({
      ...options,

      path,
      fields: options.attributes,
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
    })
  }

  getPrivacies (options: OverrideCommandOptions = {}) {
    const path = '/api/v1/video-playlists/privacies'

    return this.getRequestBody<{ [ id: number ]: string }>({
      ...options,

      path,
      implicitToken: false,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }

  videosExist (options: OverrideCommandOptions & {
    videoIds: number[]
  }) {
    const { videoIds } = options
    const path = '/api/v1/users/me/video-playlists/videos-exist'

    return this.getRequestBody<VideoExistInPlaylist>({
      ...options,

      path,
      query: { videoIds },
      implicitToken: true,
      defaultExpectedStatus: HttpStatusCode.OK_200
    })
  }
}