first commit

Author: Boris

.github/workflows/docker.yml

@@ -0,0 +1,52 @@
+name: Docker Image
+
+on:
+  push:
+    tags:
+      - '**'
+
+jobs:
+
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Docker meta
+      id: meta
+      uses: docker/metadata-action@v5
+      with:
+        images: ghcr.io/${{ github.repository }}
+        tags: |
+          type=schedule
+          type=ref,event=branch
+          type=ref,event=pr
+          type=semver,pattern={{version}}
+          type=semver,pattern={{major}}.{{minor}}
+          type=semver,pattern={{major}}
+          type=sha
+
+    - name: Set up Docker Buildx
+      id: buildx
+      uses: docker/setup-buildx-action@v3
+
+    - name: Login to Github Packages
+      uses: docker/login-action@v3
+      with:
+        registry: ghcr.io
+        username: ${{ github.repository_owner }}
+        password: ${{ secrets.GITHUB_TOKEN }}
+
+    - name: Set build date
+      run: echo "BUILD_DATE=$(date +%Y%m%d%H%M)" >> $GITHUB_ENV
+
+    - name: Build and push
+      uses: docker/build-push-action@v5
+      with:
+        context: .
+        build-args: |
+          versionflags=-X 'main.appVersion=${{ github.ref_name }}'
+        push: ${{ github.event_name != 'pull_request' }}
+        tags: ${{ steps.meta.outputs.tags }}
+        labels: ${{ steps.meta.outputs.labels }}

.github/workflows/go.yml

