From 9c55e07826cfe654198f6ee3b9bb6a43ae28319c Mon Sep 17 00:00:00 2001
From: Joel Verhagen <joel.verhagen@gmail.com>
Date: Wed, 17 Jan 2024 14:13:42 -0500
Subject: [PATCH] Allow electric poles to not be planned in web UI

---
 src/vue/src/components/ElectricPoleForm.vue | 9 +++++++--
 src/vue/src/lib/OilFieldPlanner.ts          | 2 +-
 src/vue/src/stores/OilFieldStore.ts         | 2 ++
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/vue/src/components/ElectricPoleForm.vue b/src/vue/src/components/ElectricPoleForm.vue
index 26d48a79..402777cb 100644
--- a/src/vue/src/components/ElectricPoleForm.vue
+++ b/src/vue/src/components/ElectricPoleForm.vue
@@ -1,7 +1,11 @@
 <template>
   <fieldset class="border p-3 mb-3">
     <legend>Electric poles</legend>
-    <CustomizeSelect custom-label="Entity" label="Entity" idPrefix="electric-pole"
+    <div class="form-check">
+      <input type="checkbox" class="form-check-input" id="add-electric-poles" v-model="addElectricPoles">
+      <label class="form-check-label" for="add-electric-poles">Add electric poles</label>
+    </div>
+    <CustomizeSelect v-if="addElectricPoles" custom-label="Entity" label="Entity" idPrefix="electric-pole"
       :showAdvancedOptions="showAdvancedOptions" :defaultValue="electricPoleEntityName"
       :defaultIsCustom="electricPoleIsCustom" v-model="electricPoleEntityName" v-model:isCustom="electricPoleIsCustom">
       <option value="small-electric-pole">Small electric pole</option>
@@ -9,7 +13,7 @@
       <option value="big-electric-pole">Big electric pole</option>
       <option value="substation">Substation</option>
     </CustomizeSelect>
-    <div class="row" v-show="showAdvancedOptions">
+    <div class="row" v-show="showAdvancedOptions && addElectricPoles">
       <div class="col-lg-4 mt-3">
         <label class="form-label" for="electric-pole-wire-reach">Wire reach</label>
         <input type="text" pattern="\d+(\.\d+)" min="1" max="99" class="form-control" id="electric-pole-wire-reach"
@@ -57,6 +61,7 @@ export default {
   data() {
     return pick(
       storeToRefs(useOilFieldStore()),
+      'addElectricPoles',
       'electricPoleEntityName',
       'electricPoleIsCustom',
       'electricPoleWidth',
diff --git a/src/vue/src/lib/OilFieldPlanner.ts b/src/vue/src/lib/OilFieldPlanner.ts
index 21eccac9..010e75d6 100644
--- a/src/vue/src/lib/OilFieldPlanner.ts
+++ b/src/vue/src/lib/OilFieldPlanner.ts
@@ -8,7 +8,7 @@ type RequestPropertyGetters = {
 
 const requestPropertyGetters: RequestPropertyGetters = {
   addBeacons: (state) => state.addBeacons,
-  addElectricPoles: (_) => true,
+  addElectricPoles: (state) => state.addElectricPoles,
   addFbeOffset: (_) => false,
   beaconEntityName: (state) => state.beaconEntityName.trim(),
   beaconHeight: (state) => state.beaconHeight,
diff --git a/src/vue/src/stores/OilFieldStore.ts b/src/vue/src/stores/OilFieldStore.ts
index 028690c9..cdf9ddc3 100644
--- a/src/vue/src/stores/OilFieldStore.ts
+++ b/src/vue/src/stores/OilFieldStore.ts
@@ -11,6 +11,7 @@ const defaults = {
   pumpjackModule: 'productivity-module-3',
   pumpjackModuleIsCustom: false,
   addBeacons: true,
+  addElectricPoles: true,
   overlapBeacons: true,
   beaconModule: 'speed-module-3',
   beaconModuleIsCustom: false,
@@ -51,6 +52,7 @@ const storeToQuery: StoreToQuery = {
   pumpjackModule: 'pumpMod',
   pumpjackModuleIsCustom: 'pumpModCust',
   addBeacons: 'beacons',
+  addElectricPoles: 'poles',
   overlapBeacons: 'overlapBeacons',
   beaconModule: 'beaconMod',
   beaconModuleIsCustom: 'beaconModCust',