Claude
21a0ae09e2
All checks were successful
Integration Tests / test (pull_request) Successful in 30s
107 lines
2.6 KiB
Go
107 lines
2.6 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/runescape-wiki/rsw/internal/htmlconv"
|
|
"github.com/runescape-wiki/rsw/internal/render"
|
|
"github.com/runescape-wiki/rsw/internal/wiki"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func newPageCmd() *cobra.Command {
|
|
var pageSection string
|
|
var complete bool
|
|
|
|
cmd := &cobra.Command{
|
|
Use: "page <title>",
|
|
Short: "Fetch and display a wiki page",
|
|
Long: `Fetches a wiki page and renders its table of contents by default.
|
|
Use --section to get a specific section, or --complete for the full page.
|
|
|
|
Examples:
|
|
rsw osrs page "Dragon scimitar"
|
|
rsw osrs page "Dragon scimitar" --section "Combat styles"
|
|
rsw osrs page "Mining" --complete
|
|
rsw rs3 page "Mining" --section "Training"`,
|
|
Args: cobra.MinimumNArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
title := args[0]
|
|
client := wiki.NewClient(GameBaseURL())
|
|
|
|
if Raw() {
|
|
page, err := client.GetPage(title)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to fetch page: %w", err)
|
|
}
|
|
fmt.Println(page.Wikitext)
|
|
return nil
|
|
}
|
|
|
|
page, err := client.GetPageHTML(title)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to fetch page: %w", err)
|
|
}
|
|
|
|
if pageSection != "" {
|
|
body := htmlconv.ExtractSection(page.HTML, pageSection)
|
|
if body == "" {
|
|
sections := htmlconv.ListSections(page.HTML)
|
|
return fmt.Errorf("section %q not found. Available sections: %s",
|
|
pageSection, formatSectionNames(sections))
|
|
}
|
|
md := render.New()
|
|
md.H1(page.Title)
|
|
md.Line(body)
|
|
fmt.Print(md.String())
|
|
return nil
|
|
}
|
|
|
|
md := render.New()
|
|
md.H1(page.Title)
|
|
|
|
sections := htmlconv.ListSections(page.HTML)
|
|
if len(sections) > 0 {
|
|
md.H2("Sections")
|
|
for _, s := range sections {
|
|
indent := ""
|
|
if s.Level == 3 {
|
|
indent = " "
|
|
} else if s.Level >= 4 {
|
|
indent = " "
|
|
}
|
|
md.Line(fmt.Sprintf("%s- %s", indent, s.Name))
|
|
}
|
|
md.Newline()
|
|
md.HR()
|
|
}
|
|
|
|
if complete {
|
|
md.Line(htmlconv.Convert(page.HTML))
|
|
} else {
|
|
md.Line("Use --section <name> to fetch a specific section, or --complete for the full page.")
|
|
}
|
|
|
|
fmt.Print(md.String())
|
|
return nil
|
|
},
|
|
}
|
|
|
|
cmd.Flags().StringVar(&pageSection, "section", "", "Fetch only the named section")
|
|
cmd.Flags().BoolVar(&complete, "complete", false, "Fetch the complete page content (default shows TOC only)")
|
|
return cmd
|
|
}
|
|
|
|
func formatSectionNames(sections []htmlconv.SectionInfo) string {
|
|
names := make([]string, len(sections))
|
|
for i, s := range sections {
|
|
names[i] = s.Name
|
|
}
|
|
return strings.Join(names, ", ")
|
|
}
|
|
|
|
func init() {
|
|
RegisterCommand(newPageCmd)
|
|
}
|