[Code health] Don't allow saving a duplicate vertex during polygon drawing session

app/src/main/java/org/groundplatform/android/ui/datacollection/tasks/polygon/DrawAreaTaskViewModel.kt

@@ -150,8 +150,6 @@ internal constructor(
       .stateIn(viewModelScope, WhileSubscribed(5_000), emptyList())
   }
 
-
-
   override fun initialize(
     job: Job,
     task: Task,
@@ -326,36 +324,37 @@ internal constructor(
    * This coroutine runs on [viewModelScope] to ensure lifecycle safety.
    */
   private fun refreshMap() = viewModelScope.launch {
-    if (session.vertices.isEmpty()) {
+    val vertices = session.displayVertices
+    if (vertices.isEmpty()) {
       _draftArea.emit(null)
       draftTag = null
     } else {
       if (draftTag == null) {
-        val feature = buildPolygonFeature()
+        val feature = buildPolygonFeature(vertices)
         draftTag = feature.tag
         _draftArea.emit(feature)
       } else {
-        val feature = buildPolygonFeature(id = draftTag!!.id)
+        val feature = buildPolygonFeature(vertices, id = draftTag!!.id)
         _draftUpdates.tryEmit(feature)
       }
     }
   }
 
-  private suspend fun buildPolygonFeature(id: String? = null) =
+  private suspend fun buildPolygonFeature(vertices: List<Coordinates>, id: String? = null) =
     Feature(
       id = id ?: uuidGenerator.generateUuid(),
       type = Feature.Type.USER_POLYGON,
-      geometry = LineString(session.vertices),
+      geometry = LineString(vertices),
       style = featureStyle,
       clusterable = false,
       selected = true,
-      tooltipText = getDistanceTooltipText(),
+      tooltipText = getDistanceTooltipText(vertices),
     )
 
   /** Returns the distance in meters between the last two vertices for displaying in the tooltip. */
-  private fun getDistanceTooltipText(): String? {
-    if (sessionState.value.isMarkedComplete || session.vertices.size <= 1) return null
-    val distance = session.vertices.penult().distanceTo(session.vertices.last())
+  private fun getDistanceTooltipText(vertices: List<Coordinates>): String? {
+    if (sessionState.value.isMarkedComplete || vertices.size <= 1) return null
+    val distance = vertices.penult().distanceTo(vertices.last())
     if (distance < TOOLTIP_MIN_DISTANCE_METERS) return null
     return localeAwareMeasureFormatter.formatDistance(distance, measurementUnits)
   }

app/src/main/java/org/groundplatform/android/ui/datacollection/tasks/polygon/PolygonDrawingSession.kt

@@ -28,6 +28,12 @@ interface PolygonDrawingSession {
   /** The list of committed vertices in the current drawing session. */
   val vertices: List<Coordinates>
 
+  /** The tentative vertex that tracks the uncommitted active position (e.g. camera target). */
+  val tentativeVertex: Coordinates?
+
+  /** The complete list of vertices (committed + tentative) to display on the map. */
+  val displayVertices: List<Coordinates>
+
   /** Whether the current polygon geometry self-intersects. */
   val hasSelfIntersection: Boolean
 

app/src/main/java/org/groundplatform/android/ui/datacollection/tasks/polygon/PolygonDrawingSessionImpl.kt

@@ -20,6 +20,7 @@ import kotlinx.collections.immutable.toImmutableList
 import org.groundplatform.android.ui.datacollection.tasks.polygon.PolygonDrawingSession.Companion.DISTANCE_THRESHOLD_DP
 import org.groundplatform.domain.model.geometry.Coordinates
 import org.groundplatform.domain.model.geometry.LineString
+import org.groundplatform.domain.util.isClosed
 import org.groundplatform.domain.util.isSelfIntersecting
 
 class PolygonDrawingSessionImpl : PolygonDrawingSession {
@@ -34,56 +35,60 @@ class PolygonDrawingSessionImpl : PolygonDrawingSession {
   override val vertices: List<Coordinates>
     get() = _vertices
 
-  @VisibleForTesting(otherwise = VisibleForTesting.PRIVATE)
-  internal val redoVertexStack = mutableListOf<Coordinates>()
+  private var _tentativeVertex: Coordinates? = null
 
-  override val hasSelfIntersection: Boolean
-    get() = isSelfIntersecting(_vertices)
+  override val tentativeVertex: Coordinates?
+    get() = _tentativeVertex
 
-  private fun addVertex(vertex: Coordinates, shouldOverwriteLastVertex: Boolean) {
-    val updatedVertices = _vertices.toMutableList()
-
-    if (shouldOverwriteLastVertex && updatedVertices.isNotEmpty()) {
-      updatedVertices.removeAt(updatedVertices.lastIndex)
-    }
+  override val displayVertices: List<Coordinates>
+    get() = _tentativeVertex?.let { _vertices + it } ?: _vertices
 
-    updatedVertices.add(vertex)
+  @VisibleForTesting(otherwise = VisibleForTesting.PRIVATE)
+  internal val redoVertexStack = mutableListOf<Coordinates>()
 
-    _vertices = updatedVertices.toImmutableList()
-  }
+  override val hasSelfIntersection: Boolean
+    get() = isSelfIntersecting(displayVertices)
 
   override fun setVertices(newVertices: List<Coordinates>) {
     _vertices = newVertices.toImmutableList()
+    _tentativeVertex = null
+    redoVertexStack.clear()
   }
 
   override fun updateTentativeVertex(
     target: Coordinates,
     calculateDistance: (Coordinates, Coordinates) -> Double,
   ) {
+    if (isClosed(_vertices)) {
+      _tentativeVertex = null
+      _isTooClose = true
+      return
+    }
+
     val firstVertex = _vertices.firstOrNull()
     var updatedTarget = target
-    if (firstVertex != null && _vertices.size > 2) {
+    if (firstVertex != null && _vertices.size >= 3) {
       val distance = calculateDistance(firstVertex, target)
 
       if (distance <= DISTANCE_THRESHOLD_DP) {
         updatedTarget = firstVertex
       }
     }
 
-    val prev = _vertices.dropLast(1).lastOrNull()
+    val lastCommitted = _vertices.lastOrNull()
     _isTooClose =
-      _vertices.size > 1 &&
-        prev?.let { calculateDistance(it, target) <= DISTANCE_THRESHOLD_DP } == true
+      lastCommitted?.let { calculateDistance(it, target) <= DISTANCE_THRESHOLD_DP } == true
 
-    addVertex(updatedTarget, true)
+    _tentativeVertex = updatedTarget
   }
 
   override fun commitTentativeVertex(currentCameraTarget: Coordinates?): List<Coordinates>? {
     redoVertexStack.clear()
-    val vertex = _vertices.lastOrNull() ?: currentCameraTarget
+    val vertex = _tentativeVertex ?: currentCameraTarget
     return vertex?.let {
+      _vertices = (_vertices + it).toImmutableList()
+      _tentativeVertex = null
       _isTooClose = _vertices.size > 1
-      addVertex(it, false)
       _vertices
     }
   }
@@ -96,21 +101,30 @@ class PolygonDrawingSessionImpl : PolygonDrawingSession {
     return intersected
   }
 
-  override fun isValidPolygon(): Boolean =
-    LineString(_vertices).isClosed() && !isSelfIntersecting(_vertices)
+  override fun isValidPolygon(): Boolean = LineString(displayVertices).isClosed() &&
+    !isSelfIntersecting(displayVertices) &&
+    displayVertices.distinct().size >= 3
 
   override fun complete() {
     check(isValidPolygon()) { "Polygon is not valid" }
     check(!_isMarkedComplete) { "Already marked complete" }
+    if (_tentativeVertex != null) {
+      commitTentativeVertex(null)
+    }
     _isMarkedComplete = true
   }
 
   override fun removeLastVertex(): Boolean {
-    if (_vertices.isEmpty()) return false
+    if (_tentativeVertex == null && _vertices.isEmpty()) return false
 
     _isMarkedComplete = false
-    redoVertexStack.add(_vertices.last())
-    _vertices = _vertices.dropLast(1).toImmutableList()
+    _tentativeVertex = null
+
+    if (_vertices.isNotEmpty()) {
+      redoVertexStack.add(_vertices.last())
+      _vertices = _vertices.dropLast(1).toImmutableList()
+    }
+
     return true
   }
 
@@ -120,6 +134,7 @@ class PolygonDrawingSessionImpl : PolygonDrawingSession {
     _isMarkedComplete = false
     val redoVertex = redoVertexStack.removeAt(redoVertexStack.lastIndex)
     _vertices = (_vertices + redoVertex).toImmutableList()
+    _tentativeVertex = null
     return redoVertex
   }
 

app/src/test/java/org/groundplatform/android/ui/datacollection/tasks/polygon/DrawAreaTaskScreenTest.kt

@@ -119,7 +119,6 @@ class DrawAreaTaskScreenTest : BaseHiltTest() {
               Coordinates(0.0, 0.0),
               Coordinates(10.0, 10.0),
               Coordinates(20.0, 20.0),
-              Coordinates(20.0, 20.0),
             )
           )
         )

app/src/test/java/org/groundplatform/android/ui/datacollection/tasks/polygon/DrawAreaTaskViewModelTest.kt

@@ -115,7 +115,7 @@ class DrawAreaTaskViewModelTest : BaseHiltTest() {
 
     viewModel.removeLastVertex()
 
-    assertGeometry(3, isLineString = true)
+    assertGeometry(2, isLineString = true)
   }
 
   @Test
@@ -199,11 +199,11 @@ class DrawAreaTaskViewModelTest : BaseHiltTest() {
 
     viewModel.removeLastVertex()
 
-    assertThat(featureTestObserver.value()?.tooltipText).isEqualTo("1,565,109 m")
+    assertThat(featureTestObserver.value()?.tooltipText).isNull()
 
     viewModel.removeLastVertex()
 
-    assertThat(featureTestObserver.value()?.tooltipText).isEqualTo(null)
+    assertThat(featureTestObserver.value()?.tooltipText).isNull()
   }
 
   @Test
@@ -568,10 +568,10 @@ class DrawAreaTaskViewModelTest : BaseHiltTest() {
     updateLastVertexAndAdd(COORDINATE_2)
 
     viewModel.removeLastVertex()
-    assertGeometry(2, isLineString = true)
+    assertGeometry(1, isLineString = true)
 
     viewModel.redoLastVertex()
-    assertGeometry(3, isLineString = true)
+    assertGeometry(2, isLineString = true)
   }
 
   @Test