check result type of methods

Author: smallnest

tool/xgen/parser/parser.go

@@ -57,8 +57,13 @@ func (v *visitor) Visit(n ast.Node) (w ast.Visitor) {
 		return nil
 	case *ast.FuncDecl:
 		if isExported(v.name) {
-			if n.Type.Params.NumFields() == 3 {
+			if n.Type.Params.NumFields() == 3 && n.Type.Results.NumFields() == 1 {
 				params := n.Type.Params.List
+				result := n.Type.Results.List[0]
+
+				if result.Type.(*ast.Ident).Name != "error" {
+					return nil
+				}
 
 				if p, ok := params[0].Type.(*ast.SelectorExpr); ok {
 					x := p.X.(*ast.Ident)

tool/xgen/xgen.go

@@ -48,6 +48,13 @@ func main() {
 			panic(err)
 		}
 
+		if !filepath.IsAbs(fname) {
+			fname, err = filepath.Abs(fname)
+			if err != nil {
+				panic(err)
+			}
+		}
+
 		relDir, err := filepath.Rel(filepath.Join(build.Default.GOPATH, "src"), fname)
 		if err != nil {
 			fmt.Printf("provided directory not under GOPATH (%s): %v",