fix import issue (#16)

* fix import issue * fix shadowing issue
lyft · May 22, 2018 · 33125b4 · 33125b4
1 parent a7e273b
commit 33125b4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/gatherer.go b/gatherer.go
@@ -49,8 +49,11 @@ func (g *gatherer) Generate(f *generator.FileDescriptor) {
 }
 
 func (g *gatherer) hydratePackage(f *generator.FileDescriptor, comments map[string]string) Package {
-	importPath := goImportPath(g.Generator.Unwrap(), f)
-	name := string(g.Generator.GoPackageName(importPath))
+	// TODO(btc): perhaps return error with specific info about failure
+	importPath, name, found := goPackageOption(f)
+	if !found {
+		g.Fail("could not determine import path")
+	}
 
 	g.push("package:" + name)
 	defer g.pop()