/*
 * 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
 *
 *  https://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.
 *
 * Other licenses:
 * -----------------------------------------------------------------------------
 * Commercial licenses for this work are available. These replace the above
 * Apache-2.0 license and offer limited warranties, support, maintenance, and
 * commercial database integrations.
 *
 * For more information, please visit: https://www.jooq.org/legal/licensing
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 *
 */
package org.jooq.impl;

// ...
import static org.jooq.SQLDialect.TRINO;
import static org.jooq.impl.DSL.inline;
import static org.jooq.impl.Names.N_GROUP_CONCAT;

import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Set;

import org.jooq.AggregateFunction;
import org.jooq.Context;
import org.jooq.Field;
import org.jooq.GroupConcatOrderByStep;
import org.jooq.OrderField;
import org.jooq.SQLDialect;
import org.jooq.impl.QOM.UNotYetImplemented;

/**
 * @author Lukas Eder
 */
final class GroupConcat
extends
    AbstractAggregateFunction<String>
implements
    GroupConcatOrderByStep,
    UNotYetImplemented {

    final Set<SQLDialect>       REQUIRE_WITHIN_GROUP = SQLDialect.supportedBy(TRINO);

    GroupConcat(Field<?> field) {
        this(field, false);
    }

    GroupConcat(Field<?> field, boolean distinct) {
        super(distinct, N_GROUP_CONCAT, SQLDataType.VARCHAR, field);
    }

    @Override
    public final void accept(Context<?> ctx) {
        ListAgg result;

        if (getArgument(1) == null)
            result = new ListAgg(distinct, getArgument(0), ListAgg.DEFAULT_SEPARATOR);
        else
            result = new ListAgg(distinct, getArgument(0), (Field) getArgument(1));

        if (!Tools.isEmpty(withinGroupOrderBy))
            result.withinGroupOrderBy(withinGroupOrderBy);

        // [#3045] [#11485] Dialects with mandatory WITHIN GROUP clause
        else if (REQUIRE_WITHIN_GROUP.contains(ctx.dialect()))
            result.withinGroupOrderBy(withinGroupOrderBy != null ? withinGroupOrderBy : Collections.emptyList());

        ctx.visit(fo(result));
    }

    @Override
    public final AggregateFunction<String> separator(String s) {
        return separator(inline(s));
    }

    @Override
    public final AggregateFunction<String> separator(Field<String> s) {
        if (arguments.size() < 2)
            arguments.add(s);
        else
            arguments.set(1, s);

        return this;
    }

    @Override
    public final GroupConcat orderBy(OrderField<?>... fields) {
        return orderBy(Arrays.asList(fields));
    }

    @Override
    public final GroupConcat orderBy(Collection<? extends OrderField<?>> fields) {
        withinGroupOrderBy(fields);
        return this;
    }
}
