justwatchcom · glukaza · Sep 28, 2018 · Sep 28, 2018 · Sep 28, 2018 · Sep 28, 2018
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/README.md b/README.md
@@ -22,6 +22,10 @@ To watch repositories simply add them to the list of arguments `-r=kubernetes/ku
 ```
 docker run --rm -e GITHUB_TOKEN=XXX -e SLACK_HOOK=https://hooks.slack.com/... justwatch/github-releases-notifier -r=kubernetes/kubernetes
 ```
+If you need to follow by tag you can use
+```
+docker run --rm -e TAG_CHECKER=true -e GITHUB_TOKEN=XXX -e SLACK_HOOK=https://hooks.slack.com/... justwatch/github-releases-notifier -r=kubernetes/kubernetes
+```
 
 #### docker-compose
 

diff --git a/main.go b/main.go
@@ -21,6 +21,7 @@ type Config struct {
 	LogLevel     string        `arg:"env:LOG_LEVEL"`
 	Repositories []string      `arg:"-r,separate"`
 	SlackHook    string        `arg:"env:SLACK_HOOK"`
+	IsTagChecker bool          `arg:"env:TAG_CHECKER"`
 }
 
 // Token returns an oauth2 token or an error.
@@ -63,7 +64,7 @@ func main() {
 	}
 
 	releases := make(chan Repository)
-	go checker.Run(c.Interval, c.Repositories, releases)
+	go checker.Run(c.Interval, c.Repositories, releases, c.IsTagChecker)
 
 	slack := SlackSender{Hook: c.SlackHook}
 

diff --git a/redis.go b/redis.go
@@ -0,0 +1,50 @@
+package main
+
+import (
+	"fmt"
+	"github.com/go-redis/redis"
+	"os"
+)
+
+//Client redis Client
+var Client *redis.Client
+
+//SetKey set key in redis
+func SetKey(key string, value string) {
+	var err error
+
+	if GetValue(key) != "" {
+		Client.Del(key)
+	}
+
+	err = Client.Set(key, value, 0).Err()
+	check(err)
+}
+
+//GetValue get value from redis
+func GetValue(key string) string {
+	return Client.Get(key).Val()
+}
+
+//СonnectToRedis connect to redis
+func СonnectToRedis() redis.Client {
+	var host = os.Getenv("REDIS_HOST")
+	var port = os.Getenv("REDIS_PORT")
+
+	Client = redis.NewClient(&redis.Options{
+		Addr:     host + ":" + port,
+		Password: "",
+		DB:       11,
+	})
+
+	pong, err := Client.Ping().Result()
+	fmt.Println(pong, err)
+
+	return *Client
+}
+
+func check(e error) {
+	if e != nil {
+		fmt.Println(e)
+	}
+}
diff --git a/releasechecker.go b/releasechecker.go
@@ -20,7 +20,7 @@ type Checker struct {
 }
 
 // Run the queries and comparisons for the given repositories in a given interval.
-func (c *Checker) Run(interval time.Duration, repositories []string, releases chan<- Repository) {
+func (c *Checker) Run(interval time.Duration, repositories []string, releases chan<- Repository, tag bool) {
 	if c.releases == nil {
 		c.releases = make(map[string]Repository)
 	}
@@ -31,6 +31,14 @@ func (c *Checker) Run(interval time.Duration, repositories []string, releases ch
 			owner, name := s[0], s[1]
 
 			nextRepo, err := c.query(owner, name)
+			if true == tag {
+				nextRepo, err = APIV3tagChecker(owner, name)
+				if nextRepo != (Repository{}) {
+					releases <- nextRepo
+					c.releases[repoName] = nextRepo
+				}
+			}
+
 			if err != nil {
 				level.Warn(c.logger).Log(
 					"msg", "failed to query the repository's releases",

diff --git a/tagchecker.go b/tagchecker.go
@@ -0,0 +1,71 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"github.com/google/go-github/github"
+	"golang.org/x/oauth2"
+	"net/url"
+	"os"
+	"time"
+)
+
+//APIV3tagChecker Function for working with github api v3 and check if new tags are published
+func APIV3tagChecker(owner, name string) (Repository, error) {
+	СonnectToRedis()
+	githubToken, ok := os.LookupEnv("GITHUB_TOKEN")
+
+	if !ok {
+		fmt.Errorf("GITHUB_TOKEN environment variable not set")
+	}
+
+	ctx := context.Background()
+	ts := oauth2.StaticTokenSource(
+		&oauth2.Token{AccessToken: githubToken},
+	)
+
+	tc := oauth2.NewClient(ctx, ts)
+	client := github.NewClient(tc)
+
+	opt := &github.ListOptions{}
+
+	repo, _, err := client.Repositories.Get(ctx, owner, name)
+	tags, _, err := client.Repositories.ListTags(ctx, owner, name, opt)
+
+	if err != nil {
+		return Repository{}, fmt.Errorf("cant find repo")
+	}
+
+	var LastTag github.RepositoryTag
+	for _, tag := range tags {
+		LastTag = *tag
+		break
+	}
+
+	repourl, err := url.ParseRequestURI(repo.GetURL())
+	if err != nil {
+		return Repository{}, fmt.Errorf("cant convert url")
+	}
+
+	if GetValue(name) == LastTag.GetName() {
+		return Repository{}, nil
+	}
+
+	SetKey(name, LastTag.GetName())
+
+	return Repository{
+		ID:          string(repo.GetID()),
+		Name:        string(repo.GetName()),
+		Owner:       owner,
+		Description: string(repo.GetDescription()),
+		URL:         *repourl,
+
+		Release: Release{
+			ID:          "1",
+			Name:        LastTag.GetName(),
+			Description: "It is Tag",
+			URL:         *repourl,
+			PublishedAt: time.Date(1, time.January, 1, 0, 0, 0, 0, time.UTC),
+		},
+	}, nil
+}
diff --git a/vendor/github.com/go-redis/redis/.gitignore b/vendor/github.com/go-redis/redis/.gitignore
diff --git a/vendor/github.com/go-redis/redis/.travis.yml b/vendor/github.com/go-redis/redis/.travis.yml
diff --git a/vendor/github.com/go-redis/redis/LICENSE b/vendor/github.com/go-redis/redis/LICENSE
diff --git a/vendor/github.com/go-redis/redis/Makefile b/vendor/github.com/go-redis/redis/Makefile
diff --git a/vendor/github.com/go-redis/redis/README.md b/vendor/github.com/go-redis/redis/README.md