com.github.chriskn.structurizrextension.model.SoftwareSystem.kt Maven / Gradle / Ivy

Go to download
package com.github.chriskn.structurizrextension.model

import com.structurizr.io.plantuml.C4PlantUMLWriter
import com.structurizr.model.Container
import com.structurizr.model.Location
import com.structurizr.model.SoftwareSystem
import com.structurizr.model.StaticStructureElement

/**
 * Adds a container.
 *
 * @param name          the name of the container
 * @param description   the description of the container
 * @param location      the [Location]] of the container
 * @param c4Type          the [C4Type] of the container
 * @param icon          the icon of the container. See IconRegistry for available icons or add your own
 * @param link          the link of the container
 * @param tags          the list of tags of the container
 * @param properties    the [C4Properties] of the container
 * @param uses          the list of [Dependency] to a system, container or component the container uses. A person can't be used
 * @param usedBy        the list of [Dependency] to a system, container, component or person the container is used by
 * @return the Container created and added to the system (or null)
 * @throws IllegalArgumentException if a container with the same name already exists or a person is used in an uses dependency
 */
@Suppress("LongParameterList")
fun SoftwareSystem.container(
    name: String,
    description: String,
    location: Location = this.location,
    c4Type: C4Type? = null,
    icon: String? = null,
    link: String? = null,
    technology: String = "",
    tags: List = listOf(),
    properties: C4Properties? = null,
    uses: List> = listOf(),
    usedBy: List> = listOf()
): Container {
    val container = this.addContainer(name, description, technology)
    container.c4Type = c4Type
    container.location = location
    container.configure(icon, link, tags, properties, uses, usedBy)
    return container
}

var SoftwareSystem.c4Type: C4Type?
    get() = if (this.properties.containsKey(C4PlantUMLWriter.C4_ELEMENT_TYPE)) {
        C4Type.fromC4Type(this.properties.getValue(C4PlantUMLWriter.C4_ELEMENT_TYPE))
    } else {
        null
    }
    set(value) {
        if (value != null) {
            this.addProperty(C4PlantUMLWriter.C4_ELEMENT_TYPE, value.c4Type)
        }
    }