refactor

Author: keiravillekode

exercises/practice/protein-translation/.meta/example.fut

@@ -14,18 +14,24 @@ local def contains (needle: []u8) (haystack: [][]u8): bool =
   in
     result
 
+local def is_stop (codon: []u8): bool =
+  contains codon ["UAA", "UAG", "UGA"]
+
+local def translate (codon: []u8): amino_acid =
+  if contains codon ["AUG"] then #methionine else
+  if contains codon ["UUU", "UUC"] then #phenylalanine else
+  if contains codon ["UUA", "UUG"] then #leucine else
+  if contains codon ["UCU", "UCC", "UCA", "UCG"] then #serine else
+  if contains codon ["UAU", "UAC"] then #tyrosine else
+  if contains codon ["UGU", "UGC"] then #cysteine else
+  assert (contains codon ["UGG"]) #tryptophan
+
 def proteins (strand: []u8): []amino_acid =
   let (result, _) = loop (result, index) = ([], 0) while index < length strand do
     let index = index + 3
     let codon = assert (index <= length strand) strand[index - 3:index]
     in
-      if contains codon ["AUG"] then (result ++ [#methionine], index) else
-      if contains codon ["UUU", "UUC"] then (result ++ [#phenylalanine], index) else
-      if contains codon ["UUA", "UUG"] then (result ++ [#leucine], index) else
-      if contains codon ["UCU", "UCC", "UCA", "UCG"] then (result ++ [#serine], index) else
-      if contains codon ["UAU", "UAC"] then (result ++ [#tyrosine], index) else
-      if contains codon ["UGU", "UGC"] then (result ++ [#cysteine], index) else
-      if contains codon ["UGG"] then (result ++ [#tryptophan], index) else
-      assert (contains codon ["UAA", "UAG", "UGA"]) (result, length strand)
+      if is_stop codon then (result, length strand) else
+      (result ++ [translate codon], index)
   in
     result