@@ -0,0 +1,38 @@
+name: Go
+
+on:
+  push:
+    tags:
+      - '**'
+
+jobs:
+
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Go
+      uses: actions/setup-go@v5
+      with:
+        go-version: 1.25
+
+    - name: Build
+      run: |
+        versionflags="-X main.appVersion=${{ github.ref_name }}"
+        for GOOS in linux; do
+          for GOARCH in amd64 arm64; do
+            export GOOS GOARCH
+            CGO_ENABLED=0 go build -v -a -tags netgo -ldflags="-extldflags '-static' -s -w $versionflags" -o build/osctl-${GOOS}-${GOARCH} cmd/main.go
+          done
+        done
+        cd build; sha256sum * > sha256sums.txt
+
+    - name: Release
+      uses: softprops/action-gh-release@v1
+      with:
+        generate_release_notes: true
+        token: ${{ secrets.GITHUB_TOKEN }}
+        files: |
+          build/*

.gitignore

@@ -0,0 +1,2 @@
+vendor/
+build/

Makefile

@@ -0,0 +1,34 @@
+GOPATH=$(shell pwd)/vendor:$(shell pwd)
+GOBIN=$(shell pwd)/build/
+GOFILES=./cmd/$(wildcard *.go)
+GONAME=$(shell basename "$(PWD)")
+PID=/tmp/go-$(GONAME).pid
+GOOS=linux
+#GOARCH=386
+BUILD=`date +%FT%T%z`
+MKDIR_P = mkdir -p
+
+all :: build
+
+get:
+	@GOPATH=$(GOPATH) GOBIN=$(GOBIN) go get -d $(GOFILES)
+
+build:
+	@echo "Building $(GOFILES) to ./build"
+	cd ../
+#	go mod vendor
+	@GOPATH=$(GOPATH) GOBIN=$(GOBIN) GOOS=$(GOOS) GOARCH=$(GOARCH) go build -mod=mod -ldflags "-s -w -X main.vBuild=${BUILD}" -o build/$(GONAME) $(GOFILES)
+	strip ./build/$(GONAME)
+
+run:
+	@GOPATH=$(GOPATH) GOBIN=$(GOBIN) go run $(GOFILES)
+
+clear:
+	@clear
+
+clean:
+	@echo "Cleaning"
+	@GOPATH=$(GOPATH) GOBIN=$(GOBIN) go clean
+
+.PHONY:	build get install run clean dirs
+

cmd/main.go

@@ -0,0 +1,33 @@
+package main
+
+import (
+	"flag"
+	"log"
+
+	"waitgreen/modules/config"
+	"waitgreen/modules/wg"
+)
+
+var (
+	configfile string
+	vBuild     string
+	cnf        config.Config
+)
+
+func init() {
+	flag.StringVar(&configfile, "config", "main.yml", "Read configuration from this file")
+	flag.StringVar(&configfile, "f", "main.yml", "Read configuration from this file")
+	flag.Parse()
+
+	log.SetFlags(log.LstdFlags | log.Lshortfile)
+
+	log.Println("Bootstrap: build num.", vBuild)
+
+	cnf = config.Parse(configfile)
+	log.Println("Bootstrap: successful parsing config file.")
+
+}
+
+func main() {
+	wg.Run(cnf)
+}

example/main.yml

@@ -0,0 +1,5 @@
+app:
+  port: 9900
+  bind: 0.0.0.0
+  timeout: 60
+  waitgreen: true

go.mod

@@ -0,0 +1,8 @@
+module waitgreen
+
+go 1.23.2
+
+require (
+	github.com/uzhinskiy/lib.go v0.1.7
+	gopkg.in/yaml.v2 v2.4.0
+)

go.sum

@@ -0,0 +1,6 @@
+github.com/uzhinskiy/lib.go v0.1.7 h1:up/fZOtkn0OaE2QsG0gPVsSXOGUiBwwcf1McWenOYvM=
+github.com/uzhinskiy/lib.go v0.1.7/go.mod h1:JolhUn+z8ET3PxRuHx2fMJYZEPR2nc3PE1Hu9MvAHls=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=

modules/config/config.go

@@ -0,0 +1,60 @@
+// Copyright © 2020 Uzhinskiy Boris
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package config
+
+import (
+	"io/ioutil"
+	"log"
+
+	"gopkg.in/yaml.v2"
+)
+
+type Config struct {
+	App struct {
+		Port       string `yaml:"port"`
+		Bind       string `yaml:"bind"`
+		TimeOut    int    `yaml:"-"`
+		TimeOutRaw *int   `yaml:"timeout"`
+		DefaultWG  bool   `yaml:"waitgreen"`
+	} `yaml:"app"`
+}
+
+func Parse(f string) Config {
+	var c Config
+
+	yamlBytes, err := ioutil.ReadFile(f)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	err = yaml.Unmarshal(yamlBytes, &c)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	if c.App.Port == "" {
+		c.App.Port = "9900"
+	}
+
+	if c.App.Bind == "" {
+		c.App.Bind = "0.0.0.0"
+	}
+
+	c.App.TimeOut = 30
+	if c.App.TimeOutRaw != nil {
+		c.App.TimeOut = *c.App.TimeOutRaw
+	}
+
+	return c
+}

modules/wg/wg.go

@@ -0,0 +1,81 @@
+package wg
+
+import (
+	"encoding/json"
+
+	"log"
+	"net/http"
+
+	"waitgreen/modules/config"
+
+	"github.com/uzhinskiy/lib.go/helpers"
+)
+
+type WaitGreen struct {
+	conf config.Config
+}
+
+type apiRequest struct {
+	WaitGreen bool `json:"waitgreen"`
+}
+
+var wgEnabled bool
+
+func Run(cnf config.Config) {
+	wg := WaitGreen{}
+	wg.conf = cnf
+
+	wgEnabled = cnf.App.DefaultWG
+
+	http.HandleFunc("/", wg.ApiHandler)
+	http.ListenAndServe(cnf.App.Bind+":"+cnf.App.Port, nil)
+}
+
+func (wg *WaitGreen) ApiHandler(w http.ResponseWriter, r *http.Request) {
+	var request apiRequest
+
+	defer r.Body.Close()
+	remoteIP := helpers.GetIP(r.RemoteAddr, r.Header.Get("X-Real-IP"), r.Header.Get("X-Forwarded-For"))
+
+	w.Header().Add("Access-Control-Allow-Origin", "*")
+	w.Header().Add("Access-Control-Allow-Methods", "POST,OPTIONS,GET")
+	w.Header().Add("Access-Control-Allow-Credentials", "true")
+	w.Header().Add("Access-Control-Allow-Headers", "Accept, Content-Type, Content-Length, Accept-Encoding, X-CSRF-Token, Authorization")
+	w.Header().Add("Content-Type", "application/json; charset=utf-8")
+	w.Header().Set("X-Server", "wg")
+
+	if r.Method == "OPTIONS" {
+		return
+	}
+
+	switch r.Method {
+	case http.MethodPost:
+		{
+			err := json.NewDecoder(r.Body).Decode(&request)
+			if err != nil {
+				http.Error(w, err.Error(), http.StatusInternalServerError)
+				log.Println(remoteIP, "\t", r.Method, "\t", r.URL.Path, "\t", http.StatusInternalServerError, "\t", err.Error())
+				return
+			}
+
+			wgEnabled = request.WaitGreen
+			resp := map[string]interface{}{
+				"status": "Ok",
+			}
+			j, _ := json.Marshal(resp)
+			log.Println(remoteIP, "\t", r.Method, "\t", r.URL.Path, "\t", "200", "\t", r.UserAgent())
+			w.Write(j)
+
+		}
+	case http.MethodGet:
+		{
+			resp := map[string]interface{}{
+				"waitgreen": wgEnabled,
+			}
+			j, _ := json.Marshal(resp)
+			log.Println(remoteIP, "\t", r.Method, "\t", r.URL.Path, "\t", "200", "\t", r.UserAgent())
+			w.Write(j)
+		}
+	}
+
+}