package main

import (
	"fmt"

	"github.com/pinzolo/dbmodel"
)

func main() {
	// Create DataSouce
	ds := dbmodel.NewDataSource("postgres", "9.4", "localhost", 5432, "postgres", "", "sample", map[string]string{"sslmode": "disable"})

	// Create Client
	client := dbmodel.NewClient(ds)

	// Connect to Database.
	client.Connect()
	// You must close connection.
	defer client.Disconnect()

	// AllTables returns all table in sample schema.
	// dbmodel.RequireAll is built in option.
	// When dbmodel.RequireAll is given, client loads all metadata of table.(columns, indices, constraints, foreign keys, referenced keys)
	tables, err := client.AllTables("sample", dbmodel.RequireAll)
	if err != nil {
		fmt.Println(err)
		return
	}
	for _, tbl := range tables {
		for _, col := range tbl.Columns() {
			fmt.Printf("%#v", col)
		}
		for _, idx := range tbl.Indices() {
			fmt.Printf("%#v", idx)
		}
		for _, cns := range tbl.Constraints() {
			fmt.Printf("%#v", cns)
		}
		for _, fk := range tbl.ForeignKeys() {
			fmt.Printf("%#v", fk)
		}
		for _, rk := range tbl.ReferencedKeys() {
			fmt.Printf("%#v", rk)
		}
	}

	// You can load single table users.
	// You need only columns, you can use dbmodel.RequireNone
	table, err := client.Table("sample", "users", dbmodel.RequireNone)
	fmt.Printf("%#v", table)
	if err != nil {
		fmt.Println(err)
		return
	}

	// You can load table names.
	// Returned *dbmodel.Table contains only table name and comment.
	// As well as using client.TableNames("sample", "users"), you can get tables that contains "users" in its name.
	tables, err = client.AllTableNames("sample")
	if err != nil {
		fmt.Println(err)
		return
	}
}