function: implement is_vendor function (#830)

function: implement is_vendor function

Author: ajnavarro

integration_test.go

@@ -345,6 +345,15 @@ func TestIntegration(t *testing.T) {
 			`,
 			[]sql.Row{{"worktree", int64(9)}},
 		},
+		{
+			`SELECT cf.file_path
+			FROM refs r
+			INNER JOIN commit_files cf
+			ON r.commit_hash = cf.commit_hash
+				AND r.repository_id = cf.repository_id
+			WHERE r.ref_name = 'HEAD' AND IS_VENDOR(cf.file_path)`,
+			[]sql.Row{{".gitignore"}, {"vendor/foo.go"}},
+		},
 	}
 
 	var pid uint64

internal/function/is_vendor.go

@@ -0,0 +1,58 @@
+package function
+
+import (
+	"fmt"
+
+	enry "gopkg.in/src-d/enry.v1"
+	"gopkg.in/src-d/go-mysql-server.v0/sql"
+	"gopkg.in/src-d/go-mysql-server.v0/sql/expression"
+)
+
+// IsVendor reports whether files are vendored or not.
+type IsVendor struct {
+	expression.UnaryExpression
+}
+
+// NewIsVendor creates a new IsVendor function.
+func NewIsVendor(filePath sql.Expression) sql.Expression {
+	return &IsVendor{expression.UnaryExpression{Child: filePath}}
+}
+
+// Type implements the sql.Expression interface.
+func (v *IsVendor) Type() sql.Type { return sql.Boolean }
+
+// Eval implements the sql.Expression interface.
+func (v *IsVendor) Eval(ctx *sql.Context, row sql.Row) (interface{}, error) {
+	span, ctx := ctx.Span("function.IsVendor")
+	defer span.Finish()
+
+	val, err := v.Child.Eval(ctx, row)
+	if err != nil {
+		return nil, err
+	}
+
+	if val == nil {
+		return nil, nil
+	}
+
+	val, err = sql.Text.Convert(val)
+	if err != nil {
+		return nil, err
+	}
+
+	return enry.IsVendor(val.(string)), nil
+}
+
+func (v *IsVendor) String() string {
+	return fmt.Sprintf("IS_VENDOR(%s)", v.Child)
+}
+
+// TransformUp implements the sql.Expression interface.
+func (v *IsVendor) TransformUp(f sql.TransformExprFunc) (sql.Expression, error) {
+	child, err := v.Child.TransformUp(f)
+	if err != nil {
+		return nil, err
+	}
+
+	return f(NewIsVendor(child))
+}

internal/function/is_vendor_test.go

@@ -0,0 +1,47 @@
+package function
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"gopkg.in/src-d/go-mysql-server.v0/sql"
+	"gopkg.in/src-d/go-mysql-server.v0/sql/expression"
+)
+
+func TestIsVendor(t *testing.T) {
+	testCases := []struct {
+		name     string
+		path     interface{}
+		expected interface{}
+	}{
+		{
+			"non vendored path",
+			"some/folder/foo.go",
+			false,
+		},
+		{
+			"nil",
+			nil,
+			nil,
+		},
+		{
+			"vendored path",
+			"vendor/foo.go",
+			true,
+		},
+		{
+			"vendored (no root) path",
+			"foo/bar/vendor/foo.go",
+			true,
+		},
+	}
+
+	fn := NewIsVendor(expression.NewGetField(0, sql.Text, "x", true))
+	for _, tt := range testCases {
+		t.Run(tt.name, func(t *testing.T) {
+			result, err := fn.Eval(sql.NewEmptyContext(), sql.Row{tt.path})
+			require.NoError(t, err)
+			require.Equal(t, tt.expected, result)
+		})
+	}
+}

internal/function/registry.go

@@ -13,4 +13,5 @@ var Functions = []sql.Function{
 	sql.Function2{Name: "uast_xpath", Fn: NewUASTXPath},
 	sql.Function2{Name: "uast_extract", Fn: NewUASTExtract},
 	sql.Function1{Name: "uast_children", Fn: NewUASTChildren},
+	sql.Function1{Name: "is_vendor", Fn: NewIsVendor},
 }