Dynamically generating a tree

[rchaganti]

Describe the bug When dynamically generating a tree, the child nodes are not aligned properly

Setup Please complete the following information along with version numbers, if applicable.

• OS [Ubuntu]
• Shell [bash]
• Locale [en_US.UTF-8]

To Reproduce Steps to reproduce the behavior: Run the following program.

Source Code

package main

import (
    "fmt"

    "github.com/charmbracelet/lipgloss/tree"
)

type Node struct {
    Name     string
    Category []Category
}

type Category struct {
    Name  string
    Items []Item
}

type Item struct {
    Name string
}

var categories = []Category{
    {
        Name: "cat1",
        Items: []Item{
            {Name: "item1"},
            {Name: "item2"},
        },
    },
    {
        Name: "cat2",
        Items: []Item{
            {Name: "item3"},
            {Name: "item4"},
        },
    },
}

func main() {
    data := Node{
        Name:     "root",
        Category: categories,
    }

    // Create a new tree
    root := tree.Root(data.Name)

    for _, cat := range data.Category {
        cTree := root.Child(
            tree.New().Root(cat.Name),
        )

        for _, item := range cat.Items {
            cTree.Child(item.Name)
        }
    }

    fmt.Println(root)
}

Expected behavior root, child, and leaf aligned per the relationships.

Screenshots

$ go run main.go 
root
├── cat1
├── item1
├── item2
├── cat2
├── item3
└── item4

Additional context Add any other context about the problem here.

[bashbunni]

Hey, the issue here is where you assign the output of root.Child to cTree which makes cTree a pointer to the root tree, not the child.

Here's a working version of what you're looking for:

package main

import (
    "fmt"

    "github.com/charmbracelet/lipgloss/tree"
)

type Node struct {
    Name     string
    Category []Category
}

type Category struct {
    Name  string
    Items []Item
}

type Item struct {
    Name string
}

var categories = []Category{
    {
        Name: "cat1",
        Items: []Item{
            {Name: "item1"},
            {Name: "item2"},
        },
    },
    {
        Name: "cat2",
        Items: []Item{
            {Name: "item3"},
            {Name: "item4"},
        },
    },
}

func main() {
    data := Node{
        Name:     "root",
        Category: categories,
    }

    // Create a new tree
    root := tree.Root(data.Name)

    for _, cat := range data.Category {
        cTree := tree.New().Root(cat.Name)

        for _, item := range cat.Items {
            cTree.Child(item.Name)
        }
        root.Child(cTree)
    }

    fmt.Println(root)
}

Output

 root                                                                                               
 ├── cat1                                                                                           
 │   ├── item1                                                                                      
 │   └── item2                                                                                      
 └── cat2                                                                                           
     ├── item3                                                                                      
     └── item4 

[rchaganti]

Awesome! Thank you.