james/yarr
4
0
Fork 0
mirror of https://github.com/nkanaev/yarr.git synced 2025-05-24 00:33:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

reader utility

Browse Source
This commit is contained in:
Nazar Kanaev 2021-03-30 12:23:10 +01:00
parent c958ee9116
commit ac36892150
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
bin/reader.go
View File

@ -5,6 +5,8 @@ import (
"log"
"net/http"
"os"
"io"
"strings"
"github.com/nkanaev/yarr/src/reader"
)
@ -15,13 +17,23 @@ func main() {
return
}
url := os.Args[1]
res, err := http.Get(url)
if err != nil {
log.Fatalf("failed to get url %s: %s", url, err)
}
defer res.Body.Close()
var r io.Reader
content, err := reader.ExtractContent(res.Body)
if strings.HasPrefix(url, "http") {
res, err := http.Get(url)
if err != nil {
log.Fatalf("failed to get url %s: %s", url, err)
}
r = res.Body
} else {
var err error
r, err = os.Open(url)
if err != nil {
log.Fatalf("failed to open file: %s", err)
}
}
content, err := reader.ExtractContent(r)
if err != nil {
log.Fatalf("failed to extract content: %s", err)
}