"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
    var ownKeys = function(o) {
        ownKeys = Object.getOwnPropertyNames || function (o) {
            var ar = [];
            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
            return ar;
        };
        return ownKeys(o);
    };
    return function (mod) {
        if (mod && mod.__esModule) return mod;
        var result = {};
        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
        __setModuleDefault(result, mod);
        return result;
    };
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.register = void 0;
const path = __importStar(require("path"));
const write_1 = require("@plastichub/fs/write");
const read_1 = require("@plastichub/fs/read");
const exists_1 = require("@plastichub/fs/exists");
const osr_commons_1 = require("@plastichub/osr-commons");
const osr_commons_2 = require("@plastichub/osr-commons");
const __1 = require("../");
const _cli_1 = require("../_cli");
const query_1 = require("../lib/perplexity/query");
const args_1 = require("../lib/args");
const filters_1 = require("../lib/filters");
const defaultOptions = (yargs) => {
    return yargs.option('debug', {
        default: false,
        describe: 'debug messages',
        type: 'boolean'
    }).option('append', {
        default: false,
        describe: 'append to file instead of overwriting',
        type: 'boolean'
    }).option('showPrompt', {
        default: false,
        describe: 'insert prompt in output',
        type: 'boolean'
    }).option('cache', {
        default: true,
        describe: 'Enable cache',
        type: 'boolean'
    }).option('query', {
        description: 'query',
        default: "list all towns of barcelona, as typescript enum, with gps coords, as strings"
    }).option('dst', {
        description: 'dst output path, supports XLS|CSV|HTML',
        _default: './tests/chatgpt/last.ts'
    }).option('env_key', {
        default: 'OSR-CONFIG',
        describe: 'Environment key to the config path'
    }).option('api_key', {
        describe: 'OpenAI key'
    }).option('model', {
        describe: 'OpenAI Model',
        default: "gpt-4"
    }).option('system', {
        describe: 'Path to system instructions'
    }).option('source', {
        describe: 'Path to a source file'
    }).option('response_format', {
        describe: 'Format of the response (for transcribe)',
        default: "text"
    }).option('filters', {
        describe: 'An array of filters to apply to the response: JSON',
        default: ""
    });
};
let options = (yargs) => defaultOptions(yargs);
const register = (cli) => {
    return cli.command('perplexity <verb>', 'Search Perplexity AI', options, async (argv) => {
        (0, _cli_1.defaults)();
        if (argv.help) {
            return;
        }
        const args = argv;
        const verb = args.verb || 'prompt';
        const config = (0, osr_commons_1.CONFIG_DEFAULT)(args.env_key);
        if (!config) {
            __1.logger.warn('No config found!');
            return;
        }
        const key = argv.perplexity_key || process.env.OPENAI_API_KEY || config.perplexity.key;
        if (!key) {
            __1.logger.warn('No Perplexity  key found in config!');
            return;
        }
        const opts = {
            query: argv.query,
            ...argv,
            dst: args.dst ? path.resolve((0, osr_commons_2.resolve)(args.dst)) : '',
            source: argv.source ? path.resolve((0, osr_commons_2.resolve)(args.source)) : undefined,
            system: argv.system ? path.resolve((0, osr_commons_2.resolve)(args.system)) : undefined,
            api_key: key
        };
        const filters = opts.filters.split(',');
        opts.filters = [];
        filters.forEach((f) => {
            if (filters_1.Filters[f]) {
                (opts.filters).push({
                    fn: filters_1.Filters[f],
                    name: f
                });
            }
        });
        if (opts.cache === true && (0, exists_1.sync)(opts.dst)) {
            __1.logger.debug('Output file already exists, skipping');
            return;
        }
        if (!opts.api_key) {
            __1.logger.error('No OpenAI key found in config or options!');
            return;
        }
        if (!opts.query) {
            __1.logger.error('No query specified');
            return;
        }
        opts.debug && __1.logger.debug('Options', opts);
        return new Promise(async (resolve, reject) => {
            switch (verb) {
                case 'prompt': {
                    if (opts.source && !(0, exists_1.sync)(opts.source)) {
                        __1.logger.error('Source specified but file not found', opts.source);
                        return;
                    }
                    const src = opts.source && (0, exists_1.sync)(opts.source) ? (0, read_1.sync)(opts.source, 'string') : '';
                    const stdin = await (0, args_1.readPipedInput)() || src || '';
                    let q = stdin ? `${opts.query} : "${stdin}"` : opts.query;
                    const queryPath = path.resolve((0, osr_commons_2.resolve)(opts.query));
                    if ((0, exists_1.sync)(queryPath)) {
                        q = (0, read_1.sync)(queryPath);
                    }
                    q = `${q} : ${src}`;
                    opts.debug && __1.logger.debug('Query: ' + q);
                    let ret = await (0, query_1.query)(q, opts.api_key, opts.dst, {
                        ...opts,
                        stdin: stdin
                    });
                    if (opts.filters) {
                        opts.filters.forEach((f) => {
                            try {
                                const n = f.fn(ret);
                                if (n) {
                                    ret = n;
                                }
                            }
                            catch (e) {
                                __1.logger.error(`Error applying filter: ${f.name}`, e);
                            }
                        });
                    }
                    if (opts.dst) {
                        let header = `${argv.showPrompt ? `// ${q}\n` : ''}`;
                        let content = `${argv.append ? (0, read_1.sync)((opts.dst) || '') : ''}\n${header}${ret}`;
                        (0, write_1.sync)(opts.dst, content);
                    }
                    ret && process.stdout.write(ret);
                    break;
                }
            }
            resolve(1);
            process.exit(0);
        });
    });
};
exports.register = register;
//# sourceMappingURL=perplexity.js.map