ISSUE-405: init commit with simple code gen

gradle/libs.versions.toml

@@ -13,6 +13,13 @@ androidXTest = "1.5.0"
 testOrchestrator = "1.4.2"
 lifecycle = "2.5.1"
 thirdPartyReport = "0.18.956"
+agp = "7.2.2"
+org-jetbrains-kotlin-android = "1.9.0"
+core-ktx = "1.9.0"
+junit = "4.13.2"
+androidx-test-ext-junit = "1.1.5"
+appcompat = "1.6.1"
+material = "1.8.0"
 
 [libraries]
 # plugins
@@ -54,6 +61,7 @@ kakaoCompose = { module = "io.github.kakaocup:compose", version.ref = "kakaoComp
 kakaoExtClicks = { module = "io.github.kakaocup:kakao-ext-clicks", version.ref = "kakaoExtClicks" }
 junit = "junit:junit:4.13.2"
 junitJupiter = "org.junit.jupiter:junit-jupiter:5.9.0"
+assertj = "org.assertj:assertj-core:3.11.1"
 truth = "com.google.truth:truth:1.1.3"
 
 androidXTestCore = { module = "androidx.test:core", version.ref = "androidXTest" }
@@ -80,8 +88,15 @@ allureKotlinModel = { module = "io.qameta.allure:allure-kotlin-model", version.r
 allureKotlinCommons = { module = "io.qameta.allure:allure-kotlin-commons", version.ref = "allure" }
 allureKotlinJunit4 = { module = "io.qameta.allure:allure-kotlin-junit4", version.ref = "allure" }
 allureKotlinAndroid = { module = "io.qameta.allure:allure-kotlin-android", version.ref = "allure" }
-
+core-ktx = { group = "androidx.core", name = "core-ktx", version.ref = "core-ktx" }
+junit-junit = { group = "junit", name = "junit", version.ref = "junit" }
+androidx-test-ext-junit = { group = "androidx.test.ext", name = "junit", version.ref = "androidx-test-ext-junit" }
+androidx-appcompat-appcompat = { group = "androidx.appcompat", name = "appcompat", version.ref = "appcompat" }
+com-google-android-material-material = { group = "com.google.android.material", name = "material", version.ref = "material" }
 [bundles]
 espresso = ["espressoCore", "espressoWeb", "accessibilityFramework"]
 allure = ["allureKotlinModel", "allureKotlinCommons", "allureKotlinJunit4", "allureKotlinAndroid"]
 compose = ["composeActivity", "composeUiTooling", "composeMaterial", "composeTestManifest", "composeCompiler"]
+[plugins]
+com-android-library = { id = "com.android.library", version.ref = "agp" }
+org-jetbrains-kotlin-android = { id = "org.jetbrains.kotlin.android", version.ref = "org-jetbrains-kotlin-android" }

page-object-code-gen/build.gradle.kts

@@ -0,0 +1,19 @@
+plugins {
+    id("convention.android-library")
+    id("convention.publication-android-lib")
+    id("convention.third-party-report")
+    id("convention.legal-documents")
+}
+
+publish {
+    artifactId.set("page-object-code-gen")
+}
+
+dependencies {
+    implementation(libs.kotlinStdlib)
+    implementation(libs.androidXTestCore)
+    implementation(libs.uiAutomator)
+    testImplementation(libs.junit)
+    testImplementation(libs.truth)
+    testImplementation(libs.assertj)
+}

page-object-code-gen/src/main/AndroidManifest.xml

@@ -0,0 +1 @@
+<manifest package="com.kaspersky.components.pageobjectcodegen" />

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/CreatePageObjectFromUiDump.kt

@@ -0,0 +1,133 @@
+package com.kaspresso.components.pageobjectcodegen
+
+import com.kaspresso.components.pageobjectcodegen.SupportedViews.collectableElements
+import com.kaspresso.components.pageobjectcodegen.SupportedViews.elementsWithChild
+import org.w3c.dom.Node
+import java.io.File
+import java.nio.charset.Charset
+import javax.xml.parsers.DocumentBuilderFactory
+
+/**
+ * inputs:
+ * 1. Path to xml file with UI Dump
+ * 2. Name of generated class
+ * 3. Path for generated file
+ * output:
+ * Kotlin file with screen code in the same directory as jar execute
+ */
+fun main(vararg args: String) {
+    lateinit var inputFilePath: String
+    lateinit var className: String
+
+    try {
+        inputFilePath = args[0]
+    } catch (e: Exception) {
+        throw Exception("No file path")
+    }
+
+    if (!File(inputFilePath).isFile) {
+        throw Exception("File is not exist or directory")
+    }
+
+    className = try {
+        args[1]
+    } catch (e: Exception) {
+        println("You put empty class name, we change it to \"TestClass\"")
+        "TestClass"
+    }
+
+    if (!className.contains(Regex("^[A-Z]\\S*$"))) {
+        println("You put incorrect class name, we change it to \"TestClass\"")
+        className = "TestClass"
+    }
+
+    var outputFilePath: String = try {
+        args[2]
+    } catch (e: Exception) {
+        println("Output file will be locate in directory where you ran this script with name $className.kt")
+        ""
+    }
+
+    if (!File(outputFilePath).exists() && outputFilePath.isNotEmpty()) {
+        File(outputFilePath).mkdirs()
+    }
+
+    outputFilePath = if (outputFilePath.isNotEmpty()) {
+        "$outputFilePath/$className.kt"
+    } else {
+        "$className.kt"
+    }
+
+    val filePackage = outputFilePath.findPackage()
+
+    val documentBuilderFactory = DocumentBuilderFactory.newInstance()
+    val documentBuilder = documentBuilderFactory.newDocumentBuilder()
+    val doc = documentBuilder.parse(inputFilePath)
+
+    val screenElements: List<BaseView> = findAllViewInDump(doc.firstChild.firstChild)
+    print(screenElements)
+
+    PageObjectGenerator(screenElements, filePackage, className).writeToFile(outputFilePath)
+}
+
+fun findAllViewInDump(root: Node, goToSiblings: Boolean = true): MutableList<BaseView> {
+    val result = mutableListOf<BaseView>()
+    if (root.nodeName == "node") {
+        if (root.attributes.getNamedItem("class").nodeValue in collectableElements) {
+            result.add(getViewFromNode(root))
+        }
+        if (root.attributes.getNamedItem("class").nodeValue in elementsWithChild) {
+            val res = mutableSetOf<List<BaseView>>()
+            val children = root.childNodes
+            for (i in 0 until children.length) {
+                if (children.item(i).nodeName == "node") {
+                    res.add(findAllViewInDump(children.item(i), false))
+                }
+            }
+            result.add(getViewWithChildrenFromNode(root, res))
+        }
+        if (root.hasChildNodes() && root.attributes.getNamedItem("class").nodeValue !in elementsWithChild) {
+            result.addAll(findAllViewInDump(root.firstChild))
+        }
+    }
+    if (root.nextSibling != null && goToSiblings) {
+        result.addAll(findAllViewInDump(root.nextSibling))
+    }
+    return result
+}
+
+fun getViewWithChildrenFromNode(node: Node, childViews: Set<List<BaseView>>): RecyclerView {
+    val attr = node.attributes
+    return RecyclerView(
+        attr.getNamedItem("resource-id").nodeValue.substringAfterLast("/"),
+        attr.getNamedItem("class").nodeValue.substringAfterLast("."),
+        attr.getNamedItem("package").nodeValue,
+        childViews,
+    )
+}
+
+fun getViewFromNode(node: Node): View {
+    val attr = node.attributes
+    return View(
+        attr.getNamedItem("resource-id").nodeValue.substringAfterLast("/"),
+        attr.getNamedItem("class").nodeValue.substringAfterLast("."),
+        attr.getNamedItem("package").nodeValue,
+    )
+}
+
+fun String.findPackage(): String {
+    return split("/").toMutableList()
+        .dropWhile { it != "com" }.dropLast(1).joinToString(separator = ".")
+}
+
+fun Generator.writeToFile(filePath: String) {
+    val writer = TextWriter()
+    generate(writer)
+    val file = File(filePath)
+    val printWriter = file.printWriter(Charset.forName("UTF-8"))
+    try {
+        printWriter.print(writer.toString())
+    } finally {
+        printWriter.close()
+    }
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/Generator.kt

@@ -0,0 +1,5 @@
+package com.kaspresso.components.pageobjectcodegen
+
+interface Generator {
+    fun generate(writer: TextWriter)
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/KotlinCodeGenerator.kt

@@ -0,0 +1,33 @@
+package com.kaspresso.components.pageobjectcodegen
+
+abstract class KotlinCodeGenerator(val elements: List<BaseView>, val filePackage: String) : Generator {
+    override fun generate(writer: TextWriter) {
+        with(writer) {
+            if (filePackage.isNotEmpty()) {
+                append("package $filePackage", 2)
+            }
+            createImports(elements).forEach {
+                append(it)
+            }
+            nextLine()
+        }
+    }
+
+    private fun createImports(screenElements: List<BaseView>): List<String> {
+        val importsList = mutableSetOf("import com.screen.common.KScreen", "import ${screenElements.first().packages}.R")
+
+        for (element in screenElements) {
+            when (element.viewType) {
+                "ImageView" -> importsList.add("import io.guthub.kakaocup.kakao.image.KImageView")
+                "Button" -> importsList.add("import io.guthub.kakaocup.kakao.text.KButton")
+                "TextView" -> importsList.add("import io.guthub.kakaocup.kakao.text.KTextView")
+                "EditText" -> importsList.add("import io.guthub.kakaocup.kakao.edit.KEditText")
+            }
+            if (element is RecyclerView) {
+                importsList.addAll(mutableSetOf("import io.guthub.kakaocup.kakao.recycler.KRecyclerView", "import io.guthub.kakaocup.kakao.recycler.KRecyclerItem"))
+                importsList.addAll(createImports(element.childView.flatten()))
+            }
+        }
+        return importsList.sorted()
+    }
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/PageObjectGenerator.kt

@@ -0,0 +1,46 @@
+package com.kaspresso.components.pageobjectcodegen
+
+class PageObjectGenerator(elements: List<BaseView>, filePackage: String, private val className: String) :
+    KotlinCodeGenerator(elements, filePackage) {
+    override fun generate(writer: TextWriter) {
+        super.generate(writer)
+        with(writer) {
+            codeBlock("object $className : KScreen<$className>()") {
+                append(LAYOUT)
+                append(VIEWCLASS, 2)
+                createElements(elements).forEach {
+                    append(it)
+                }
+                nextLine()
+                elements.forEach { view ->
+                    if (view is RecyclerView) {
+                        for (i in 0 until view.childView.size) {
+                            codeBlock(
+                                "class ${view.childClassNames[i]}(matcher: Matcher<View>) : KRecyclerItem<${view.childClassNames[i]}>(matcher)",
+                                countOfLinesAfterBegin = 1,
+                                countOfLinesAfterEnd = 2,
+                            ) {
+                                createElements(view.childView.elementAt(i)).forEach {
+                                    append(it, countOfLinesAfterText = 0, countOfLinesBeforeText = 1)
+                                }
+                            }
+                        }
+                    }
+                }
+                codeBlock("override fun BaseTestContext.waitForScreen()", countOfLinesAfterBegin = 1) {
+                    append(TODO, 0)
+                }
+            }
+        }
+    }
+
+    private fun createElements(screenElements: List<BaseView>): List<String> {
+        return screenElements.map { it.toKaspressoExpression() }
+    }
+
+    companion object Constants {
+        private const val LAYOUT = "override val layoutId: Int? = TODO(\"Need To Implement\")"
+        private const val VIEWCLASS = "override val viewClass: Class<*>? = TODO(\"Need To Implement\")"
+        private const val TODO = "TODO(\"Need To Implement\")"
+    }
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/SupportedViews.kt

@@ -0,0 +1,6 @@
+package com.kaspresso.components.pageobjectcodegen
+
+object SupportedViews {
+    val collectableElements = listOf("android.widget.Button", "android.widget.TextView", "android.widget.ImageView")
+    val elementsWithChild = listOf("androidx.recyclerview.widget.RecyclerView")
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/TextWriter.kt

@@ -0,0 +1,62 @@
+package com.kaspresso.components.pageobjectcodegen
+
+import java.lang.StringBuilder
+
+class TextWriter(private val indentation: Int = 0) {
+
+    private val line = StringBuilder()
+    private val lines = mutableListOf<Any>()
+
+    init {
+        append(" ".repeat(indentation), 0)
+    }
+
+    fun append(text: String, countOfLinesAfterText: Int = 1, countOfLinesBeforeText: Int = 0): TextWriter = apply {
+        nextLine(countOfLinesBeforeText)
+        line.append(text)
+        nextLine(countOfLinesAfterText)
+    }
+
+    fun nextLine(count: Int = 1): TextWriter = apply {
+        for (i in 0 until count) {
+            commitLine()
+            initNewLine()
+        }
+    }
+
+    private fun commitLine() {
+        if (!line.all { it == ' ' }) {
+            lines.add(line.toString())
+        } else {
+            lines.add("")
+        }
+    }
+
+    private fun initNewLine() {
+        line.setLength(0)
+        line.append(" ".repeat(indentation))
+    }
+
+    private fun withIncreasedIndentation(): TextWriter {
+        val writer = TextWriter(indentation + INDENTATION_STEP)
+        lines.add(writer)
+        return writer
+    }
+
+    override fun toString(): String {
+        commitLine()
+        return lines.joinToString("\n")
+    }
+
+    fun codeBlock(header: String, countOfLinesAfterBegin: Int = 2, countOfLinesAfterEnd: Int = 0, block: TextWriter.() -> Unit) {
+        append("$header {", countOfLinesAfterBegin)
+        with(withIncreasedIndentation()) {
+            block()
+        }
+        append("}", countOfLinesAfterEnd)
+    }
+
+    companion object Constants {
+        private const val INDENTATION_STEP: Int = 4
+    }
+}

page-object-code-gen/src/main/java/com/kaspresso/components/pageobjectcodegen/View.kt

@@ -0,0 +1,43 @@
+package com.kaspresso.components.pageobjectcodegen
+
+data class View(
+    override val resourceId: String,
+    override val viewType: String,
+    override val packages: String,
+) : BaseView {
+
+    override fun toKaspressoExpression(): String {
+        return "val ${resourceId.toCamelCase()} = K$viewType { withId(R.id.$resourceId) }"
+    }
+}
+
+data class RecyclerView(
+    override val resourceId: String,
+    override val viewType: String,
+    override val packages: String,
+    val childView: Set<List<BaseView>>,
+) : BaseView {
+
+    val childClassNames = List(childView.size) { if (it == 0) "RecyclerViewItem" else "RecyclerViewItem$it" }
+
+    override fun toKaspressoExpression(): String {
+        return """val ${resourceId.toCamelCase()} = KRecyclerView(
+        builder = { withId(R.id.$resourceId) },
+        itemTypeBuilder = { ${childClassNames.joinToString(separator = ",\n" + "\t".repeat(AMOUNT_OF_TABS)) { "itemType(::$it)" }} },
+    )"""
+    }
+
+    companion object {
+        private const val AMOUNT_OF_TABS = 7
+    }
+}
+
+interface BaseView {
+
+    val resourceId: String
+    val viewType: String
+    val packages: String
+    fun toKaspressoExpression(): String
+
+    fun String.toCamelCase() = replace("_[a-z]".toRegex()) { it.value.last().uppercase() }
+}