From a435036064da168da325792d3f7fb8bee51505c3 Mon Sep 17 00:00:00 2001
From: Steven Tan <steven@steventan.com.au>
Date: Thu, 17 Jun 2021 17:52:44 +1000
Subject: [PATCH] refactor fuel tech groups

---
 components/ui/FuelTechGroupSelector.vue | 18 +++++-------------
 constants/energy-fuel-techs/index.js    |  2 ++
 constants/fuel-tech-group-queries.js    | 10 ++++++++++
 3 files changed, 17 insertions(+), 13 deletions(-)
 create mode 100644 constants/fuel-tech-group-queries.js

diff --git a/components/ui/FuelTechGroupSelector.vue b/components/ui/FuelTechGroupSelector.vue
index fde712a7..62efe9c9 100755
--- a/components/ui/FuelTechGroupSelector.vue
+++ b/components/ui/FuelTechGroupSelector.vue
@@ -19,21 +19,13 @@
 
 <script>
 import { mapGetters } from 'vuex'
-
-const groups = [
-  'Default',
-  'Simplified',
-  'Coal/Gas/Renewables',
-  'Flexibility',
-  'Renewable/Fossil',
-  'Solar/Residual'
-]
+import { ftGroups, GROUP_DEFAULT } from '@/constants/energy-fuel-techs/'
 
 export default {
   data() {
     return {
-      groups,
-      selected: groups[0]
+      groups: ftGroups,
+      selected: ftGroups[0]
     }
   },
 
@@ -73,8 +65,8 @@ export default {
   mounted() {
     if (this.fuelTechGroupName === 'regions') {
       if (this.regionId !== 'nem') {
-        this.$store.dispatch('fuelTechGroupName', 'Default')
-        this.selected = 'Default'
+        this.$store.dispatch('fuelTechGroupName', GROUP_DEFAULT)
+        this.selected = GROUP_DEFAULT
         this.$router.push({ path: `/energy/${this.regionId}` })
       } else {
         this.selected = this.fuelTechGroupName
diff --git a/constants/energy-fuel-techs/index.js b/constants/energy-fuel-techs/index.js
index deec9524..3eb17329 100644
--- a/constants/energy-fuel-techs/index.js
+++ b/constants/energy-fuel-techs/index.js
@@ -21,6 +21,8 @@ Groups[GROUP_FLEXIBILITY] = FlexibilityGroup
 Groups[GROUP_RENEWABLE_FOSSIL] = RenewableFossilGroup
 Groups[GROUP_SOLAR_RESIDUAL] = SolarResidualGroup
 
+export const ftGroups = Object.keys(Groups)
+
 export function getAllGroups(powerEnergyDomains, type) {
   const parsed = {}
   Object.keys(Groups).forEach(key => {
diff --git a/constants/fuel-tech-group-queries.js b/constants/fuel-tech-group-queries.js
new file mode 100644
index 00000000..7bbc61c2
--- /dev/null
+++ b/constants/fuel-tech-group-queries.js
@@ -0,0 +1,10 @@
+import _includes from 'lodash.includes'
+import { ftGroups, GROUP_DEFAULT } from '@/constants/energy-fuel-techs/'
+
+export function isValidFtGroupQuery(query) {
+  return _includes(ftGroups, query)
+}
+
+export function getDefaultFtGroup() {
+  return GROUP_DEFAULT
+}