wi/main.go

110 lines
2.5 KiB
Go
Raw Normal View History

2016-07-21 19:40:19 +00:00
/**
* @file main.go
* @author Mikhail Klementyev jollheef<AT>riseup.net
* @license GNU GPLv3
* @date July, 2016
* @brief Tiny non-interactive cli browser
*/
2016-07-19 22:08:19 +00:00
package main
import (
2016-11-20 14:57:27 +00:00
"os"
2016-07-30 08:05:50 +00:00
"strings"
2016-07-25 15:35:09 +00:00
"github.com/jollheef/wi/commands"
2016-07-22 12:44:12 +00:00
"github.com/jollheef/wi/storage"
2016-07-22 12:43:23 +00:00
2016-11-20 14:57:27 +00:00
cookiejar "github.com/juju/persistent-cookiejar"
2016-07-19 22:08:19 +00:00
kingpin "gopkg.in/alecthomas/kingpin.v2"
)
2016-07-30 08:05:50 +00:00
type searchList []string
func (l *searchList) Set(value string) (err error) {
*l = append(*l, value)
return
}
func (l *searchList) String() (s string) {
return ""
}
func (l *searchList) IsCumulative() bool {
return true
}
func SearchList(settings kingpin.Settings) (target *[]string) {
target = new([]string)
settings.SetValue((*searchList)(target))
return
}
2016-07-19 22:08:19 +00:00
var (
2016-07-24 08:01:55 +00:00
get = kingpin.Command("get", "Get url")
getUrl = get.Arg("url", "Url").Required().String()
2016-11-20 10:22:00 +00:00
form = kingpin.Command("form", "Fill form")
formID = form.Arg("id", "Form ID").Required().Int64()
formArgs = SearchList(form.Arg("args", "Form arguments"))
2016-08-01 00:11:48 +00:00
2016-07-25 15:41:12 +00:00
link = kingpin.Command("link", "Get link")
linkNo = link.Arg("no", "Number").Required().Int64()
linkFromHistory = link.Flag("history", "Item from history").Bool()
2016-07-24 08:01:55 +00:00
historyList = kingpin.Command("history", "List history")
historyListItems = historyList.Arg("items", "Amount of items").Int64()
2016-07-24 08:23:24 +00:00
historyListAll = historyList.Flag("all", "Show all items").Bool()
2016-07-30 08:05:50 +00:00
search = kingpin.Command("search", "Search engine (google by default)")
searchArgs = SearchList(search.Arg("string", "String for search"))
2016-07-19 22:08:19 +00:00
)
2016-07-22 12:43:23 +00:00
func main() {
2016-11-20 15:09:02 +00:00
homePath, exists := os.LookupEnv("HOME")
var wiDir, widbPath, wijarPath string
if exists {
wiDir = homePath + "/.wi"
widbPath = wiDir + "/wi.db"
wijarPath = wiDir + "/wi.jar"
} else {
wiDir = "/tmp"
widbPath = "/tmp/wi.db"
wijarPath = "/tmp/wi.jar"
}
err := os.MkdirAll(wiDir, 0700)
if err != nil {
panic(err)
}
db, err := storage.OpenDB(widbPath)
2016-07-22 12:43:23 +00:00
if err != nil {
panic(err)
}
2016-07-24 08:01:55 +00:00
defer db.Close()
2016-11-20 15:09:02 +00:00
os.Setenv("GOCOOKIES", wijarPath)
2016-11-20 14:57:27 +00:00
jar, err := cookiejar.New(nil)
if err != nil {
panic(err)
}
defer jar.Save()
2016-07-24 08:01:55 +00:00
switch kingpin.Parse() {
case "get":
2016-11-20 14:57:27 +00:00
commands.Get(db, jar, *getUrl)
2016-11-20 10:22:00 +00:00
case "form":
2016-11-20 14:57:27 +00:00
commands.Form(db, jar, *formID, *formArgs)
2016-07-24 08:01:55 +00:00
case "link":
2016-11-20 14:57:27 +00:00
commands.Link(db, jar, *linkNo, *linkFromHistory)
2016-07-24 08:01:55 +00:00
case "history":
2016-07-25 15:35:09 +00:00
commands.History(db, *historyListItems, 20, *historyListAll)
2016-07-30 08:05:50 +00:00
case "search":
// FIXME: currenlty supports only Google
2016-11-20 14:57:27 +00:00
commands.Get(db, jar, "https://google.com/search?q="+strings.Join(*searchArgs, "+"))
2016-07-22 12:43:23 +00:00
}
}