ViewModel 테스트 템플릿 생성

kimdohun0104 · kimdohun0104 · commit b7a133e51fdc · 2021-02-19T11:48:26.000+09:00
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/action/CreateFindaTemplateAction.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/action/CreateFindaTemplateAction.kt
@@ -11,9 +11,10 @@ import kr.co.finda.androidtemplate.type.PluginError
 import kr.co.finda.androidtemplate.feature.createFindaTemplate.dialog.CreateFindaTemplateDialog
 import kr.co.finda.androidtemplate.model.ActionRouterImpl
 
-class CreateFindaTemplateAction : AnAction(), CreateFindaTemplateActionContract.View {
+class CreateFindaTemplateAction :
+    AnAction(), CreateFindaTemplateActionContract.View {
 
-    private val presenter: CreateFindaTemplateActionPresenter by lazy {
+    private val presenter: CreateFindaTemplateActionContract.Presenter by lazy {
         CreateFindaTemplateActionPresenter(this, ActionRouterImpl())
     }
 
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/action/CreateFindaTemplateActionPresenter.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/action/CreateFindaTemplateActionPresenter.kt
@@ -35,6 +35,6 @@ class CreateFindaTemplateActionPresenter(
     }
 
     private fun isValidSelectedDirectoryPath(path: String): Boolean {
-        return path.contains("^.*src/main/(java|kotlin).*$".toRegex())
+        return actionRouter.isCreateFindaTemplateEnable(path)
     }
 }
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/dialog/CreateFindaTemplateDialog.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createFindaTemplate/dialog/CreateFindaTemplateDialog.kt
@@ -37,7 +37,7 @@ class CreateFindaTemplateDialog(
     override fun createCenterPanel(): JComponent {
         return panel {
 
-            row("화면 이름") {
+            row("화면 이름:") {
                 nameTextField = textField({ "" }, {}).component
             }
 
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestAction.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestAction.kt
@@ -0,0 +1,45 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.action
+
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.CommonDataKeys
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.ui.Messages
+import com.intellij.openapi.vfs.VirtualFile
+import icons.Icons
+import kr.co.finda.androidtemplate.feature.createViewModelTest.dialog.CreateViewModelTestDialog
+import kr.co.finda.androidtemplate.model.ActionRouterImpl
+import kr.co.finda.androidtemplate.type.PluginError
+
+class CreateViewModelTestAction :
+    AnAction(), CreateViewModelTestActionContract.View {
+
+    private val presenter: CreateViewModelTestActionContract.Presenter by lazy {
+        CreateViewModelTestActionPresenter(this, ActionRouterImpl())
+    }
+
+    override fun actionPerformed(e: AnActionEvent) {
+        val project = e.project!!
+        val selectedDirectory = e.getData(CommonDataKeys.VIRTUAL_FILE)
+
+        presenter.onCreateViewModelTestTemplateActionPerformed(project, selectedDirectory)
+    }
+
+    override fun update(event: AnActionEvent) {
+        val selectedDirectory = event.getData(CommonDataKeys.VIRTUAL_FILE)
+        presenter.onActionUpdate(event, selectedDirectory)
+    }
+
+    override fun showErrorDialog(project: Project, pluginError: PluginError) {
+        Messages.showMessageDialog(
+            project,
+            pluginError.message,
+            pluginError.title,
+            Icons.FindaLogo
+        )
+    }
+
+    override fun showCreateViewModelTestTemplateDialog(project: Project, selectedDirectory: VirtualFile) {
+        CreateViewModelTestDialog(project, selectedDirectory).show()
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestActionContract.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestActionContract.kt
@@ -0,0 +1,29 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.action
+
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import kr.co.finda.androidtemplate.type.PluginError
+
+interface CreateViewModelTestActionContract {
+
+    interface View {
+
+        fun showErrorDialog(project: Project, pluginError: PluginError)
+
+        fun showCreateViewModelTestTemplateDialog(project: Project, selectedDirectory: VirtualFile)
+    }
+
+    interface Presenter {
+
+        fun onActionUpdate(
+            event: AnActionEvent,
+            selectedDirectory: VirtualFile?
+        )
+
+        fun onCreateViewModelTestTemplateActionPerformed(
+            project: Project,
+            selectedDirectory: VirtualFile?
+        )
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestActionPresenter.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/action/CreateViewModelTestActionPresenter.kt
@@ -0,0 +1,39 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.action
+
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import kr.co.finda.androidtemplate.model.ActionRouter
+import kr.co.finda.androidtemplate.type.PluginError
+
+class CreateViewModelTestActionPresenter(
+    private val view: CreateViewModelTestActionContract.View,
+    private val actionRouter: ActionRouter
+) : CreateViewModelTestActionContract.Presenter {
+
+    override fun onActionUpdate(
+        event: AnActionEvent,
+        selectedDirectory: VirtualFile?
+    ) {
+        event.presentation.isEnabledAndVisible =
+            if (selectedDirectory != null) actionRouter.isCreateViewModelTestTemplateEnable(selectedDirectory.path)
+            else false
+    }
+
+    override fun onCreateViewModelTestTemplateActionPerformed(
+        project: Project,
+        selectedDirectory: VirtualFile?
+    ) {
+        if (selectedDirectory == null) {
+            view.showErrorDialog(project, PluginError.VTT101)
+            return
+        }
+
+        if (!actionRouter.isCreateViewModelTestTemplateEnable(selectedDirectory.path)) {
+            view.showErrorDialog(project, PluginError.VTT102)
+            return
+        }
+
+        view.showCreateViewModelTestTemplateDialog(project, selectedDirectory)
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialog.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialog.kt
@@ -0,0 +1,56 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.dialog
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.ui.DialogWrapper
+import com.intellij.openapi.ui.Messages
+import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.ui.components.JBTextField
+import com.intellij.ui.layout.panel
+import icons.Icons
+import kr.co.finda.androidtemplate.model.FileHelperImpl
+import kr.co.finda.androidtemplate.model.ReplacerImpl
+import kr.co.finda.androidtemplate.type.PluginError
+import javax.swing.JComponent
+
+class CreateViewModelTestDialog(
+    private val project: Project,
+    private val selectedDirectory: VirtualFile
+) : DialogWrapper(true), CreateViewModelTestDialogContract.View {
+
+    private val presenter: CreateViewModelTestDialogContract.Presenter by lazy {
+        CreateViewModelTestDialogPresenter(this, FileHelperImpl(ReplacerImpl()))
+    }
+
+    private lateinit var nameTextField: JBTextField
+
+    init {
+        init()
+        title = "Create ViewModel Test Template"
+    }
+
+    override fun createCenterPanel(): JComponent {
+        return panel {
+            row("ViewModel 이름:") {
+                nameTextField = textField({ "" }, {}).component
+            }
+        }
+    }
+
+    override fun doOKAction() {
+        presenter.onCreateViewModelTest(
+            project = project,
+            name = nameTextField.text,
+            selectedDirectory = selectedDirectory
+        )
+        super.doOKAction()
+    }
+
+    override fun showConflictNameDialog(project: Project, pluginError: PluginError, conflictedFileName: String) {
+        Messages.showMessageDialog(
+            project,
+            "${pluginError.message}\n${conflictedFileName}",
+            pluginError.title,
+            Icons.FindaLogo
+        )
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialogContract.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialogContract.kt
@@ -0,0 +1,24 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.dialog
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import kr.co.finda.androidtemplate.type.PluginError
+
+interface CreateViewModelTestDialogContract {
+
+    interface View {
+        fun showConflictNameDialog(
+            project: Project,
+            pluginError: PluginError,
+            conflictedFileName: String
+        )
+    }
+
+    interface Presenter {
+        fun onCreateViewModelTest(
+            project: Project,
+            name: String,
+            selectedDirectory: VirtualFile
+        )
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialogPresenter.kt b/src/main/kotlin/kr/co/finda/androidtemplate/feature/createViewModelTest/dialog/CreateViewModelTestDialogPresenter.kt
@@ -0,0 +1,61 @@
+package kr.co.finda.androidtemplate.feature.createViewModelTest.dialog
+
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.VirtualFile
+import kr.co.finda.androidtemplate.model.FileHelper
+import kr.co.finda.androidtemplate.model.Replacements
+import kr.co.finda.androidtemplate.type.FileExtension
+import kr.co.finda.androidtemplate.type.PluginError
+
+class CreateViewModelTestDialogPresenter(
+    private val view: CreateViewModelTestDialogContract.View,
+    private val fileHelper: FileHelper
+) : CreateViewModelTestDialogContract.Presenter {
+
+    override fun onCreateViewModelTest(
+        project: Project,
+        name: String,
+        selectedDirectory: VirtualFile
+    ) {
+        getConflictedFileName(name, selectedDirectory.path)?.let {
+            view.showConflictNameDialog(project, PluginError.VTT103, it)
+            return
+        }
+
+        val packageName = fileHelper.getPackageNameByPath(selectedDirectory.path)
+        createViewModelTest(selectedDirectory, name, packageName)
+    }
+
+    private fun getConflictedFileName(
+        name: String,
+        selectedDirectoryPath: String
+    ): String? {
+        val fileNameWithExtension = "${name}ViewModelTest.kt"
+        val isFileExist = fileHelper.isFileExistWithPath(
+            "${selectedDirectoryPath}/${fileNameWithExtension}"
+        )
+
+        return if (isFileExist) {
+            fileNameWithExtension
+        } else {
+            null
+        }
+    }
+
+    private fun createViewModelTest(
+        selectedDirectory: VirtualFile,
+        name: String,
+        packageName: String
+    ) {
+        fileHelper.createFileWithTemplate(
+            selectedDirectory,
+            "${name}ViewModelTest",
+            FileExtension.Kotlin,
+            "ViewModelTestTemplate",
+            Replacements(
+                packageName = packageName,
+                name = name
+            )
+        )
+    }
+}
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/model/ActionRouter.kt b/src/main/kotlin/kr/co/finda/androidtemplate/model/ActionRouter.kt
@@ -3,10 +3,16 @@ package kr.co.finda.androidtemplate.model
 interface ActionRouter {
 
     fun isCreateFindaTemplateEnable(path: String): Boolean
+
+    fun isCreateViewModelTestTemplateEnable(path: String): Boolean
 }
 
-class ActionRouterImpl: ActionRouter {
+class ActionRouterImpl : ActionRouter {
     override fun isCreateFindaTemplateEnable(path: String): Boolean {
         return path.contains("src/main/(java|kotlin)".toRegex())
     }
+
+    override fun isCreateViewModelTestTemplateEnable(path: String): Boolean {
+        return path.contains("src/test/(java|kotlin)".toRegex())
+    }
 }
diff --git a/src/main/kotlin/kr/co/finda/androidtemplate/type/PluginError.kt b/src/main/kotlin/kr/co/finda/androidtemplate/type/PluginError.kt
@@ -11,5 +11,12 @@ enum class PluginError(
     FT101("선택된 위치를 찾을 수 없습니다", "FT101"),
     FT102("유효하지 않은 경로입니다", "FT102-'src/main/(java|kotlin)'경로에 해당하지 않습니다"),
     FT103("중복된 이름이 존재합니다", "FT103-이미 생성된 Activity 혹은 ViewModel을 확인해주세요"),
-    FT104("레이아웃 폴더를 찾을 수 없습니다", "FT104-레이아웃 파일이 생성되지 못했습니다")
+    FT104("레이아웃 폴더를 찾을 수 없습니다", "FT104-레이아웃 파일이 생성되지 못했습니다"),
+
+    /**
+     * ViewModel Test Template
+     */
+    VTT101("선택된 위치를 찾을 수 없습니다", "VTT101"),
+    VTT102("유효하지 않은 경로입니다", "VTT102-'src/main/(java|kotlin)/test' 경로에 해당하지 않습니다"),
+    VTT103("중복된 이름이 존재합니다", "VTT103-이미 생성된 테스트 파일을 확인해주세요"),
 }
diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
@@ -13,10 +13,16 @@
                 icon="Icons.FindaLogo">
             <add-to-group group-id="NewGroup" anchor="first"/>
         </action>
+        <action id="kr.co.finda.androidtemplate.feature.createViewModelTest.action.CreateViewModelTestAction"
+                class="kr.co.finda.androidtemplate.feature.createViewModelTest.action.CreateViewModelTestAction"
+                text="New ViewModel Test Template"
+                icon="Icons.FindaLogo">
+            <add-to-group group-id="NewGroup" anchor="first"/>
+        </action>
     </actions>
 
     <extensions defaultExtensionNs="com.intellij">
-        <defaultLiveTemplates file="/liveTemplates/FindaTest.xml" />
-        <liveTemplateContext implementation="kr.co.finda.androidtemplate.feature.FindaTestTemplateContextType" />
+        <defaultLiveTemplates file="/liveTemplates/FindaTest.xml"/>
+        <liveTemplateContext implementation="kr.co.finda.androidtemplate.feature.FindaTestTemplateContextType"/>
     </extensions>
 </idea-plugin>
diff --git a/src/main/resources/templates/ViewModelTestTemplate.txt b/src/main/resources/templates/ViewModelTestTemplate.txt
@@ -0,0 +1,14 @@
+package $PACKAGE$
+
+import kr.co.finda.BasicTest
+
+class $NAME$ViewModelTest : BasicTest() {
+
+    private lateinit var viewModel: $NAME$ViewModel
+
+    fun buildViewModel(): $NAME$ViewModel {
+        return $NAME$ViewModel()
+    }
+
+
+}

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,6 @@ class CreateFindaTemplateActionPresenter(`
`35`	`35`	`}`
`36`	`36`
`37`	`37`	`private fun isValidSelectedDirectoryPath(path: String): Boolean {`
`38`		`- return path.contains("^.src/main/(java\|kotlin).$".toRegex())`
	`38`	`+ return actionRouter.isCreateFindaTemplateEnable(path)`
`39`	`39`	`}`
`40`	`40`	`}`
Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ class CreateFindaTemplateDialog(`
`37`	`37`	`override fun createCenterPanel(): JComponent {`
`38`	`38`	`return panel {`
`39`	`39`
`40`		`- row("화면 이름") {`
	`40`	`+ row("화면 이름:") {`
`41`	`41`	`nameTextField = textField({ "" }, {}).component`
`42`	`42`	`}`
`43`	`43`
Original file line number	Diff line number	Diff line change
`@@ -3,10 +3,16 @@ package kr.co.finda.androidtemplate.model`
`3`	`3`	`interface ActionRouter {`
`4`	`4`
`5`	`5`	`fun isCreateFindaTemplateEnable(path: String): Boolean`
	`6`	`+`
	`7`	`+ fun isCreateViewModelTestTemplateEnable(path: String): Boolean`
`6`	`8`	`}`
`7`	`9`
`8`		`-class ActionRouterImpl: ActionRouter {`
	`10`	`+class ActionRouterImpl : ActionRouter {`
`9`	`11`	`override fun isCreateFindaTemplateEnable(path: String): Boolean {`
`10`	`12`	`return path.contains("src/main/(java\|kotlin)".toRegex())`
`11`	`13`	`}`
	`14`	`+`
	`15`	`+ override fun isCreateViewModelTestTemplateEnable(path: String): Boolean {`
	`16`	`+ return path.contains("src/test/(java\|kotlin)".toRegex())`
	`17`	`+ }`
`12`	`18`	`}`