import { Type, Static, TSchema } from '@sinclair/typebox'


export type RequestHeaders = Static<typeof RequestHeaders>
export const RequestHeaders = Type.Record(Type.String(), Type.Any())

export type SharedSummarizeOpts<T extends TSchema> = Static<ReturnType<typeof SharedSummarizeOpts<T>>>
export const SharedSummarizeOpts = <T extends TSchema>(T: T) => Type.Object({
extraOpts: Type.Optional(T),
headers: Type.Optional(RequestHeaders)
})

export type SharedSummarizeExtraOpts = Static<typeof SharedSummarizeExtraOpts>
export const SharedSummarizeExtraOpts = Type.Object({
bypassCache: Type.Optional(Type.Boolean()),
sessionId: Type.Optional(Type.String())
})

export enum EnumSummarizeStatus { UNKNOWN, GENERATING, SUCCESS, FAILED, NOT_FOUND_IN_CACHE }

export type SummarizeStatus = Static<typeof SummarizeStatus>
export const SummarizeStatus = Type.Enum(EnumSummarizeStatus)

export type SummarizeType = Static<typeof SummarizeType>
export const SummarizeType = Type.Union([
Type.Literal("article"),
Type.Literal("text"),
Type.Literal("file"),
Type.Literal("video")
])

export type MinimalSummarizeResponse<T extends TSchema> = Static<ReturnType<typeof MinimalSummarizeResponse<T>>>
export const MinimalSummarizeResponse = <T extends TSchema>(T: T) => Type.Object({
pollIntervalMs: Type.Number(),
sessionId: Type.String(),
statusCode: SummarizeStatus,
title: Type.String(),
type: T
})

export type SummarizeResponse<T extends TSchema> = Static<ReturnType<typeof SummarizeResponse<T>>>
export const SummarizeResponse = <T extends TSchema>(T: T) => Type.Intersect([
MinimalSummarizeResponse(T),
Type.Object({
chapters: Type.Array(SummarizeChapter(SummarizeThesisWithLink)),
haveChapters: Type.Boolean(),
normalizedUrl: Type.String(),
sharingUrl: Type.String(),
summaryAgeSeconds: Type.Number(),
thesis: Type.Array(SummarizeThesis)
})
])

export type SummarizeThesis = Static<typeof SummarizeThesis>
export const SummarizeThesis = Type.Object({
id: Type.Number(),
content: Type.String()
})

export type SummarizeThesisWithLink = Static<typeof SummarizeThesisWithLink>
export const SummarizeThesisWithLink = Type.Intersect([
SummarizeThesis,
Type.Object({
link: Type.String()
})
])

export type SummarizeThesisWithTime = Static<typeof SummarizeThesisWithTime>
export const SummarizeThesisWithTime = Type.Intersect([
SummarizeThesis,
Type.Object({
startTime: Type.Number()
})
])

export type SummarizeChapter<T extends TSchema> = Static<ReturnType<typeof SummarizeChapter<T>>>
export const SummarizeChapter = <T extends TSchema>(T: T) => Type.Object({
id: Type.Number(),
content: Type.String(),
theses: Type.Array(T)
})

export type VideoSummarizeChapter = Static<typeof VideoSummarizeChapter>
export const VideoSummarizeChapter = Type.Intersect([
SummarizeChapter(SummarizeThesis),
Type.Object({
startTime: Type.Number()
})
])

export type VideoSummarizeExtraOpts = Static<typeof VideoSummarizeExtraOpts>
export const VideoSummarizeExtraOpts = Type.Intersect([
SharedSummarizeExtraOpts,
Type.Object({
videoTitle: Type.Optional(Type.String())
})
])

export type VideoSummarizeOpts = Static<typeof VideoSummarizeOpts>
export const VideoSummarizeOpts = Type.Intersect([
SharedSummarizeOpts(VideoSummarizeExtraOpts),
Type.Object({
url: Type.String(),
language: Type.String()
})
])

export type VideoSummarizeResponse = Static<typeof VideoSummarizeResponse>
export const VideoSummarizeResponse = Type.Intersect([
MinimalSummarizeResponse(Type.Literal("video")),
Type.Object({
chapters: Type.Array(VideoSummarizeChapter),
unknown0: Type.Optional(Type.String()),
haveChapters: Type.Literal(true),
thesis: Type.Array(SummarizeThesisWithTime)
})
])

export type ArticleSummarizeExtraOpts = Static<typeof ArticleSummarizeExtraOpts>
export const ArticleSummarizeExtraOpts = SharedSummarizeExtraOpts

export type ArticleSummarizeOpts = Static<typeof ArticleSummarizeOpts>
export const ArticleSummarizeOpts = Type.Intersect([
SharedSummarizeOpts(ArticleSummarizeExtraOpts),
Type.Object({
url: Type.String()
})
])

export type ArticleSummarizeResponse = Static<typeof ArticleSummarizeResponse>
export const ArticleSummarizeResponse = SummarizeResponse(Type.Literal("article"))

export type TextSummarizeExtraOpts = Static<typeof TextSummarizeExtraOpts>
export const TextSummarizeExtraOpts = SharedSummarizeExtraOpts

export type TextSummarizeOpts = Static<typeof TextSummarizeOpts>
export const TextSummarizeOpts = Type.Intersect([
SharedSummarizeOpts(ArticleSummarizeExtraOpts),
Type.Object({
text: Type.String()
})
])

export type TextSummarizeResponse = Static<typeof TextSummarizeResponse>
export const TextSummarizeResponse = SummarizeResponse(Type.Literal("text"))

export type GetSharingContentOpts = Static<typeof GetSharingContentOpts>
export const GetSharingContentOpts = Type.Object({
token: Type.String(),
headers: Type.Optional(RequestHeaders)
})

export type SharingVideoSummarizeResponse = Static<typeof SharingVideoSummarizeResponse>
export const SharingVideoSummarizeResponse = Type.Intersect([
MinimalSummarizeResponse(Type.Literal("video")),
Type.Object({
contentId: Type.String(),
keypoints: Type.Array(VideoSummarizeChapter),
normalizedUrl: Type.String(),
sharingUrl: Type.String(),
summaryAgeSeconds: Type.Number()
})
])