package main import ( "encoding/base64" "encoding/json" "fmt" "io" "log" "net/http" "os" "text/template" "github.com/alexflint/go-arg" ) type NodeData struct { Id string Title string } type Node struct { Children []*Node Data NodeData } func (n *Node) writeIndex(w io.Writer) { tmpl, err := template.New("root").Parse(` {{ define "tree" }} {{ if gt (len .Children) 0 }} {{ end }} {{ end }} {{ template "tree" . }} `) if err != nil { fmt.Println("ERROR creating template") } err = tmpl.Execute(w, n) if err != nil { fmt.Printf("ERROR executing template: %v", err) } } func (n *Node) writeHTMLFiles() { if len(n.Children) > 0 { for _, child := range n.Children { writeHTML(n) child.writeHTMLFiles() } } // leaf node writeHTML(n) } func (n *Node) getChildWithID(id string) (node *Node, hasChild bool) { for _, c := range n.Children { if c.Data.Id == id { return c, true } } empty := Node{} return &empty, false } func (n *Node) addChild(child *Node) { n.Children = append(n.Children, child) } func writeHTML(n *Node) { f, err := os.Create(n.Data.Id + ".html") if err != nil { log.Fatalf("Error creating file for ID %v", n.Data.Id) } defer f.Close() f.Write([]byte(n.Data.Title)) } type spaceResult struct { Results []struct { ID string `json:"id"` Type string `json:"type"` Status string `json:"status"` Title string `json:"title"` Ancestors []struct { ID string `json:"id"` Type string `json:"type"` Status string `json:"status"` Title string `json:"title"` MacroRenderedOutput struct { } `json:"macroRenderedOutput"` Extensions struct { Position int `json:"position"` } `json:"extensions"` Expandable struct { Container string `json:"container"` Metadata string `json:"metadata"` Restrictions string `json:"restrictions"` History string `json:"history"` Body string `json:"body"` Version string `json:"version"` Descendants string `json:"descendants"` Space string `json:"space"` ChildTypes string `json:"childTypes"` Operations string `json:"operations"` SchedulePublishDate string `json:"schedulePublishDate"` Children string `json:"children"` Ancestors string `json:"ancestors"` } `json:"_expandable"` Links struct { Self string `json:"self"` Tinyui string `json:"tinyui"` Editui string `json:"editui"` Webui string `json:"webui"` } `json:"_links"` } `json:"ancestors"` MacroRenderedOutput struct { } `json:"macroRenderedOutput"` Extensions struct { Position int `json:"position"` } `json:"extensions"` Expandable struct { ChildTypes string `json:"childTypes"` Container string `json:"container"` Metadata string `json:"metadata"` Operations string `json:"operations"` SchedulePublishDate string `json:"schedulePublishDate"` Children string `json:"children"` Restrictions string `json:"restrictions"` History string `json:"history"` Body string `json:"body"` Version string `json:"version"` Descendants string `json:"descendants"` Space string `json:"space"` } `json:"_expandable"` Links struct { Self string `json:"self"` Tinyui string `json:"tinyui"` Editui string `json:"editui"` Webui string `json:"webui"` } `json:"_links"` } `json:"results"` Start int `json:"start"` Limit int `json:"limit"` Size int `json:"size"` Links struct { Base string `json:"base"` Context string `json:"context"` Next string `json:"next"` Self string `json:"self"` } `json:"_links"` } func getSpaceChildren(url, user, token string) spaceResult { b64Auth := base64.StdEncoding.EncodeToString([]byte(user + ":" + token)) client := &http.Client{} req, err := http.NewRequest(http.MethodGet, url, nil) if err != nil { log.Fatal(err) } req.Header.Add("Authorization", "Basic "+b64Auth) resp, err := client.Do(req) if err != nil { log.Fatal(err) } if resp.StatusCode != http.StatusOK { log.Fatalf("Error fetching url: %v, status code: %v", url, resp.StatusCode) } body, err := io.ReadAll(resp.Body) if err != nil { log.Fatal(err) } var result spaceResult json.Unmarshal(body, &result) return result } func createPageTree(s *spaceResult) { var parent *Node root := Node{Data: NodeData{Title: "root"}} for _, page := range s.Results { parent = &root // ignore the root node if len(page.Ancestors) == 0 { continue } // create branches from ancestors for i, ancestor := range page.Ancestors { // skip root node if i == 0 { continue } node, hasChild := parent.getChildWithID(ancestor.ID) if hasChild { parent = node continue } n := Node{Data: NodeData{Id: ancestor.ID, Title: ancestor.Title}} parent.addChild(&n) parent = &n } // create leaf node n := Node{Data: NodeData{Id: page.ID, Title: page.Title}} parent.addChild(&n) } fname := "index.html" f, err := os.Create(fname) if err != nil { log.Fatalf("Failed to open %v", fname) } defer f.Close() root.writeIndex(f) root.writeHTMLFiles() } func main() { var args struct { Baseurl string `arg:"required" help:"Base URL of the Confluence instance (required)"` Spacekey string `arg:"required" help:"Spacekey to export (required)"` User string `arg:"required" help:"User used for authentication (required)"` Token string `arg:"required" help:"Token used for authentication (required)"` } arg.MustParse(&args) url := args.Baseurl + "/wiki/rest/api/space/" + args.Spacekey + "/content/page?expand=ancestors&limit=9999" result := getSpaceChildren(url, args.User, args.Token) createPageTree(&result) }