import {ApiRequest, ApiResponse} from "@gongt/ts-stl-library/request/protocol";
import {ErrorResponse} from "@gongt/ts-stl-library/request/request-error";
import {Context, Middleware, Request, Response} from "koa";
import {Middleware as ComposeMiddleware} from "koa-compose";

export function responseJson<Res extends ApiResponse>(response: Response, data: Res) {
	response.body = {
		status: 0,
		message: '',
		...data as object,
	};
	response.type = 'json';
}

export interface TypedRequest<Req, Res, Ses, Cok> extends Request {
	ctx: TypedContext<Req, Res, Ses, Cok>;
	response: TypedResponse<Req, Res, Ses, Cok>;
	body: Req;
}

export interface TypedResponse<Req, Res, Ses, Cok> extends Response {
	ctx: TypedContext<Req, Res, Ses, Cok>;
	request: TypedRequest<Req, Res, Ses, Cok>;
	body: Res;
}

export interface TypedContext<Req, Res, Ses, Cok> extends Context {
	request: TypedRequest<Req, Res, Ses, Cok>;
	response: TypedResponse<Req, Res, Ses, Cok>;
	session: Ses;
	cookie: Cok;
}

export type AnyTypedContext = TypedContext<any, any, any, any>;

export type TypedMiddleware
	<RequestType=any, ResponseType=any, SessionType=any, CookieType=any>
	= ComposeMiddleware<TypedContext<RequestType, ResponseType, SessionType, CookieType>>;

export function tsHint<RequestType extends ApiRequest, ResponseType extends ApiResponse>(middleware: TypedMiddleware<RequestType, ResponseType>): Middleware {
	return middleware;
}