Exercise: HTML table DSL
You are working on a project that requires you to generate HTML tables.
fun createTable(): TableBuilder {
val td1 = TdBuilder()
td1.text = "A"
val td2 = TdBuilder()
td2.text = "B"
val tr1 = TrBuilder()
tr1.tds += td1
tr1.tds += td2
val td3 = TdBuilder()
td3.text = "C"
val td4 = TdBuilder()
td4.text = "D"
val tr2 = TrBuilder()
tr2.tds += td3
tr2.tds += td4
val html = TableBuilder()
html.trs += tr1
html.trs += tr2
return html
}
You want to make it more readable and easier to use. You decide to create a DSL for it. You want to be able to create a table in the following way:
fun createTable(): TableBuilder = table {
tr {
td { +"A" }
td { +"B" }
}
tr {
td { +"C" }
td { +"D" }
}
}
To make it possible, you need to four functions: table, tr, td and text. Some of them need to be added inside builder classes.
This problem can either be solved in the below playground or you can clone kotlin-exercises project and solve it locally. In the project, you can find code template for this exercise in functional/dsl/Table.kt. You can find there starting code, example usage and unit tests.
Once you are done with the exercise, you can check your solution here.
Playground
import org.junit.Test
import kotlin.test.assertEquals
fun createTable(): TableBuilder = table {
tr {
td { +"A" }
td { +"B" }
}
tr {
td { +"C" }
td { +"D" }
}
}
data class TableBuilder(
var trs: List<TrBuilder> = emptyList()
) {
override fun toString(): String =
"<table>${trs.joinToString(separator = "")}</table>"
}
data class TrBuilder(
var tds: List<TdBuilder> = emptyList()
) {
override fun toString(): String =
"<tr>${tds.joinToString(separator = "")}</tr>"
}
data class TdBuilder(
var text: String = ""
) {
override fun toString(): String = "<td>$text</td>"
}
fun main() {
// println(createTable()) //<table><tr><td>This is row 1</td><td>This is row 2</td></tr></table>
}
class HtmlDslTest {
@Test
fun createTableTest() {
val html = TableBuilder().apply {
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "A" }
tds += TdBuilder().apply { text = "B" }
}
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "C" }
tds += TdBuilder().apply { text = "D" }
}
}
assertEquals(html, createTable())
}
@Test
fun `Table can be created and it is empty by default`() {
val expected = TableBuilder()
val actual = table {}
assertEquals(expected, actual)
}
@Test
fun `Tr can be created and it is empty`() {
val expected = TableBuilder().apply {
trs += TrBuilder()
}
val actual = table {
tr {}
}
assertEquals(expected, actual)
}
@Test
fun `Multiple tr can be created`() {
val expected = TableBuilder().apply {
trs += TrBuilder()
trs += TrBuilder()
}
val actual = table {
tr {}
tr {}
}
assertEquals(expected, actual)
}
@Test
fun `Td can be created and it is empty`() {
val expected = TableBuilder().apply {
trs += TrBuilder().apply {
tds += TdBuilder()
}
}
val actual = table {
tr {
td {}
}
}
assertEquals(expected, actual)
}
@Test
fun dslTestStandard() {
val expected = TableBuilder().apply {
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "A" }
tds += TdBuilder().apply { text = "B" }
}
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "C" }
tds += TdBuilder().apply { text = "D" }
}
}
val actual = table {
tr {
td { +"A" }
td { +"B" }
}
tr {
td { +"C" }
td { +"D" }
}
}
assertEquals(expected, actual)
}
@Test
fun dslTestMoreColons() {
val expected = TableBuilder().apply {
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "A" }
tds += TdBuilder().apply { text = "B" }
tds += TdBuilder().apply { text = "C" }
}
trs += TrBuilder().apply {
tds += TdBuilder().apply { text = "C" }
tds += TdBuilder().apply { text = "D" }
}
}
val actual = table {
tr {
td { +"A" }
td { +"B" }
td { +"C" }
}
tr {
td { +"C" }
td { +"D" }
}
}
assertEquals(expected, actual)
}
}
