'use strict';

import * as db from 'db/index';
import toObjectId from 'misc/toobjectid';
import { ObjectId } from 'mongodb';
import { CredentialType } from 'struct/credential';
import { InsertResult } from 'struct/db';

export type TokenExchangeMethod = 'post' | 'basic';

export type Credential = {
	_id?: ObjectId;
	orgId: ObjectId;
	teamId: ObjectId;
	type: CredentialType;
	credentials: {
		key?: string;
		endpointURL?: string;
		clientId?: string;
		clientSecret?: string;
		authURL?: string;
		tokenURL?: string;
		scope?: string;
		tokenExchangeMethod?: TokenExchangeMethod;
	};
    name: string;
    createdDate: Date;
}

export function CredentialCollection(): any {
	return db.db().collection('credentials');
}

export function getCredentialById(teamId: db.IdOrStr, credentialId: db.IdOrStr): Promise<Credential> {
	return CredentialCollection().findOne({
		_id: toObjectId(credentialId),
		teamId: toObjectId(teamId),
	}, {
		projection: {
			'credentials.key': 0,
		}
	});
}

export function getCredentialsByTeam(teamId: db.IdOrStr): Promise<Credential> {
	return CredentialCollection().find({
		teamId: toObjectId(teamId),
	}, {
		projection: {
			'credentials.key': 0,
		}
	}).toArray();
}

export function getCredentialsById(teamId: db.IdOrStr, credentialIds: db.IdOrStr[]): Promise<Credential[]> {
	return CredentialCollection().find({
		_id: {
			$in: credentialIds.map(toObjectId),
		},
		teamId: toObjectId(teamId),
	}, {
		projection: {
			'credentials.key': 0,
		}
	}).toArray();
}

export async function addCredential(credential: Credential): Promise<InsertResult> {
	return CredentialCollection().insertOne(credential);
}

export function deleteCredentialById(teamId: db.IdOrStr, credentialId: db.IdOrStr): Promise<any> {
	return CredentialCollection().deleteOne({
		_id: toObjectId(credentialId),
		teamId: toObjectId(teamId),
	});
}