schema_selector.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

package main

import (
	"fmt"

	"github.com/rivo/tview"

	"bnbl.io/pgqt/postgres"
)

func newSchemaSelector(db *postgres.DB) *tview.List {
	v := tview.NewList()
	v.SetTitle("Schemata")
	v.SetBorder(true)
	v.ShowSecondaryText(false)
	v.SetHighlightFullLine(true)

	schemata, err := db.GetSchemata()
	if err != nil {
		v.AddItem(fmt.Sprintf("could not find schemata: %v", err), "", 'a', nil)
		return v
	}

	for _, s := range schemata {
		v.AddItem(s.SchemaName, "", 0, nil)
	}

	return v
}