docs: draw graph

cmd/graph/main.go

@@ -0,0 +1,158 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"time"
+
+	"github.com/shurcooL/githubv4"
+	"github.com/sirupsen/logrus"
+	"golang.org/x/oauth2"
+	"gonum.org/v1/plot"
+	"gonum.org/v1/plot/plotter"
+	"gonum.org/v1/plot/plotutil"
+	"gonum.org/v1/plot/vg"
+	"gopkg.in/yaml.v2"
+)
+
+func main() {
+	if err := core(); err != nil {
+		log.Fatal(err)
+	}
+}
+
+func core() error {
+	// start: 2021-09-02 https://github.com/aquaproj/aqua-registry/commit/199bdfb520dcb27f3338cc0fb01ce525c12c21c7
+	startDate := time.Date(2021, 9, 2, 3, 49, 0, 0, time.UTC)
+
+	ctx := context.Background()
+
+	// list releases
+	src := oauth2.StaticTokenSource(
+		&oauth2.Token{AccessToken: os.Getenv("GITHUB_TOKEN")},
+	)
+	httpClient := oauth2.NewClient(ctx, src)
+	client := githubv4.NewClient(httpClient)
+	logrus.Infof("getting the list of releases")
+
+	releases, err := listReleases(ctx, client)
+	if err != nil {
+		return err
+	}
+
+	logrus.WithField("num_of_releases", len(releases)).Info("got the list of releases")
+
+	pts, err := getPlots(ctx, startDate, releases)
+	if err != nil {
+		return err
+	}
+
+	return draw(pts)
+}
+
+func getPlots(ctx context.Context, startDate time.Time, releases []*Release) (plotter.XYs, error) {
+	pts := make(plotter.XYs, len(releases))
+	for i, release := range releases {
+		duration := release.PublishedAt.Sub(startDate).Hours() / 24
+		name := *release.Name
+		logrus.WithFields(logrus.Fields{
+			"release":      name,
+			"published_at": release.PublishedAt,
+			"duration":     duration,
+		}).Info("getting the number of packages")
+		numOfPkgs, err := getNumOfPackages(ctx, release)
+		if err != nil {
+			return nil, err
+		}
+		logrus.WithFields(logrus.Fields{
+			"release":         name,
+			"published_at":    release.PublishedAt,
+			"num_of_packages": numOfPkgs,
+			"duration":        duration,
+		}).Info("got the number of packages")
+		pts[i].X = duration
+		pts[i].Y = float64(numOfPkgs)
+	}
+	return pts, nil
+}
+
+type Release struct {
+	Name        *githubv4.String
+	PublishedAt *githubv4.DateTime
+}
+
+func listReleases(ctx context.Context, client *githubv4.Client) ([]*Release, error) {
+	var q struct {
+		Repository struct {
+			Releases struct {
+				Nodes    []*Release
+				PageInfo struct {
+					EndCursor   githubv4.String
+					HasNextPage bool
+				}
+			} `graphql:"releases(first: 100, after: $releasesCursor)"`
+		} `graphql:"repository(owner: $repositoryOwner, name: $repositoryName)"`
+	}
+	variables := map[string]interface{}{
+		"repositoryOwner": githubv4.String("aquaproj"),
+		"repositoryName":  githubv4.String("aqua-registry"),
+		"releasesCursor":  (*githubv4.String)(nil), // Null after argument to get first page.
+	}
+
+	var allReleases []*Release
+	for {
+		if err := client.Query(ctx, &q, variables); err != nil {
+			return nil, fmt.Errorf("list releases by GitHub API: %w", err)
+		}
+		allReleases = append(allReleases, q.Repository.Releases.Nodes...)
+		if !q.Repository.Releases.PageInfo.HasNextPage {
+			break
+		}
+		variables["releasesCursor"] = githubv4.NewString(q.Repository.Releases.PageInfo.EndCursor)
+	}
+	return allReleases, nil
+}
+
+func draw(pts plotter.XYs) error {
+	p := plot.New()
+	path := "points.png"
+
+	p.Title.Text = "The transition of the number of packages"
+	p.X.Label.Text = "Date and aqua-registry version"
+	p.Y.Label.Text = "The number of packages"
+
+	if err := plotutil.AddLinePoints(p, pts); err != nil {
+		return fmt.Errorf("add a line points to the plot: %w", err)
+	}
+
+	logrus.Info("saving the plot to a PNG file")
+	if err := p.Save(4*vg.Inch, 4*vg.Inch, path); err != nil {
+		return fmt.Errorf("save the plot to a PNG file: %w", err)
+	}
+	return nil
+}
+
+type Registry struct {
+	Packages []struct{}
+}
+
+func getNumOfPackages(ctx context.Context, release *Release) (int, error) {
+	u := fmt.Sprintf("https://raw.githubusercontent.com/aquaproj/aqua-registry/%s/registry.yaml", *release.Name)
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
+	if err != nil {
+		return 0, nil
+	}
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return 0, nil
+	}
+	defer resp.Body.Close()
+	registry := &Registry{}
+	if err := yaml.NewDecoder(resp.Body).Decode(registry); err != nil {
+		return 0, err
+	}
+	return len(registry.Packages), nil
+}

go.mod

@@ -0,0 +1,28 @@
+module github.com/aquaproj/aqua-registry
+
+go 1.18
+
+require (
+	github.com/shurcooL/githubv4 v0.0.0-20220520033151-0b4e3294ff00
+	github.com/sirupsen/logrus v1.8.1
+	golang.org/x/oauth2 v0.0.0-20220622183110-fd043fe589d2
+	gonum.org/v1/plot v0.11.0
+	gopkg.in/yaml.v2 v2.2.2
+)
+
+require (
+	git.sr.ht/~sbinet/gg v0.3.1 // indirect
+	github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b // indirect
+	github.com/go-fonts/liberation v0.2.0 // indirect
+	github.com/go-latex/latex v0.0.0-20210823091927-c0d11ff05a81 // indirect
+	github.com/go-pdf/fpdf v0.6.0 // indirect
+	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
+	github.com/golang/protobuf v1.4.2 // indirect
+	github.com/shurcooL/graphql v0.0.0-20220606043923-3cf50f8a0a29 // indirect
+	golang.org/x/image v0.0.0-20220302094943-723b81ca9867 // indirect
+	golang.org/x/net v0.0.0-20220127200216-cd36cc0744dd // indirect
+	golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e // indirect
+	golang.org/x/text v0.3.7 // indirect
+	google.golang.org/appengine v1.6.6 // indirect
+	google.golang.org/protobuf v1.25.0 // indirect
+)