Merge pull request #12 from nao1215/add-header-command

Add header command

Author: nao1215

README.md

@@ -43,14 +43,15 @@ If the --sql option is not specified, the sqly shell is started. sqly command ar
 The command beginning with a dot is the sqly helper command; I plan to add more features in the future to make the sqly shell run more comfortably.
 ```
 $ sqly 
-sqly v0.0.7 (work in progress)
+sqly v0.2.1 (work in progress)
 
 enter "SQL query" or "sqly command that beginning with a dot".
 .help print usage, .exit exit sqly.
 
-sqly> .help                                                                                        
+sqly> .help
       .dump: dump db table to csv file
       .exit: exit sqly
+    .header: print table header
       .help: print help message
     .import: import csv file(s)
     .tables: print tables

shell/command.go

@@ -27,6 +27,7 @@ func NewCommands() CommandList {
 	c := CommandList{}
 	c[".dump"] = command{execute: c.dumpCommand, name: ".dump", description: "dump db table to csv file"}
 	c[".exit"] = command{execute: c.exitCommand, name: ".exit", description: "exit sqly"}
+	c[".header"] = command{execute: c.headerCommand, name: ".header", description: "print table header"}
 	c[".help"] = command{execute: c.helpCommand, name: ".help", description: "print help message"}
 	c[".import"] = command{execute: c.importCommand, name: ".import", description: "import csv file(s)"}
 	c[".tables"] = command{execute: c.tablesCommand, name: ".tables", description: "print tables"}

shell/header.go

@@ -0,0 +1,39 @@
+package shell
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/nao1215/sqly/domain/model"
+	"github.com/olekukonko/tablewriter"
+)
+
+// headerCommand print table header.
+func (c CommandList) headerCommand(s *Shell, argv []string) error {
+	if len(argv) == 0 {
+		fmt.Fprintln(Stdout, "[Usage]")
+		fmt.Fprintln(Stdout, "  .header TABLE_NAME")
+		return nil
+	}
+
+	table, err := s.sqlite3Interactor.List(s.Ctx, argv[0])
+	if err != nil {
+		return err
+	}
+	table.Name = argv[0]
+	printHeader(os.Stdout, table)
+	return nil
+}
+
+// printHeader print header
+func printHeader(out *os.File, t *model.Table) {
+	table := tablewriter.NewWriter(out)
+	table.SetHeader([]string{t.Name})
+	table.SetAutoWrapText(false)
+	table.SetAutoFormatHeaders(false)
+
+	for _, v := range t.Header {
+		table.Append([]string{v})
+	}
+	table.Render()
+}

shell/shell.go

@@ -187,6 +187,20 @@ func (s *Shell) completer(d prompt.Document) []prompt.Suggest {
 			Text:        v.Name,
 			Description: v.Name + " table",
 		})
+
+		table, err := s.sqlite3Interactor.List(s.Ctx, v.Name)
+		if err != nil {
+			// TODO: error logging
+			continue
+		}
+		table.Name = v.Name
+
+		for _, h := range table.Header {
+			suggest = append(suggest, prompt.Suggest{
+				Text:        h,
+				Description: "header in " + table.Name + " table",
+			})
+		}
 	}
 	return prompt.FilterHasPrefix(suggest, d.GetWordBeforeCursor(), true)
 }