Initial import (#1)

Author: fank

.gitignore

@@ -0,0 +1,19 @@
+# Created by .ignore support plugin (hsz.mobi)
+### Go template
+# Binaries for programs and plugins
+*.exe
+*.dll
+*.so
+*.dylib
+
+# Test binary, build with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Project-local glide cache, RE: https://github.com/Masterminds/glide/issues/736
+.glide/
+
+# IDE
+.idea/

Dockerfile

@@ -0,0 +1,8 @@
+FROM golang:alpine
+
+RUN apk add --no-cache git
+
+ADD . /go/src/github.com/fankserver/docker-factorio-watchdog
+RUN go get github.com/fankserver/docker-factorio-watchdog/... \
+ && go install github.com/fankserver/docker-factorio-watchdog
+ENTRYPOINT /go/bin/docker-factorio-watchdog

docker-factorio-watchdog.go

@@ -0,0 +1,152 @@
+package main
+
+import (
+"fmt"
+"os"
+
+"github.com/blang/semver"
+"github.com/robfig/cron"
+"github.com/sirupsen/logrus"
+)
+
+var latestVersionProceeded map[string]string
+
+func main() {
+latestVersionProceeded = map[string]string{}
+checkVersion()
+c := cron.New()
+c.AddFunc("@every 5m", func() { checkVersion() })
+c.Run()
+}
+
+func checkVersion() {
+version, err := GetAvailableVersions()
+if err != nil {
+logrus.Panic(err)
+}
+
+// Get all available versions
+versions := semver.Versions{}
+for _, version := range version.CoreLinuxHeadless64 {
+if version.To == "" {
+continue
+}
+v, err := semver.Make(version.To)
+if err != nil {
+logrus.Error(err)
+continue
+}
+
+versions = append(versions, v)
+}
+semver.Sort(versions)
+
+// Filter only latest version based on minor
+var lastVersion semver.Version
+firstRun := true
+lastVersions := semver.Versions{}
+for i, version := range versions {
+if firstRun {
+lastVersion = version
+firstRun = false
+}
+
+if lastVersion.Major != version.Major || lastVersion.Minor != version.Minor {
+lastVersions = append(lastVersions, lastVersion)
+} else if i+1 == len(versions) {
+lastVersions = append(lastVersions, version)
+}
+
+lastVersion = version
+}
+
+// Remove all tags which are published on docker
+tags, err := getTags()
+for _, tag := range tags {
+tagVersion, err := semver.Make(tag.Name)
+if err != nil {
+continue
+}
+for i, version := range lastVersions {
+if version.String() == tag.Name {
+lastVersions = append(lastVersions[:i], lastVersions[i+1:]...)
+}
+}
+
+key := fmt.Sprintf("%d.%d", tagVersion.Major, tagVersion.Minor)
+if val, ok := latestVersionProceeded[key]; ok {
+if val == tagVersion.String() {
+logrus.Info("Delete ", tagVersion.String(), " from latest proceeded")
+delete(latestVersionProceeded, key)
+}
+}
+}
+
+for _, version := range lastVersions {
+if version.Major == 0 && version.Minor < 13 {
+continue
+}
+key := fmt.Sprintf("%d.%d", version.Major, version.Minor)
+
+logrus.Info(latestVersionProceeded)
+if val, ok := latestVersionProceeded[key]; ok {
+logrus.Info("OK ", val)
+if val == version.String() {
+logrus.Info("Version exists in cache SKIP")
+continue
+}
+}
+
+updateVersion(version)
+}
+}
+
+func updateVersion(version semver.Version) {
+logrus.Info("Start ", version.String())
+latestVersionProceeded[fmt.Sprintf("%d.%d", version.Major, version.Minor)] = version.String()
+pathRepo := fmt.Sprintf("/tmp/factorio-%s-repo", version)
+
+err := gitCloneRepo(pathRepo)
+if err != nil {
+logrus.Panic(err)
+}
+defer os.RemoveAll(pathRepo)
+logrus.Info("Cloned repo")
+
+err = gitCheckoutBranch(pathRepo, "update-"+version.String())
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Checkout branch ", version)
+
+checksum, err := factorioGetChecksum(fmt.Sprintf("https://www.factorio.com/get-download/%s/headless/linux64", version))
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Got checksum ", checksum)
+
+err = editDockerfile(pathRepo, version, checksum)
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Edited Dockerfile")
+err = editReadme(pathRepo, version)
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Edited README")
+
+err = gitCreateCommit(pathRepo, "update")
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Committed")
+
+err = gitPushBranch(pathRepo, "update-"+version.String())
+if err != nil {
+logrus.Panic(err)
+}
+logrus.Info("Pushed")
+
+createPullRequest(version, "update-"+version.String())
+}

dockerhub.go

@@ -0,0 +1,21 @@
+package main
+
+import "encoding/json"
+
+type imageTag struct {
+Layer string
+Name string
+}
+
+func getTags() ([]imageTag, error) {
+var tags []imageTag
+r, err := myClient.Get("https://index.docker.io/v1/repositories/dtandersen/factorio/tags")
+if err != nil {
+return tags, err
+}
+defer r.Body.Close()
+
+err = json.NewDecoder(r.Body).Decode(&tags)
+
+return tags, err
+}

factorio.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+"crypto/sha1"
+"encoding/hex"
+"encoding/json"
+"io"
+"net/http"
+"time"
+)
+
+func factorioGetChecksum(url string) (string, error) {
+var returnSHA1String string
+
+//Open a new SHA1 hash interface to write to
+hash := sha1.New()
+
+resp, err := http.Get(url)
+if err != nil {
+return returnSHA1String, err
+}
+defer resp.Body.Close()
+
+_, err = io.Copy(hash, resp.Body)
+if err != nil {
+return returnSHA1String, err
+}
+
+//Get the 20 bytes hash
+hashInBytes := hash.Sum(nil)[:20]
+
+//Convert the bytes to a string
+returnSHA1String = hex.EncodeToString(hashInBytes)
+
+return returnSHA1String, nil
+}
+
+var myClient = &http.Client{Timeout: 10 * time.Second}
+
+func GetAvailableVersions() (AvailableVersions, error) {
+var availableVersions AvailableVersions
+r, err := myClient.Get("https://www.factorio.com/updater/get-available-versions?apiVersion=2")
+if err != nil {
+return availableVersions, err
+}
+defer r.Body.Close()
+
+err = json.NewDecoder(r.Body).Decode(&availableVersions)
+
+return availableVersions, err
+}
+
+type AvailableVersions struct {
+CoreLinuxHeadless64 []AvailableVersionsStep `json:"core-linux_headless64"`
+}
+type AvailableVersionsStep struct {
+From string
+To string
+Stable string
+}

