From 8192a6e56d7a110b22b5f7629dab8d38a5f902d3 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Fri, 13 Apr 2018 13:09:54 +0300
Subject: [PATCH] Teach cleanup remove files with one particular extension

---
 .idea/inspectionProfiles/Project_Default.xml  |  1 +
 src/main/kotlin/nk/patchsets/git/cleanup.kt   | 67 ++++++++++++-------
 .../kotlin/nk/patchsets/git/restore-branch.kt |  3 +-
 3 files changed, 47 insertions(+), 24 deletions(-)

diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml
index ecc04fc..11e643c 100644
--- a/.idea/inspectionProfiles/Project_Default.xml
+++ b/.idea/inspectionProfiles/Project_Default.xml
@@ -1,6 +1,7 @@
 <component name="InspectionProjectProfileManager">
   <profile version="1.0">
     <option name="myName" value="Project Default" />
+    <inspection_tool class="FoldInitializerAndIfToElvis" enabled="true" level="INFORMATION" enabled_by_default="true" />
     <inspection_tool class="PackageDirectoryMismatch" enabled="true" level="INFORMATION" enabled_by_default="true" />
     <inspection_tool class="Since15" enabled="true" level="ERROR" enabled_by_default="true">
       <effectiveLL value="JDK_1_7" />
diff --git a/src/main/kotlin/nk/patchsets/git/cleanup.kt b/src/main/kotlin/nk/patchsets/git/cleanup.kt
index 872bfbe..4a100a3 100644
--- a/src/main/kotlin/nk/patchsets/git/cleanup.kt
+++ b/src/main/kotlin/nk/patchsets/git/cleanup.kt
@@ -8,23 +8,31 @@ import nk.patchsets.git.restore.isGitDir
 import nk.patchsets.git.restore.isGradleDir
 import java.io.File
 
-const val DEFAULT_CLEANUP_COMMIT_TITLE = "~~~~ cleanup ~~~~"
+const val EXT_PATTERN = "{ext}"
+const val DEFAULT_CLEANUP_COMMIT_TITLE = "~~~~ cleanup$EXT_PATTERN ~~~~"
 const val NO_COMMIT_ = "--no-commit"
+const val EXT_ = "-ext="
 
-data class Settings(val repoPath: String, val commitTitle: String? = DEFAULT_CLEANUP_COMMIT_TITLE, val isNoCommit: Boolean)
+data class Settings(
+        val repoPath: String,
+        val extension: String?,
+        val commitTitle: String? = DEFAULT_CLEANUP_COMMIT_TITLE,
+        val isNoCommit: Boolean)
 
 fun main(args: Array<String>) {
     cleanup(args)
 }
 
 fun cleanup(args: Array<String>) {
-    if (args.size != 1 && args.size != 2) {
+    if (!(args.size in 1..3)) {
         System.err.println("""
-            Usage: <git-path> <commit-title>? $NO_COMMIT_?
+            Usage: <git-path> ($EXT_<file-extension>)? (<commit-title>|$NO_COMMIT_)?
 
-            <git-path>      - Directory with repository (parent directory for .git folder).
-            <commit-title>  - Title for cleanup commit. "$DEFAULT_CLEANUP_COMMIT_TITLE" is used by default.
-            $NO_COMMIT_     - option to cancel cleanup commit.
+            <git-path>            - Directory with repository (parent directory for .git folder).
+            $EXT_<file-extension> - Particular extension for remove.
+                                    All files with extensions found in .bunch file will be removed if not set.
+            <commit-title>        - Title for cleanup commit. "$DEFAULT_CLEANUP_COMMIT_TITLE" is used by default.
+            $NO_COMMIT_           - option to cancel cleanup commit.
 
             Example:
             <program> C:/Projects/kotlin $NO_COMMIT_
@@ -33,32 +41,43 @@ fun cleanup(args: Array<String>) {
         return
     }
 
-    val isNoCommit = args.size == 2 && args[1] == NO_COMMIT_
-    val settings = if (!isNoCommit) {
-        Settings(
-                repoPath = args[0],
-                commitTitle = args.getOrNull(1) ?: DEFAULT_CLEANUP_COMMIT_TITLE,
-                isNoCommit = false
-        )
-    } else {
-        Settings(
-                repoPath = args[0],
-                commitTitle = null,
-                isNoCommit = true
-        )
+    val repoPath = args[0]
+    val extension = args.getOrNull(1)?.takeIf { it.startsWith(EXT_) }?.substringAfter(EXT_)
+    val commitIndex = if (extension == null) 1 else 2
+
+    val commitTitleOrNoCommit = args.getOrNull(commitIndex)
+    if (commitIndex == 1 && args.size == 3) {
+        System.err.println("Unknown parameter: '${args[2]}'")
+        return
     }
 
+    val isNoCommit = commitTitleOrNoCommit == NO_COMMIT_
+    val commitTitle = if (!isNoCommit) commitTitleOrNoCommit ?: DEFAULT_CLEANUP_COMMIT_TITLE else null
+
+    val settings = Settings(
+            repoPath = repoPath,
+            extension = extension,
+            commitTitle = commitTitle,
+            isNoCommit = isNoCommit
+    )
+
     cleanup(settings)
 }
 
 fun cleanup(settings: Settings) {
-    val extensions = readExtensionFromFile(settings.repoPath)?.toSet() ?: return
+    val extensions = if (settings.extension != null) {
+        setOf(settings.extension)
+    } else {
+        readExtensionFromFile(settings.repoPath)?.toSet()
+    }?.map { ".$it" }
+
+    if (extensions == null) return
 
     val root = File(settings.repoPath)
     val filesWithExtensions = root
             .walkTopDown()
             .onEnter { dir -> !(isGitDir(dir) || isGradleDir(dir)) }
-            .filter { child -> child.extension in extensions }
+            .filter { child -> extensions.any { child.name.endsWith(it) } }
             .toList()
 
     val changedFiles = ArrayList<FileChange>()
@@ -73,6 +92,8 @@ fun cleanup(settings: Settings) {
     }
 
     if (!settings.isNoCommit) {
-        commitChanges(settings.repoPath, changedFiles, settings.commitTitle!!)
+        val extValue = if (settings.extension != null) " ${settings.extension}" else ""
+        val commitTitle = settings.commitTitle!!.replace(EXT_PATTERN, extValue)
+        commitChanges(settings.repoPath, changedFiles, commitTitle)
     }
 }
diff --git a/src/main/kotlin/nk/patchsets/git/restore-branch.kt b/src/main/kotlin/nk/patchsets/git/restore-branch.kt
index 6b6318d..268c77d 100644
--- a/src/main/kotlin/nk/patchsets/git/restore-branch.kt
+++ b/src/main/kotlin/nk/patchsets/git/restore-branch.kt
@@ -66,7 +66,8 @@ fun restore(args: Array<String>) {
     }
 
     if (settings.doCleanup) {
-        cleanup(nk.patchsets.git.cleanup.Settings(settings.repoPath, RESTORE_CLEANUP_COMMIT_TITLE, false))
+        cleanup(nk.patchsets.git.cleanup.Settings(
+                settings.repoPath, extension = null, commitTitle =  RESTORE_CLEANUP_COMMIT_TITLE, isNoCommit = false))
     }
 }