import { AbstractPredictionsProvider } from "../types/Providers";
import {
    AmazonAIConvertPredictionsProvider,
    AmazonAIInterpretPredictionsProvider,
    AmazonAIIdentifyPredictionsProvider
} from ".";
import { TranslateTextInput, TextToSpeechInput, SpeechToTextInput, PredictionsOptions, IdentifyTextInput, 
    IdentifyTextOutput, IdentifyLabelsInput, IdentifyLabelsOutput, IdentifyEntitiesInput, IdentifyEntitiesOutput, 
    isIdentifyTextInput, isIdentifyLabelsInput, isIdentifyEntitiesInput, TranslateTextOutput,
    TextToSpeechOutput, isTranslateTextInput, SpeechToTextOutput, isTextToSpeechInput, isSpeechToTextInput,
    InterpretTextInput, InterpretTextOutput, isInterpretTextInput
 } from "../types";

export default class AmazonAIPredictionsProvider extends AbstractPredictionsProvider {

    private convertProvider: AmazonAIConvertPredictionsProvider;
    private identifyProvider: AmazonAIIdentifyPredictionsProvider;
    private interpretProvider: AmazonAIInterpretPredictionsProvider;

    constructor() {
        super();
        this.convertProvider = new AmazonAIConvertPredictionsProvider();
        this.identifyProvider = new AmazonAIIdentifyPredictionsProvider();
        this.interpretProvider = new AmazonAIInterpretPredictionsProvider();
    }

    getCategory(): string {
        return "Predictions";
    }

    getProviderName(): string {
        return "AmazonAIPredictionsProvider";
    }

    configure(config: PredictionsOptions) {
        this.convertProvider.configure(config.convert);
        this.identifyProvider.configure(config.identify);
        this.interpretProvider.configure(config.interpret);
        return config;
    }

    interpret(input: InterpretTextInput): Promise<InterpretTextOutput> {
        return this.interpretProvider.interpret(input);
    }

    convert(input: TranslateTextInput | TextToSpeechInput | SpeechToTextInput)
        : Promise<TextToSpeechOutput | TranslateTextOutput | SpeechToTextOutput> {
        return this.convertProvider.convert(input);
    }

    identify(input: IdentifyTextInput | IdentifyLabelsInput | IdentifyEntitiesInput)
        : Promise<IdentifyTextOutput | IdentifyLabelsOutput | IdentifyEntitiesOutput> {
        return this.identifyProvider.identify(input);
    }

}