import * as Promise from 'bluebird';

import {
    HudAiCreateAttributes,
    HudAiListAttributes,
    HudAiUpdateAttributes,
    Resource
} from '../utils/Resource';
import { RequestManager } from '../RequestManager';
import { TextCorpus } from '../entities';

export interface TextCorpusListAttributes extends HudAiListAttributes {
    type?: string;
    userId?: string;
}

export interface TextCorpusCreateAttributes extends HudAiCreateAttributes {
    type: string;
    body: string;
    userId: string;
}

export interface TextCorpusUpdateAttributes extends HudAiUpdateAttributes {
    type?: string;
    body?: string;
    userId?: string;
}

export class TextCorpusResource extends Resource<
    TextCorpus,
    TextCorpusListAttributes,
    TextCorpusCreateAttributes,
    TextCorpusUpdateAttributes
> {
    constructor(requestManager: RequestManager) {
        super('/text-corpora', requestManager);
    }

    public list(listArgs: TextCorpusListAttributes): Promise<{ count: number, rows: TextCorpus[] }> {
        return this._list(listArgs);
    }

    public create(createArgs: TextCorpusCreateAttributes): Promise<TextCorpus> {
        return this._create(createArgs);
    }

    public get(id: string | number): Promise<TextCorpus> {
        return this._get(id);
    }

    public update(id: string | number, updateArgs: TextCorpusUpdateAttributes): Promise<TextCorpus> {
        return this._update(id, updateArgs);
    }

    public del(id: string | number): Promise<void> {
        return this.destroy(id);
    }

    public destroy(id: string | number): Promise<void> {
        return this._destroy(id);
    }
}