steve/pkg/table/mapper.go

package table

import (
	"github.com/rancher/naok/pkg/attributes"
	"github.com/rancher/norman/pkg/data"
	"github.com/rancher/norman/pkg/types"
)

type Column struct {
	Name  string `json:"name,omitempty"`
	Field string `json:"field,omitempty"`
	Type  string `json:"type,omitempty"`
}

type Table struct {
	Columns  []Column
	Computed func(data.Object)
}

type ColumnMapper struct {
	definition Table
	types.EmptyMapper
}

func (t *ColumnMapper) FromInternal(d data.Object) {
	d.Map("metadata").Set("columns", t.definition.Columns)
	if t.definition.Computed != nil {
		t.definition.Computed(d)
	}
}

func (t *ColumnMapper) ModifySchema(schema *types.Schema, schemas *types.Schemas) error {
	attributes.SetColumns(schema, t.definition.Columns)
	return nil
}