116 lines
2.2 KiB
Go
116 lines
2.2 KiB
Go
|
package main
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"path"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/Sirupsen/logrus"
|
||
|
"github.com/codegangsta/cli"
|
||
|
|
||
|
"eriol.xyz/piken/format"
|
||
|
"eriol.xyz/piken/sql"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
unicodeDataUrl = "http://www.unicode.org/Public/UNIDATA/UnicodeData.txt"
|
||
|
pikenHome = ".piken"
|
||
|
defaultDatabaseFile = "piken.sqlite3"
|
||
|
defaultDataFile = "UnicodeData.txt"
|
||
|
version = "0.1a"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
baseDir = path.Join(getHome(), pikenHome)
|
||
|
databaseFile = path.Join(baseDir, defaultDatabaseFile)
|
||
|
dataFile = path.Join(baseDir, defaultDataFile)
|
||
|
store sql.Store
|
||
|
)
|
||
|
|
||
|
func main() {
|
||
|
|
||
|
app := cli.NewApp()
|
||
|
app.Name = "piken"
|
||
|
app.Version = version
|
||
|
app.Author = "Daniele Tricoli"
|
||
|
app.Email = "eriol@mornie.org"
|
||
|
app.Usage = "unicode search tool"
|
||
|
|
||
|
if _, err := os.Stat(baseDir); os.IsNotExist(err) {
|
||
|
os.Mkdir(baseDir, 0755)
|
||
|
}
|
||
|
|
||
|
if err := store.Open(databaseFile); err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
defer store.Close()
|
||
|
|
||
|
app.Commands = []cli.Command{
|
||
|
{
|
||
|
Name: "update",
|
||
|
Usage: "Update unicode data",
|
||
|
Action: func(c *cli.Context) {
|
||
|
modifiedTime, err := checkLastModified(unicodeDataUrl)
|
||
|
if err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
|
||
|
lastUpdate, err := store.GetLastUpdate(defaultDataFile)
|
||
|
if err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if lastUpdate.Before(modifiedTime) {
|
||
|
download(unicodeDataUrl, dataFile)
|
||
|
|
||
|
records, err := readCsvFile(dataFile)
|
||
|
if err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if err := store.LoadFromRecords(records); err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if err := store.CreateLastUpdate(defaultDataFile,
|
||
|
modifiedTime); err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
} else {
|
||
|
logrus.Info("Already up to date.")
|
||
|
}
|
||
|
|
||
|
},
|
||
|
},
|
||
|
{
|
||
|
Name: "search",
|
||
|
Aliases: []string{"s"},
|
||
|
Usage: "Search for unicode",
|
||
|
Action: func(c *cli.Context) {
|
||
|
args := strings.Join(c.Args(), " ")
|
||
|
rows, err := store.SearchUnicode(args)
|
||
|
if err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
|
||
|
formatter := format.NewTextFormatter(
|
||
|
[]string{"CodePoint", "Name"},
|
||
|
" -- ",
|
||
|
true)
|
||
|
for _, row := range rows {
|
||
|
|
||
|
b, err := formatter.Format(&row)
|
||
|
if err != nil {
|
||
|
logrus.Fatal(err)
|
||
|
}
|
||
|
fmt.Println(b)
|
||
|
}
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
app.Run(os.Args)
|
||
|
|
||
|
}
|