com.thoughtworks.go.server.dashboard.AbstractDashboardGroup.java Source code

Introduction

Here is the source code for com.thoughtworks.go.server.dashboard.AbstractDashboardGroup.java

Source

/*
 * Copyright 2019 ThoughtWorks, Inc.
 *
 * 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.
 */

package com.thoughtworks.go.server.dashboard;

import com.thoughtworks.go.server.domain.Username;
import org.apache.commons.codec.binary.Hex;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.io.output.NullOutputStream;

import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.UncheckedIOException;
import java.security.DigestOutputStream;
import java.security.MessageDigest;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import static java.util.Comparator.comparing;
import static java.util.stream.Collectors.toList;

public abstract class AbstractDashboardGroup implements DashboardGroup {
    private String name;
    private boolean hasDefinedPipelines;
    private Map<String, GoDashboardPipeline> pipelines = new LinkedHashMap<>();

    AbstractDashboardGroup(String name, boolean hasDefinedPipelines) {
        this.name = name;
        this.hasDefinedPipelines = hasDefinedPipelines;
    }

    @Override
    public String name() {
        return name;
    }

    @Override
    public List<String> pipelines() {
        return pipelines.values().stream().sorted(comparing(GoDashboardPipeline::getdisplayOrderWeight))
                .map(pipeline -> pipeline.name().toString()).collect(toList());
    }

    public abstract boolean canAdminister(Username username);

    public abstract String etag();

    public boolean hasDefinedPipelines() {
        return hasDefinedPipelines;
    }

    @Override
    public Collection<GoDashboardPipeline> allPipelines() {
        return pipelines.values();
    }

    @Override
    public boolean hasPipelines() {
        return !pipelines.isEmpty();
    }

    @Override
    public void addPipeline(GoDashboardPipeline pipeline) {
        if (pipeline != null) {
            pipelines.put(pipeline.name().toString(), pipeline);
        }
    }

    protected String digest(String permissionsSegment) {
        try {
            MessageDigest digest = DigestUtils.getSha256Digest();
            OutputStreamWriter outputStreamWriter = new OutputStreamWriter(
                    new DigestOutputStream(new NullOutputStream(), digest));
            outputStreamWriter.write(getClass().getSimpleName());
            outputStreamWriter.write("$");
            outputStreamWriter.write(name());
            outputStreamWriter.write("/");
            outputStreamWriter.write(permissionsSegment);
            outputStreamWriter.write("[");

            for (GoDashboardPipeline pipeline : allPipelines()) {
                outputStreamWriter.write(pipeline.cacheSegment());
                outputStreamWriter.write(",");
            }

            outputStreamWriter.write("]");
            outputStreamWriter.flush();

            return Hex.encodeHexString(digest.digest());
        } catch (IOException e) {
            throw new UncheckedIOException(e);
        }
    }
}