// -*- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// This file is part of Genie
//
// Copyright 2019-2020 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: Giovanni Campagna <gcampagn@cs.stanford.edu>

import assert from 'assert';
import { Ast, Type } from 'thingtalk';
import * as C from '../ast_manip';
import ThingpediaLoader from '../load-thingpedia';
import type * as $Genie from "genie-toolkit";
export default async function($runtime : typeof $Genie.SentenceGeneratorRuntime, $ttUtils : typeof $Genie.ThingTalkUtils, $options : $Genie.SentenceGeneratorTypes.GrammarOptions, $locale : $Genie.I18n.LanguagePack, $grammar : $Genie.SentenceGenerator<any, any>, $loader : ThingpediaLoader) : Promise<void> {
   $grammar.declareSymbol("projection_Number");
   if ($options.flags.aggregation) {
       $grammar.addRule("projection_Number", ["the", new $runtime.Choice(["number", "total number", "count"]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["is_list" as ("is_list" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), (true) as ReturnType<typeof C.expressionKeyFn>["is_list"]])], (($0 : undefined, $1 : undefined, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        assert(t.schema!.is_list);
        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT)
            .addArguments([new Ast.ArgumentDef(null, Ast.ArgDirection.OUT, 'count', Type.Number)]);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, '*', 'count', newSchema);
        return new Ast.ProjectionExpression(null, agg, ['count'], [], [], newSchema);
    }), C.expressionKeyFn, {  });
   }
   $grammar.declareSymbol("projection_Any");
   if ($options.flags.aggregation) {
       $grammar.addRule("projection_Any", [new $runtime.NonTerminal("thingpedia_complete_query", undefined)], ((q : Ast.Expression) : (Ast.Expression)|null => {
        if (!(q instanceof Ast.AggregationExpression))
            return null;
        return q;
    }), C.expressionKeyFn, {  });
       if (!$options.flags.schema_org) {
           $grammar.addRule("projection_Any", ["the", "total", new $runtime.NonTerminal("out_param_Any", ["is_numeric" as ("is_numeric" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void), (true) as ReturnType<typeof C.paramKeyFn>["is_numeric"]]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["functionName" as ("functionName" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), 2, "functionName" as ("functionName" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void)])], (($0 : undefined, $1 : undefined, p : C.ParamSlot, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        if (!t.schema!.out[p.name] || !t.schema!.out[p.name].isNumeric())
            return null;
        if (!t.schema!.is_list)
            return null;

        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT || arg.name === p.name);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, p.name, 'sum', newSchema);
        return new Ast.ProjectionExpression(null, agg, [p.name], [], [], newSchema);
    }), C.expressionKeyFn, {  });
       }
       if (!$options.flags.turking) {
           $grammar.addRule("projection_Any", ["the", new $runtime.Choice(["average", "mean"]), new $runtime.NonTerminal("out_param_Any", ["is_numeric" as ("is_numeric" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void), (true) as ReturnType<typeof C.paramKeyFn>["is_numeric"]]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["functionName" as ("functionName" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), 2, "functionName" as ("functionName" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void)])], (($0 : undefined, $1 : undefined, p : C.ParamSlot, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        if (!t.schema!.out[p.name] || !t.schema!.out[p.name].isNumeric())
            return null;
        if (!t.schema!.is_list)
            return null;

        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT || arg.name === p.name);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, p.name, 'avg', newSchema);
        return new Ast.ProjectionExpression(null, agg, [p.name], [], [], newSchema);
    }), C.expressionKeyFn, {  });
       }
       if ($options.flags.turking) {
           $grammar.addRule("projection_Any", ["the", "average", new $runtime.NonTerminal("out_param_Any", ["is_numeric" as ("is_numeric" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void), (true) as ReturnType<typeof C.paramKeyFn>["is_numeric"]]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["functionName" as ("functionName" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), 2, "functionName" as ("functionName" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void)])], (($0 : undefined, $1 : undefined, p : C.ParamSlot, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        if (!t.schema!.out[p.name] || !t.schema!.out[p.name].isNumeric())
            return null;
        if (!t.schema!.is_list)
            return null;

        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT || arg.name === p.name);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, p.name, 'avg', newSchema);
        return new Ast.ProjectionExpression(null, agg, [p.name], [], [], newSchema);
    }), C.expressionKeyFn, {  });
       }
       $grammar.addRule("projection_Any", ["the", new $runtime.Choice(["minimum", "lowest"]), new $runtime.NonTerminal("out_param_Any", ["is_numeric" as ("is_numeric" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void), (true) as ReturnType<typeof C.paramKeyFn>["is_numeric"]]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["functionName" as ("functionName" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), 2, "functionName" as ("functionName" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void)])], (($0 : undefined, $1 : undefined, p : C.ParamSlot, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        if (!t.schema!.out[p.name] || !t.schema!.out[p.name].isNumeric())
            return null;
        if (!t.schema!.is_list)
            return null;

        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT || arg.name === p.name);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, p.name, 'min', newSchema);
        return new Ast.ProjectionExpression(null, agg, [p.name], [], [], newSchema);
    }), C.expressionKeyFn, {  });
       $grammar.addRule("projection_Any", ["the", new $runtime.Choice(["maximum", "highest"]), new $runtime.NonTerminal("out_param_Any", ["is_numeric" as ("is_numeric" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void), (true) as ReturnType<typeof C.paramKeyFn>["is_numeric"]]), "of", new $runtime.Choice(["", "the", "this", "that"]), new $runtime.NonTerminal("complete_table", ["functionName" as ("functionName" extends keyof ReturnType<typeof C.expressionKeyFn> ? string : void), 2, "functionName" as ("functionName" extends keyof ReturnType<typeof C.paramKeyFn> ? string : void)])], (($0 : undefined, $1 : undefined, p : C.ParamSlot, $2 : undefined, $3 : undefined, t : Ast.Expression) : (Ast.Expression)|null => {
        if (!t.schema!.out[p.name] || !t.schema!.out[p.name].isNumeric())
            return null;
        if (!t.schema!.is_list)
            return null;

        const newSchema = t.schema!.filterArguments((arg) => arg.direction !== Ast.ArgDirection.OUT || arg.name === p.name);
        newSchema.is_list = false;
        const agg = new Ast.AggregationExpression(null, t, p.name, 'max', newSchema);
        return new Ast.ProjectionExpression(null, agg, [p.name], [], [], newSchema);
    }), C.expressionKeyFn, {  });
   }
};
