fetch all project data

vilmibm · vilmibm · commit 824ef3af53f4 · 2020-11-25T14:27:05.000-08:00
diff --git a/api/queries_project.go b/api/queries_project.go
@@ -10,16 +10,12 @@ import (
 	"github.com/cli/cli/internal/ghrepo"
 )
 
-func (c Client) GetProject(baseRepo ghrepo.Interface, projectID string) ([]byte, error) {
-	url := fmt.Sprintf("%sprojects/%s",
-		ghinstance.RESTPrefix(baseRepo.RepoHost()), projectID)
-
+func (c Client) projectREST(url string) ([]byte, error) {
 	req, err := http.NewRequest("GET", url, nil)
 	if err != nil {
 		return nil, err
 	}
 
-	// TODO api() { gh api -H 'accept: application/vnd.github.inertia-preview+json' "$@"; }
 	req.Header.Set("Accept", "application/vnd.github.inertia-preview+json; charset=utf-8")
 
 	resp, err := c.http.Do(req)
@@ -40,3 +36,24 @@ func (c Client) GetProject(baseRepo ghrepo.Interface, projectID string) ([]byte,
 
 	return data, nil
 }
+
+func (c Client) GetProject(baseRepo ghrepo.Interface, projectID int) ([]byte, error) {
+	url := fmt.Sprintf("%sprojects/%d", ghinstance.RESTPrefix(baseRepo.RepoHost()), projectID)
+
+	return c.projectREST(url)
+}
+
+func (c Client) GetProjectColumns(baseRepo ghrepo.Interface, projectID int) ([]byte, error) {
+	url := fmt.Sprintf("%sprojects/%d/columns",
+		ghinstance.RESTPrefix(baseRepo.RepoHost()), projectID)
+
+	return c.projectREST(url)
+
+}
+
+func (c Client) GetProjectCards(baseRepo ghrepo.Interface, columnID int) ([]byte, error) {
+	url := fmt.Sprintf("%sprojects/columns/%d/cards",
+		ghinstance.RESTPrefix(baseRepo.RepoHost()), columnID)
+
+	return c.projectREST(url)
+}
diff --git a/pkg/cmd/project/project.go b/pkg/cmd/project/project.go
@@ -49,7 +49,7 @@ func NewCmdProject(f *cmdutil.Factory, runF func(*ProjectOptions) error) *cobra.
 
 func projectRun(opts *ProjectOptions) error {
 	// TODO interactively ask which project they want since these IDs are not easy to get
-	projectID := "3514315"
+	projectID := 3514315
 
 	c, err := opts.HttpClient()
 	if err != nil {
@@ -69,14 +69,31 @@ func projectRun(opts *ProjectOptions) error {
 
 	fmt.Printf("DBG %#v\n", project)
 
+	for _, c := range project.Columns {
+		fmt.Printf("DBG %s: %d cards\n", c.Name, len(c.Cards))
+	}
+
 	return nil
 }
 
+type Card struct {
+	Note string
+	ID   int
+}
+
+type Column struct {
+	Name  string
+	ID    int
+	Cards []*Card
+}
+
 type Project struct {
-	Name string
+	Name    string
+	ID      int
+	Columns []*Column
 }
 
-func getProject(client *api.Client, baseRepo ghrepo.Interface, projectID string) (*Project, error) {
+func getProject(client *api.Client, baseRepo ghrepo.Interface, projectID int) (*Project, error) {
 	data, err := client.GetProject(baseRepo, projectID)
 	if err != nil {
 		return nil, err
@@ -89,5 +106,27 @@ func getProject(client *api.Client, baseRepo ghrepo.Interface, projectID string)
 		return nil, err
 	}
 
+	data, err = client.GetProjectColumns(baseRepo, projectID)
+	if err != nil {
+		return nil, err
+	}
+
+	err = json.Unmarshal(data, &project.Columns)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, column := range project.Columns {
+		data, err := client.GetProjectCards(baseRepo, column.ID)
+		if err != nil {
+			return nil, err
+		}
+
+		err = json.Unmarshal(data, &column.Cards)
+		if err != nil {
+			return nil, err
+		}
+	}
+
 	return project, nil
 }
