react-native-api-ai/index.ios.js

A React-Native Bridge for the Google API AI SDK

import { NativeModules, NativeAppEventEmitter } from 'react-native';
import {ApiAiClient} from 'api-ai-javascript';
import ResetContextsRequest from './js/ResetContextsRequest';

var SpeechToText = NativeModules.RNSpeechToTextIos;

class ApiAi {


    setConfiguration(clientAccessToken: String, languageTag: String) {
        this.language = languageTag;
        this.client = new ApiAiClient({accessToken: clientAccessToken, apiLang: languageTag});
    }

    onListeningStarted(callback: ()=>{}) {

    }

    onListeningCanceled(callback: ()=>{}) {

    }

    onListeningFinished(callback: ()=>{}) {

    }

    startListening(onResult: ()=>{}, onError: ()=>{}) {

        this.subscription = NativeAppEventEmitter.addListener(
            'SpeechToText',
            (result) => {

                console.log(result);

                if (result.error) {
                    onError(result.error);
                } else {
                    if (result.isFinal) {
                        this.requestQuery(result.bestTranscription.formattedString, onResult, onError);
                    }

                }

            }
        );

        SpeechToText.startRecognition(this.language);
    }

    finishListening() {
        SpeechToText.finishRecognition();
    }

    setContexts(contexts) {
        this.contexts = contexts;
    }

    setPermanentContexts(contexts) {
        // set lifespan to 1 if it's not set
        contexts.forEach((c,i,a)=>{
            if (!c.lifespan) {
                a[i] = {...c, lifespan: 1};
            }
        });

        this.permanentContexts = contexts;
    }

    resetContexts(onResult: ()=>{}, onError: ()=>{}) {
        let request = new ResetContextsRequest(this.client.getAccessToken(), this.client.getSessionId(), null);
        request.perform().then(res=>onResult(res)).catch(err=>onError(err));
    };

    setEntities(entities) {
        this.entities = entities;
    }

    requestQuery(query: String, onResult: ()=>{}, onError: ()=>{}) {
        if (this.contexts || this.permanentContexts || this.entities) {
            this.client.textRequest(query, {
                contexts: this.mergeContexts(this.contexts, this.permanentContexts),
                entities: this.entities}
            ).then(res=>onResult(res)).catch(err=>onError(err));
            this.contexts = null;
            this.entities = null;
        } else {
            this.client.textRequest(query).then(res=>onResult(res)).catch(err=>onError(err));
        }
    }

    requestEvent(eventName, eventData = {}, onResult: ()=>{}, onError: ()=>{}) {
        this.client.eventRequest(eventName, eventData, {}).then(res=>onResult(res)).catch(err=>onError(err));
    }

    onAudioLevel(callback) {

    }

    mergeContexts(context1, context2) {
        if (!context1) {
            return context2;
        } else if (!context2) {
            return context1;
        } else {
            return [...context1, ...context2];
        }
    }

    LANG_CHINESE_CHINA = "zh-CN";
    LANG_CHINESE_HONGKONG = "zh-HK";
    LANG_CHINESE_TAIWAN = "zh-TW";
    LANG_DUTCH = "nl";
    LANG_ENGLISH = "en";
    LANG_ENGLISH_GB = "en-GB";
    LANG_ENGLISH_US = "en-US";
    LANG_FRENCH = "fr";
    LANG_GERMAN = "de";
    LANG_ITALIAN = "it";
    LANG_JAPANESE = "ja";
    LANG_KOREAN = "ko";
    LANG_PORTUGUESE = "pt";
    LANG_PORTUGUESE_BRAZIL = "pt-BR";
    LANG_RUSSIAN = "ru";
    LANG_SPANISH = "es";
    LANG_UKRAINIAN = "uk";
}

export default new ApiAi();