org.graylog2.contentpacks.model.entities.DashboardEntity Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of graylog2-server Show documentation
Graylog
There is a newer version: 6.0.1
/*
 * Copyright (C) 2020 Graylog, Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program. If not, see
 * .
 */
package org.graylog2.contentpacks.model.entities;

import com.fasterxml.jackson.annotation.JsonAutoDetect;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.auto.value.AutoValue;
import com.google.common.collect.ImmutableMap;
import org.graylog.autovalue.WithBeanGetter;
import org.graylog.plugins.views.search.views.Titles;
import org.graylog.plugins.views.search.views.WidgetPositionDTO;
import org.graylog2.contentpacks.model.entities.references.ValueReference;

import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

@AutoValue
@WithBeanGetter
@JsonAutoDetect
public abstract class DashboardEntity {
    @JsonProperty("title")
    @NotBlank
    public abstract ValueReference title();

    @JsonProperty("description")
    public abstract ValueReference description();

    @JsonProperty("widgets")
    @NotNull
    public abstract List widgets();

    public static Map  positionMap(
            Map parameters,
            Map> widgetMap) {
        final Map widgetPositionMap = new HashMap<>();
        for (Map.Entry> widgetEntityListEntry: widgetMap.entrySet()) {
            final DashboardWidgetEntity dashboardWidgetEntity = widgetEntityListEntry.getKey();
            Optional position = dashboardWidgetEntity.position();
            widgetEntityListEntry.getValue().forEach(widgetEntity ->
                position.ifPresent(value -> widgetPositionMap.put(widgetEntity.id(), value.convert(parameters))));
        }
        return ImmutableMap.copyOf(widgetPositionMap);
    }

    public static Titles widgetTitles(
            Map> widgetMap,
            Map parameters) {

        final Map widgetTitleMap = new HashMap<>();

        for (Map.Entry> widgetEntityListEntry: widgetMap.entrySet()) {
            widgetEntityListEntry.getValue().forEach(widgetEntity -> {
                widgetTitleMap.put(widgetEntity.id(), widgetEntityListEntry.getKey().description().asString(parameters));
            });
        }
        return Titles.withWidgetTitle(widgetTitleMap);
    }

    @JsonCreator
    public static DashboardEntity create(
            @JsonProperty("title") @NotBlank ValueReference title,
            @JsonProperty("description") ValueReference description,
            @JsonProperty("widgets") @NotNull List widgets) {
        return new AutoValue_DashboardEntity(title, description, widgets);
    }
}