// -*- mode: typescript; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// This file is part of Genie
//
// Copyright 2019-2021 The Board of Trustees of the Leland Stanford Junior University
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Author: Silei Xu <silei@cs.stanford.edu>

import * as argparse from 'argparse';
import * as Tp from 'thingpedia';
import * as fs from 'fs';
import FormData from 'form-data';
import csvstringify from 'csv-stringify';

import { getConfig, DEFAULT_THINGPEDIA_URL } from './lib/argutils';

export function initArgparse(subparsers : argparse.SubParser) {
    const parser = subparsers.add_parser('upload-entity-values', {
        add_help: true,
        description: "Upload a new entity type to Thingpedia."
    });
    parser.add_argument('-l', '--locale', {
        required: false,
        default: 'en-US',
        help: `BGP 47 locale tag of the natural language (defaults to 'en-US', American English)`
    });
    parser.add_argument('--thingpedia-url', {
        required: false,
        help: `base URL of Thingpedia server to contact; defaults to '${DEFAULT_THINGPEDIA_URL}'`
    });
    parser.add_argument('--access-token', {
        required: false,
        help: `OAuth access token to use when contacting Thingpedia`
    });
    parser.add_argument('--csv', {
        required: false,
        help: "The .csv file of the entity values to upload."
    });
    parser.add_argument('--json', {
        required: false,
        help: "The .json file of the entity values to upload"
    });
    parser.add_argument('--entity-id', {
        required: true,
        help: "The id (type) of the string dataset."
    });
    parser.add_argument('--entity-name', {
        required: true,
        help: "The name of the string dataset."
    });
    parser.add_argument('--no-ner-support', {
        action: 'store_true',
        help: 'If this entity is an opaque identifier that cannot be used from natural language.',
        default: false
    });
}

function createUpload(args : any) {
    const fd = new FormData();
    if (args.no_ner_support !== '1') {
        if (args.csv) {
            fd.append('upload', fs.createReadStream(args.csv), {
                filename: 'entity.csv',
                contentType: 'text/csv;charset=utf8'
            });
        } else if (args.json) {
            const json = JSON.parse(fs.readFileSync(args.json, 'utf8'));
            const csv = json.data.map((row : { value : string, name : string }) => [row.value, row.name]);
            const string = csvstringify(csv, { delimiter: ',' });
            fd.append('upload', string, {
                filename: 'entity.csv',
                contentType: 'text/csv;charset=utf8'
            });
        } else {
            throw new Error(`Either a json file or a csv file is needed.`);
        }
    }
    for (const key of ['entity_id', 'entity_name', 'no_ner_support'])
        fd.append(key, args[key]);
    return fd;
}

export async function execute(args : any) {
    if (!args.thingpedia_url)
        args.thingpedia_url = await getConfig('thingpedia.url', process.env.THINGPEDIA_URL || DEFAULT_THINGPEDIA_URL);
    if (!args.access_token)
        args.access_token = await getConfig('thingpedia.access-token', process.env.THINGPEDIA_ACCESS_TOKEN || null);

    if (!args.access_token)
        throw new Error(`You must pass a valid OAuth access token to talk to Thingpedia`);

    args.no_ner_support = args.no_ner_support ? '1' : '';
    const fd = createUpload(args);
    await Tp.Helpers.Http.postStream(args.thingpedia_url + '/api/v3/entities/create', fd, {
        dataContentType:  'multipart/form-data; boundary=' + fd.getBoundary(),
        auth: 'Bearer ' + args.access_token
    });

    console.log('Success!');
}