Implement type-to-string-to-type mappings

pkg/eia/eia_reflection.go

@@ -18,6 +18,7 @@ import (
 // parameters by type
 type MethodSubs struct {
 	TypedParams      map[reflect.Type]reflect.Value // Replace field of specific type with value, must be ptr to type
+	StrTypedParams   map[string]reflect.Value       // Parameter types by string name from eiaapi mappings
 	RequestEditorFns []eiaapi.RequestEditorFn       // Optional request editor functions
 }
 
@@ -25,9 +26,13 @@ type MethodSubs struct {
 // To be more specific, set NameParams, which will overwrite
 // NameContainsParams
 var defaultMethodSubs = MethodSubs{
-	TypedParams: map[reflect.Type]reflect.Value{
-		reflect.TypeOf((*eiaapi.Route1)(nil)): reflect.ValueOf(
-			strconv.Itoa(time.Now().AddDate(-1, 0, 0).Year())),
+	StrTypedParams: map[string]reflect.Value{
+		"Route1": reflect.ValueOf(eiaapi.Route1(
+			strconv.Itoa(time.Now().AddDate(-1, 0, 0).Year()),
+		)),
+		"Route2": reflect.ValueOf(eiaapi.Route1(
+			strconv.Itoa(time.Now().AddDate(-1, 0, 0).Year()),
+		)),
 	},
 }
 
@@ -42,8 +47,7 @@ func (client *Client) GetFacet(ctx context.Context, route string, facet string,
 		return nil, fmt.Errorf("method %s not found", methodName)
 	}
 
-	// Populate facet param anywhere in func call
-	subs.TypedParams[reflect.TypeOf((*eiaapi.FacetId)(nil))] = reflect.ValueOf(facet)
+	subs.StrTypedParams["FacetId"] = reflect.ValueOf(eiaapi.FacetId(facet))
 
 	args := prepMethodArgs(method, methodName, subs)
 
@@ -174,7 +178,9 @@ func (client *Client) GetFacets(ctx context.Context, route string, subs *MethodS
 func DefaultMethodSubs(ctx context.Context) *MethodSubs {
 	subs := defaultMethodSubs
 
-	subs.TypedParams[reflect.TypeOf((*context.Context)(nil)).Elem()] = reflect.ValueOf(ctx)
+	subs.TypedParams = map[reflect.Type]reflect.Value{
+		reflect.TypeOf((*context.Context)(nil)).Elem(): reflect.ValueOf(ctx),
+	}
 
 	return &subs
 }
@@ -226,18 +232,14 @@ func prepMethodArgs(method reflect.Value, name string, subs *MethodSubs) []refle
 			break
 		}
 
-		// Retrieve arg from generated arg mapping due to use of
-		// type aliases (e.g. type Route1 = string) rather than use of
-		// discrete types (e.g. type Route1 string) in generated code
-		funcArgs, ok := eiaapi.FunctionParams[name]
-		if ok {
-			if i > len(funcArgs) {
-				// WARN: This indicates a generator issues
-			} else {
-				argType = funcArgs[i]
+		// Perform type lookups by string match
+		if paramType := eiaapi.GetFuncParamType(name, i); paramType != "" {
+			for name, val := range subs.StrTypedParams {
+				if paramType == name {
+					args = append(args, val)
+					goto next
+				}
 			}
-		} else {
-			// WARN: This indicates a generator issues
 		}
 
 		// Run through type substitutions