57 lines
748 B
Go
57 lines
748 B
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
|
|
"github.com/prologic/go-gopher"
|
|
)
|
|
|
|
var (
|
|
json = flag.Bool("json", false, "display gopher directory as JSON")
|
|
)
|
|
|
|
func main() {
|
|
var uri string
|
|
|
|
flag.Parse()
|
|
|
|
if len(flag.Args()) == 1 {
|
|
uri = flag.Arg(0)
|
|
} else {
|
|
uri = "gopher://gopher.floodgap.com/"
|
|
}
|
|
|
|
res, err := gopher.Get(uri)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if res.Body != nil {
|
|
contents, err := ioutil.ReadAll(res.Body)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Print(string(contents))
|
|
} else {
|
|
var (
|
|
bytes []byte
|
|
err error
|
|
)
|
|
|
|
if *json {
|
|
bytes, err = res.Dir.ToJSON()
|
|
} else {
|
|
bytes, err = res.Dir.ToText()
|
|
}
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
fmt.Println(string(bytes))
|
|
}
|
|
}
|