chore: refactor

moul · Aug 11, 2019 · b7a0ce5 · b7a0ce5
1 parent f64f9e5
commit b7a0ce5
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 10 deletions.
diff --git a/cryptoguess/cryptoguess.go b/cryptoguess/cryptoguess.go
@@ -116,6 +116,26 @@ type Result interface {
 	Rest() []byte
 }
 
+//
+// recursive
+//
+
+func recursiveResults(parent Result, input []byte) []Result {
+	results := []Result{}
+	for _, childExperiment := range New(input).Experiments {
+		for _, childResult := range childExperiment.Results() {
+			results = append(
+				results,
+				&stackedResult{
+					parent: parent,
+					child:  childResult,
+				},
+			)
+		}
+	}
+	return results
+}
+
 //
 // stackedResult
 //

diff --git a/cryptoguess/guess_pem.go b/cryptoguess/guess_pem.go
@@ -21,16 +21,8 @@ func runPEMBlock(exp Experiment) []Result {
 		// FIXME: data.Type
 		// FIXME: data.Headers
 		result := &baseResult{exp: exp, rest: rest, data: data}
-		for _, childExperiment := range New(data.Bytes).Experiments {
-			for _, childResult := range childExperiment.Results() {
-				results = append(
-					results,
-					&stackedResult{
-						parent: result,
-						child:  childResult,
-					},
-				)
-			}
+		for _, childResult := range recursiveResults(result, data.Bytes) {
+			results = append(results, childResult)
 		}
 		results = append(results, result)
 	}