import { PlantUMLPresenter } from "./PlantUMLPresenter";
import { CodeFile, CodeStructure } from "../CodeElement";
import { LANGUAGE_LINE_COMMENT_MAP } from "../../language/LanguageCommentMap";

export class CommentedUmlPresenter extends PlantUMLPresenter {
	present(file: CodeFile): string {
		const commentSymbol = LANGUAGE_LINE_COMMENT_MAP[file.language];
		const plantUml = this.render(file);

		return plantUml.split("\n").map((line) => {
			return `${commentSymbol} ${line}`;
		}).join("\n");
	}

	presentClass(clazz: CodeStructure, lang: string): string {
		const commentSymbol = LANGUAGE_LINE_COMMENT_MAP[lang];
		const plantUml = this.renderClass(clazz);

		return plantUml.split("\n").map((line) => {
			return `${commentSymbol} ${line}`;
		}).join("\n");
	}
}