2 files changed, 7 insertions, 2 deletions
diff --git a/src/main/kotlin/moe/nea/firmament/features/diana/DianaWaypoints.kt b/src/main/kotlin/moe/nea/firmament/features/diana/DianaWaypoints.kt
index 6422f67..9f72967 100644
--- a/src/main/kotlin/moe/nea/firmament/features/diana/DianaWaypoints.kt
+++ b/src/main/kotlin/moe/nea/firmament/features/diana/DianaWaypoints.kt
@@ -17,11 +17,12 @@ import moe.nea.firmament.features.FirmamentFeature
 import moe.nea.firmament.gui.config.ManagedConfig
 
 object DianaWaypoints : FirmamentFeature {
-    override val identifier get() = "diana-waypoints"
+    override val identifier get() = "diana"
     override val config get() = TConfig
 
     object TConfig : ManagedConfig(identifier) {
-        val ancestralSpadeSolver by toggle("ancestral-spade") { false }
+        val ancestralSpadeSolver by toggle("ancestral-spade") { true }
+        val nearbyWaypoints by toggle("nearby-waypoints") { true }
     }
 
     override fun onLoad() {
diff --git a/src/main/kotlin/moe/nea/firmament/features/diana/NearbyBurrowsSolver.kt b/src/main/kotlin/moe/nea/firmament/features/diana/NearbyBurrowsSolver.kt
index 9fd7792..1ab7e01 100644
--- a/src/main/kotlin/moe/nea/firmament/features/diana/NearbyBurrowsSolver.kt
+++ b/src/main/kotlin/moe/nea/firmament/features/diana/NearbyBurrowsSolver.kt
@@ -63,6 +63,7 @@ object NearbyBurrowsSolver {
     }
 
     fun onParticles(event: ParticleSpawnEvent) {
+        if (!DianaWaypoints.TConfig.nearbyWaypoints) return
 
         val position: BlockPos = event.position.toBlockPos().down()
 
@@ -106,6 +107,7 @@ object NearbyBurrowsSolver {
     }
 
     fun onRender(event: WorldRenderLastEvent) {
+        if (!DianaWaypoints.TConfig.nearbyWaypoints) return
         renderInWorld(event) {
             for ((location, burrow) in burrows) {
                 when (burrow) {
@@ -126,6 +128,8 @@ object NearbyBurrowsSolver {
     }
 
     fun onBlockClick(blockPos: BlockPos) {
+        if (!DianaWaypoints.TConfig.nearbyWaypoints) return
+        burrows.remove(blockPos)
         lastBlockClick = blockPos
     }
 }