1 files changed, 9 insertions, 0 deletions

diff --git a/src/Sequence/Utils.hs b/src/Sequence/Utils.hs
index 1b52630..dc6657a 100644
--- a/src/Sequence/Utils.hs
+++ b/src/Sequence/Utils.hs
@@ -131,3 +131,12 @@ instance Argument (Set Hitzone) GameState where
     hitzones <- Set.map (view _Hitzone) . Map.keysSet <$> MaybeT (preuse $ gFocus' . eStats . sHitzones)
     guard (hasGlob || ws `Set.isSubsetOf` hitzones)
     return . Set.map (review _Hitzone) $ if hasGlob then hitzones else ws
+
+instance Completion DamageType GameState where
+  completableLabel _ = "<damageType>"
+  complete _ _ prefix = return . filter ((isPrefixOf `on` CI.foldCase) prefix) $ map show ([minBound .. maxBound] :: [DamageType])
+
+instance Argument DamageType GameState where
+  arg (CI.mk -> word) = return $ Map.lookup word types
+    where
+      types = Map.fromList [(CI.mk $ show dType, dType) | dType <- [minBound .. maxBound]]