git.go

@@ -0,0 +1,95 @@
+package main
+
+import (
+"errors"
+"fmt"
+"io/ioutil"
+"os"
+"os/exec"
+"regexp"
+
+"github.com/blang/semver"
+)
+
+var (
+reDockerfileVersion = regexp.MustCompile(`(VERSION=)(\d+\.\d+\.\d+)`)
+reDockerfileSHA1 = regexp.MustCompile(`(SHA1=)([a-z0-9]+)`)
+)
+
+func gitCloneRepo(path string) error {
+cmd := exec.Command("git", "clone", fmt.Sprintf("https://%s:%s@github.com/%s/%s.git", os.Getenv("GITHUB_USER"), os.Getenv("GITHUB_TOKEN"), githubRepoOwner, githubRepoName), path)
+output, err := cmd.CombinedOutput()
+if err != nil {
+return errors.New(string(output))
+}
+return nil
+}
+
+func gitCheckoutBranch(path string, branch string) error {
+cmd := exec.Command("git", "checkout", "-b", branch)
+cmd.Dir = path
+output, err := cmd.CombinedOutput()
+if err != nil {
+return errors.New(string(output))
+}
+return nil
+}
+
+func gitCreateCommit(path string, commitMessage string) error {
+cmd := exec.Command("git", "commit", "-am", "fk_update")
+cmd.Dir = path
+output, err := cmd.CombinedOutput()
+if err != nil {
+return errors.New(string(output))
+}
+return nil
+}
+
+func gitPushBranch(path string, branch string) error {
+cmd := exec.Command("git", "push", "--set-upstream", "origin", branch)
+cmd.Dir = path
+output, err := cmd.CombinedOutput()
+if err != nil {
+return errors.New(string(output))
+}
+return nil
+}
+
+func editDockerfile(path string, version semver.Version, checksum string) error {
+filename := fmt.Sprintf("%s/%d.%d/Dockerfile", path, version.Major, version.Minor)
+
+file, err := ioutil.ReadFile(filename)
+if err != nil {
+return err
+}
+file = reDockerfileVersion.ReplaceAll(file, []byte("${1}"+version.String()))
+file = reDockerfileSHA1.ReplaceAll(file, []byte("${1}"+checksum))
+
+err = ioutil.WriteFile(filename, file, 0666)
+if err != nil {
+return err
+}
+
+return nil
+}
+
+func editReadme(path string, version semver.Version) error {
+file, err := ioutil.ReadFile(path + "/README.md")
+if err != nil {
+return err
+}
+
+reReadmeVersion, err := regexp.Compile(fmt.Sprintf("(`)%d\\.%d\\.\\d+(`)", version.Major, version.Minor))
+if err != nil {
+return err
+}
+
+file = reReadmeVersion.ReplaceAll(file, []byte("${1}"+version.String()+"${2}"))
+
+err = ioutil.WriteFile(path+"/README.md", file, 0666)
+if err != nil {
+return err
+}
+
+return nil
+}