From 6fc0d183f04f6b2b801b0d08321c8b9872a7d0e9 Mon Sep 17 00:00:00 2001
From: Oleg Atamanenko <oleg.atamanenko@gmail.com>
Date: Tue, 4 Apr 2023 21:34:20 -0700
Subject: [PATCH] Add remote version check

---
 cmd/version.go | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/cmd/version.go b/cmd/version.go
index 6fbe9c07e..a97986543 100644
--- a/cmd/version.go
+++ b/cmd/version.go
@@ -1,9 +1,13 @@
 package cmd
 
 import (
+	"context"
+	"errors"
 	"fmt"
 	"log"
 
+	"github.com/google/go-github/v35/github"
+	"github.com/hashicorp/go-version"
 	"github.com/spf13/afero"
 	"github.com/terraform-linters/tflint/plugin"
 	"github.com/terraform-linters/tflint/tflint"
@@ -12,6 +16,8 @@ import (
 func (cli *CLI) printVersion(opts Options) int {
 	fmt.Fprintf(cli.outStream, "TFLint version %s\n", tflint.Version)
 
+	cli.printLatestReleaseVersion()
+
 	workingDirs, err := findWorkingDirs(opts)
 	if err != nil {
 		cli.formatter.Print(tflint.Issues{}, fmt.Errorf("Failed to find workspaces; %w", err), map[string][]byte{})
@@ -81,3 +87,35 @@ func getPluginVersions(opts Options) []string {
 
 	return versions
 }
+
+// Checks GitHub releases and prints new version, if current version is outdated.
+// requires GitHub releases to follow semver.
+func (cli *CLI) printLatestReleaseVersion() {
+	latest, err := getLatestVersion()
+	if err != nil {
+		cli.formatter.Print(tflint.Issues{}, fmt.Errorf("Failed to check updates; %w", err), map[string][]byte{})
+	}
+	latestVersion, err := version.NewSemver(*latest.Name)
+	compare := tflint.Version.Compare(latestVersion)
+	if compare < 0 {
+		fmt.Fprintf(cli.outStream, "New version available: %s\n", *latest.HTMLURL)
+	}
+}
+
+func getLatestVersion() (*github.RepositoryRelease, error) {
+	ghClient := github.NewClient(nil)
+	releases, _, err := ghClient.Repositories.ListReleases(context.Background(),
+		"terraform-linters", "tflint", &github.ListOptions{})
+	if err != nil {
+		return nil, err
+	}
+
+	// GitHub sorts releases results. Select first non-prerelease version and return it.
+	for i := range releases {
+		release := releases[i]
+		if !*release.Prerelease {
+			return release, nil
+		}
+	}
+	return nil, errors.New("not found")
+}