package schema

import (
	"sort"
)

// Table is a single SQLite table.
type Table struct {
	TableName string `db:"name"`

	// One of "table", "view", "shadow", or "virtual"
	TableType      string `db:"table_type"`
	IsStrict       bool   `db:"is_strict"`
	IsWithoutRowid bool   `db:"is_without_rowid"`

	Columns []Column

	TypeIDName string

	// Default variable name for variables of this type to use when generating Go code
	VarName string

	// Name of corresponding model type to be generated
	GoTypeName string
}

// PrimaryKeyColumns returns the ordered list of columns in this table's primary key.
// This can be useful for "without rowid" tables with composite primary keys.
//
// TODO: needs test
func (t Table) PrimaryKeyColumns() []Column {
	pks := make([]Column, 0)
	for _, c := range t.Columns {
		if c.IsPrimaryKey {
			pks = append(pks, c)
		}
	}
	sort.Slice(pks, func(i, j int) bool {
		return pks[i].PrimaryKeyRank < pks[j].PrimaryKeyRank
	})
	return pks
}

func (t Table) GetColumnByName(name string) Column {
	for _, c := range t.Columns {
		if c.Name == name {
			return c
		}
	}
	panic("no such column: " + name)
}

func (t Table) HasAutoTimestamps() (hasCreatedAt bool, hasUpdatedAt bool) {
	for _, c := range t.Columns {
		if c.Name == "created_at" && c.Type == "integer" {
			hasCreatedAt = true
		}
		if c.Name == "updated_at" && c.Type == "integer" {
			hasUpdatedAt = true
		}
	}
	return
}