/**
 * Copyright (c) 2020-present, Goldman Sachs
 *
 * 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.
 */
import { PRIMITIVE_TYPE, Enumeration, PrimitiveType, } from '@finos/legend-graph';
import { QueryBuilderAggregateOperator } from '../QueryBuilderAggregateOperator.js';
import { QueryBuilderSimpleProjectionColumnState, } from '../../projection/QueryBuilderProjectionColumnState.js';
import { buildAggregateColumnState, buildAggregateExpression, } from './QueryBuilderAggregateOperatorValueSpecificationBuilder.js';
import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graph/QueryBuilderMetaModelConst.js';
import { hashArray } from '@finos/legend-shared';
import { QUERY_BUILDER_STATE_HASH_STRUCTURE } from '../../../../QueryBuilderStateHashUtils.js';
export class QueryBuilderAggregateOperator_Count extends QueryBuilderAggregateOperator {
    getLabel(projectionColumnState) {
        return 'count';
    }
    isCompatibleWithColumn(projectionColumnState) {
        if (projectionColumnState instanceof QueryBuilderSimpleProjectionColumnState) {
            const propertyType = projectionColumnState.propertyExpressionState.propertyExpression.func
                .value.genericType.value.rawType;
            return ([
                PRIMITIVE_TYPE.STRING,
                PRIMITIVE_TYPE.BOOLEAN,
                PRIMITIVE_TYPE.NUMBER,
                PRIMITIVE_TYPE.INTEGER,
                PRIMITIVE_TYPE.DECIMAL,
                PRIMITIVE_TYPE.FLOAT,
                PRIMITIVE_TYPE.DATE,
                PRIMITIVE_TYPE.STRICTDATE,
                PRIMITIVE_TYPE.DATETIME,
            ].includes(propertyType.path) || propertyType instanceof Enumeration);
        }
        return true;
    }
    buildAggregateExpression(propertyExpression, variableName, graph) {
        return buildAggregateExpression(QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT, variableName, graph);
    }
    buildAggregateColumnState(expression, lambdaParam, projectionColumnState) {
        return buildAggregateColumnState(projectionColumnState, lambdaParam, expression, QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT, this);
    }
    getReturnType(aggregateColumnState) {
        return PrimitiveType.INTEGER;
    }
    get hashCode() {
        return hashArray([
            QUERY_BUILDER_STATE_HASH_STRUCTURE.AGGREGATE_OPERATOR_COUNT,
        ]);
    }
}
//# sourceMappingURL=QueryBuilderAggregateOperator_Count.js.map