diff --git a/lib/src/main/groovy/com/jessebrault/ssg/part/PartsMap.groovy b/lib/src/main/groovy/com/jessebrault/ssg/part/PartsMap.groovy index e8df0c8..c89b027 100644 --- a/lib/src/main/groovy/com/jessebrault/ssg/part/PartsMap.groovy +++ b/lib/src/main/groovy/com/jessebrault/ssg/part/PartsMap.groovy @@ -1,22 +1,18 @@ package com.jessebrault.ssg.part +import groovy.transform.ToString + +@ToString(includeFields = true) class PartsMap { + @Delegate private final Map partsMap = [:] PartsMap(Collection parts) { Objects.requireNonNull(parts) parts.each { - this.partsMap.put(it.name, new EmbeddablePart(it)) + this.put(it.name, new EmbeddablePart(it)) } } - EmbeddablePart get(String name) { - this.partsMap.get(Objects.requireNonNull(name)) - } - - EmbeddablePart getAt(String name) { - this.get(Objects.requireNonNull(name)) - } - } diff --git a/lib/src/main/groovy/com/jessebrault/ssg/text/TextsMap.groovy b/lib/src/main/groovy/com/jessebrault/ssg/text/TextsMap.groovy index 050ae08..368a628 100644 --- a/lib/src/main/groovy/com/jessebrault/ssg/text/TextsMap.groovy +++ b/lib/src/main/groovy/com/jessebrault/ssg/text/TextsMap.groovy @@ -1,22 +1,18 @@ package com.jessebrault.ssg.text +import groovy.transform.ToString + +@ToString(includeFields = true) class TextsMap { + @Delegate private final Map textsMap = [:] TextsMap(Collection texts) { Objects.requireNonNull(texts) texts.each { - this.textsMap.put(it.path, new EmbeddableText(it)) + this.put(it.path, new EmbeddableText(it)) } } - EmbeddableText get(String path) { - this.textsMap.get(Objects.requireNonNull(path)) - } - - EmbeddableText getAt(String path) { - this.get(Objects.requireNonNull(path)) - } - }