tcbot-engine/src/main/java/org/apache/ignite/tcbot/engine/conf/TcBotJsonConfig.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */

package org.apache.ignite.tcbot.engine.conf;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import javax.annotation.Nullable;

/**
 * TC Bot main JSON config file, Historically.
 * Config file for tracked branches.
 */
public class TcBotJsonConfig implements ITrackedBranchesConfig {
    /** Branches. */
    private List<BranchTracked> branches = new ArrayList<>();

    /** Primary server ID. */
    @Nullable private String primaryServerCode;

    /** Flaky rate to consider test as a flaky test. */
    @Nullable private Integer flakyRate;

    /** Сonfidence (used with flaky tests). */
    @Nullable private Double confidence;

    /** Additional list Servers to be used for validation of PRs, but not for tracking any branches. */
    private List<TcServerConfig> tcServers = new ArrayList<>();

    /** JIRA config to be used . */
    private List<JiraServerConfig> jiraServers = new ArrayList<>();

    /** JIRA config to be used . */
    private List<GitHubConfig> gitHubConfigs = new ArrayList<>();

    /** Notifications settings & tokens. */
    private NotificationsConfig notifications = new NotificationsConfig();

    @Override
    public Stream<ITrackedBranch> branchesStream() {
        return branches.stream().map(t->t);
    }

    /**
     *
     */
    public Set<String> getServerIds() {
        Stream<String> srvsInTracked = branchesStream()
            .flatMap(ITrackedBranch::chainsStream)
            .map(ITrackedChain::serverCode);

        return Stream.concat(srvsInTracked,
            tcServers.stream().map(TcServerConfig::getCode))
            .collect(Collectors.toSet());
    }

    public List<BranchTracked> getBranches() {
        return Collections.unmodifiableList(branches);
    }

    public void addBranch(BranchTracked branch) {
        branches.add(branch);
    }

    /**
     * @return Primary server code.
     */
    @Nullable public String primaryServerCode() {
        return primaryServerCode;
    }

    /**
     * @return Flaky rate to consider test as a flaky test.
     */
    @Nullable public Integer flakyRate() {
        return flakyRate;
    }

    /**
     * @return Сonfidence.
     */
    @Nullable public Double confidence() {
        return confidence;
    }

    public Optional<TcServerConfig> getTcConfig(String code) {
        return tcServers.stream().filter(s -> Objects.equals(code, s.getCode())).findAny();
    }

    public Optional<JiraServerConfig> getJiraConfig(String code) {
        return jiraServers.stream().filter(s -> Objects.equals(code, s.getCode())).findAny();
    }

    public Optional<GitHubConfig> getGitHubConfig(String code) {
        return gitHubConfigs.stream().filter(s -> Objects.equals(code, s.getCode())).findAny();
    }

    @Nullable
    public NotificationsConfig notifications() {
        return notifications;
    }
}