git: add objects table, closes #54 (#77)

Author: ajnavarro

git/database.go

@@ -13,6 +13,7 @@ const (
 	tagsTableName        = "tags"
 	blobsTableName       = "blobs"
 	treeEntriesTableName = "tree_entries"
+	objectsTableName     = "objects"
 )
 
 type Database struct {
@@ -22,6 +23,7 @@ type Database struct {
 	rr   sql.Table
 	ter  sql.Table
 	br   sql.Table
+	or   sql.Table
 }
 
 func NewDatabase(name string, r *git.Repository) sql.Database {
@@ -32,6 +34,7 @@ func NewDatabase(name string, r *git.Repository) sql.Database {
 		tr:   newTagsTable(r),
 		br:   newBlobsTable(r),
 		ter:  newTreeEntriesTable(r),
+		or:   newObjectsTable(r),
 	}
 }
 
@@ -46,5 +49,6 @@ func (d *Database) Tables() map[string]sql.Table {
 		referencesTableName:  d.rr,
 		blobsTableName:       d.br,
 		treeEntriesTableName: d.ter,
+		objectsTableName:     d.or,
 	}
 }

git/database_test.go

@@ -38,6 +38,7 @@ func TestDatabase_Tables(t *testing.T) {
 		treeEntriesTableName,
 		tagsTableName,
 		blobsTableName,
+		objectsTableName,
 	}
 	sort.Strings(expected)
 

git/objects.go

@@ -0,0 +1,71 @@
+package git
+
+import (
+	"github.com/gitql/gitql/sql"
+
+	"gopkg.in/src-d/go-git.v4"
+)
+
+type objectsTable struct {
+	r *git.Repository
+}
+
+func newObjectsTable(r *git.Repository) sql.Table {
+	return &objectsTable{r: r}
+}
+
+func (objectsTable) Resolved() bool {
+	return true
+}
+
+func (objectsTable) Name() string {
+	return objectsTableName
+}
+
+func (objectsTable) Schema() sql.Schema {
+	return sql.Schema{
+		sql.Field{"id", sql.String},
+		sql.Field{"type", sql.String},
+	}
+}
+
+func (r *objectsTable) TransformUp(f func(sql.Node) sql.Node) sql.Node {
+	return f(r)
+}
+
+func (r *objectsTable) TransformExpressionsUp(f func(sql.Expression) sql.Expression) sql.Node {
+	return r
+}
+
+func (r objectsTable) RowIter() (sql.RowIter, error) {
+	oIter, err := r.r.Objects()
+	if err != nil {
+		return nil, err
+	}
+	iter := &objectIter{i: oIter}
+	return iter, nil
+}
+
+func (objectsTable) Children() []sql.Node {
+	return []sql.Node{}
+}
+
+type objectIter struct {
+	i *git.ObjectIter
+}
+
+func (i *objectIter) Next() (sql.Row, error) {
+	o, err := i.i.Next()
+	if err != nil {
+		return nil, err
+	}
+
+	return objectToRow(o), nil
+}
+
+func objectToRow(o git.Object) sql.Row {
+	return sql.NewMemoryRow(
+		o.ID().String(),
+		o.Type().String(),
+	)
+}

git/objects_test.go

@@ -0,0 +1,43 @@
+package git
+
+import (
+	"testing"
+
+	"github.com/gitql/gitql/sql"
+
+	"github.com/stretchr/testify/assert"
+	"gopkg.in/src-d/go-git.v4/fixtures"
+)
+
+func TestObjectsTable_Name(t *testing.T) {
+	assert := assert.New(t)
+
+	f := fixtures.Basic().One()
+	table := getTable(assert, f, objectsTableName)
+	assert.Equal(objectsTableName, table.Name())
+}
+
+func TestObjectsTable_Children(t *testing.T) {
+	assert := assert.New(t)
+
+	f := fixtures.Basic().One()
+	table := getTable(assert, f, objectsTableName)
+	assert.Equal(0, len(table.Children()))
+}
+
+func TestObjectsTable_RowIter(t *testing.T) {
+	assert := assert.New(t)
+
+	f := fixtures.Basic().One()
+	table := getTable(assert, f, objectsTableName)
+
+	rows, err := sql.NodeToRows(table)
+	assert.Nil(err)
+	assert.Len(rows, 31)
+
+	schema := table.Schema()
+	for idx, row := range rows {
+		err := schema.CheckRow(row)
+		assert.Nil(err, "row %d doesn't conform to schema", idx)
+	}
+}