import { Command, Session } from '@odyssey/models'
import Pagination from './pagination'
import {
  CommandObject,
  SessionDetail,
  SessionDocument,
  SessionObject,
} from '../interface.js'

function formatTime(time: number) {
  const seconds = Math.floor(time / 1000)
  const minutes = Math.floor(seconds / 60)
  const hours = Math.floor(minutes / 60)
  return `${hours}h ${minutes % 60}m ${seconds % 60}s`
}

export async function sessionList(
  pagination: Pagination
): Promise<SessionObject[]> {
  const sessions = await Session.aggregate([
    pagination.match('_id'),
    {
      $lookup: {
        from: 'users',
        localField: 'user',
        foreignField: '_id',
        as: 'user',
      },
    },
    { $unwind: '$user' },
    pagination.sort('_id'),
    pagination.limit(),
  ])

  return sessions.map((session) => {
    const {
      _id,
      user,
      quest,
      locale,
      status,
      createdAt,
      finishedAt,
      lastActivityAt,
    } = session
    return {
      _id,
      user: { _id: user._id, username: user.username },
      quest,
      locale: locale ?? 'zh-TW',
      status,
      createdAt: createdAt?.toLocaleString(),
      finishedAt: finishedAt?.toLocaleString(),
      lastActivityAt: lastActivityAt?.toLocaleString(),
      usedTime:
        lastActivityAt && createdAt
          ? formatTime(lastActivityAt.getTime() - createdAt.getTime())
          : 'N/A', // Ensure both dates are present
      commandCount: 0,
    }
  })
}

export async function sessionDetail(id: string): Promise<SessionDetail> {
  const session = (await Session.findById(id)
    .populate('user')
    .exec()) as unknown as SessionDocument
  const commands: CommandObject[] = (await Command.find({ session: id })).map(
    ({ command, pwd, output, error, createdAt, stage }): CommandObject => ({
      command: command?.slice(0, 20),
      pwd,
      stage,
      output: output?.slice(0, 50),
      error: error?.slice(0, 50),
      createdAt: createdAt?.toLocaleString(),
    })
  )
  const {
    _id,
    user,
    quest,
    locale,
    status,
    createdAt,
    finishedAt,
    lastActivityAt,
  } = session
  return {
    _id,
    user: { _id: user._id, username: user.username },
    quest,
    locale: locale ?? 'zh-TW',
    status,
    createdAt: createdAt?.toLocaleString(),
    finishedAt: finishedAt?.toLocaleString(),
    lastActivityAt: lastActivityAt?.toLocaleDateString(),
    commandCount: commands.length,
    usedTime: finishedAt
      ? formatTime(finishedAt.getTime() - createdAt.getTime())
      : '',
    commands,
  }
}