43 lines
1.2 KiB
TypeScript
43 lines
1.2 KiB
TypeScript
/*
|
|
* Copyright (C) Sapphirecode - All Rights Reserved
|
|
* This file is part of graphviz-builder which is released under MIT.
|
|
* See file 'LICENSE' for full license details.
|
|
* Created by Timo Hocker <timo@scode.ovh>, May 2020
|
|
*/
|
|
|
|
import { EdgeStyles } from '../enums/Styles';
|
|
import { Color } from './Color';
|
|
|
|
export class Edge {
|
|
public origin: string;
|
|
public target: string;
|
|
public style?: EdgeStyles;
|
|
public color?: Color;
|
|
private _directional: boolean;
|
|
|
|
public constructor (origin: string, target: string, directional: boolean) {
|
|
this.origin = origin;
|
|
this.target = target;
|
|
this._directional = directional;
|
|
}
|
|
|
|
// eslint-disable-next-line @typescript-eslint/naming-convention
|
|
public toString (): string {
|
|
const attributes = [];
|
|
|
|
if (this.style)
|
|
attributes.push ({ name: 'style', value: this.style.toString () });
|
|
if (this.color)
|
|
attributes.push ({ name: 'color', value: this.color.toString () });
|
|
|
|
const attr_string = ` [${attributes.map ((v) => `${v.name}="${v.value}"`)
|
|
.join (',')}]`;
|
|
|
|
return `${this.origin} -${
|
|
this._directional ? '>' : '-'
|
|
} ${this.target}${attributes.length > 0
|
|
? attr_string
|
|
: ''}`;
|
|
}
|
|
}
|