mirror of
https://github.com/NohamR/papeer.git
synced 2026-05-25 12:27:20 +00:00
49 lines
901 B
Go
49 lines
901 B
Go
package cmd
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
urllib "net/url"
|
|
"os"
|
|
|
|
"github.com/jedib0t/go-pretty/v6/table"
|
|
cobra "github.com/spf13/cobra"
|
|
|
|
"github.com/lapwat/papeer/book"
|
|
)
|
|
|
|
var listCmd = &cobra.Command{
|
|
Use: "ls",
|
|
Short: "Print table of content",
|
|
Args: func(cmd *cobra.Command, args []string) error {
|
|
if len(args) < 1 {
|
|
return errors.New("requires an URL argument")
|
|
}
|
|
return nil
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
base, err := urllib.Parse(args[0])
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
links := book.GetLinks(base, selector)
|
|
|
|
t := table.NewWriter()
|
|
t.SetOutputMirror(os.Stdout)
|
|
t.AppendHeader(table.Row{"#", "Name", "Url", "Class"})
|
|
|
|
for index, link := range links {
|
|
u, err := base.Parse(link.Href())
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
t.AppendRow([]interface{}{index + 1, link.Text(), u.String(), link.Class()})
|
|
}
|
|
|
|
t.Render()
|
|
|
|
},
|
|
}
|