From 70601daac0909ee2256b3e7c15ae7963ea13091a Mon Sep 17 00:00:00 2001 From: Jacob Gunther Date: Sat, 30 Jul 2022 22:13:16 -0500 Subject: [PATCH] Add initial code --- .gitignore | 3 + LICENSE | 21 +++ Makefile | 16 +++ README.md | 32 +++++ config.example.yml | 7 + go.mod | 33 +++++ go.sum | 73 ++++++++++ src/config.go | 28 ++++ src/default-icon.png | Bin 0 -> 11083 bytes src/main.go | 125 ++++++++++++++++++ src/redis.go | 155 ++++++++++++++++++++++ src/routes.go | 308 +++++++++++++++++++++++++++++++++++++++++++ src/status.go | 251 +++++++++++++++++++++++++++++++++++ src/util.go | 161 ++++++++++++++++++++++ 14 files changed, 1213 insertions(+) create mode 100644 .gitignore create mode 100644 LICENSE create mode 100644 Makefile create mode 100644 README.md create mode 100644 config.example.yml create mode 100644 go.mod create mode 100644 go.sum create mode 100644 src/config.go create mode 100644 src/default-icon.png create mode 100644 src/main.go create mode 100644 src/redis.go create mode 100644 src/routes.go create mode 100644 src/status.go create mode 100644 src/util.go diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..85cb51e --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +bin/ +config.yml +.DS_Store \ No newline at end of file diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..cf1682c --- /dev/null +++ b/LICENSE @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2022 Jacob Gunther + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. \ No newline at end of file diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..2aa50e2 --- /dev/null +++ b/Makefile @@ -0,0 +1,16 @@ +build: + go build -o bin/main src/*.go + +build-linux: + GOOS=linux go build -o bin/main src/*.go + +build-windows: + GOOS=windows go build -o bin/main src/*.go + +run: + go run src/*.go + +flush-cache: + go run src/*.go --flush-cache + +all: build \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..eb7211f --- /dev/null +++ b/README.md @@ -0,0 +1,32 @@ +# API Server +The REST server that powers the API for mcstatus.io. This repository is open source to allow developers to run their own Minecraft server status API server. + +## Requirements + +- [Go](https://go.dev/) +- [Redis](https://redis.io/) +- [GNU Make](https://www.gnu.org/software/make/) (*optional*) + +## Installation + +1. Clone the repository to a folder + - `git clone https://github.com/mcstatus-io/api-server.git` +2. Move the working directory into the folder + - `cd api-server` +3. Install all required dependencies + - `go get ...` +4. Build the executable + - Using GNU make + - `make build` + - Without GNU make + - `go build -o .\bin\main.exe .\src\*.go` (Windows) + - `go build -o bin/main src/*.go` (Unix) +5. Copy the `config.example.yml` file to `config.yml` and edit the details +6. Start the API server + - `.\bin\main.exe` (Windows) + - `bin/main` (Unix) + +## License + +[MIT License](https://github.com/mcstatus-io/api-server/blob/main/LICENSE) + \ No newline at end of file diff --git a/config.example.yml b/config.example.yml new file mode 100644 index 0000000..87bab60 --- /dev/null +++ b/config.example.yml @@ -0,0 +1,7 @@ +environment: development +host: 127.0.0.1 +port: 3001 +redis: redis://127.0.0.1:6379/0 +cache_enable: false +status_cache_ttl: 5m +favicon_cache_ttl: 24h \ No newline at end of file diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..ec469f6 --- /dev/null +++ b/go.mod @@ -0,0 +1,33 @@ +module main + +go 1.18 + +require ( + github.com/PassTheMayo/mcstatus/v4 v4.0.0 + github.com/go-yaml/yaml v2.1.0+incompatible +) + +require ( + github.com/buaazp/fasthttprouter v0.1.1 + gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect +) + +require ( + github.com/andybalholm/brotli v1.0.4 // indirect + github.com/cespare/xxhash/v2 v2.1.2 // indirect + github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect + github.com/go-redis/redis/v8 v8.11.5 + github.com/klauspost/compress v1.15.6 // indirect + github.com/valyala/bytebufferpool v1.0.0 // indirect + github.com/valyala/fasthttp v1.37.0 +) + +require ( + github.com/fatih/color v1.13.0 // indirect + github.com/kr/pretty v0.3.0 // indirect + github.com/mattn/go-colorable v0.1.9 // indirect + github.com/mattn/go-isatty v0.0.14 // indirect + github.com/rogpeppe/go-internal v1.8.1 // indirect + golang.org/x/net v0.0.0-20220622184535-263ec571b305 // indirect + golang.org/x/sys v0.0.0-20220622161953-175b2fd9d664 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..85c4088 --- /dev/null +++ b/go.sum @@ -0,0 +1,73 @@ +github.com/PassTheMayo/mcstatus/v4 v4.0.0 h1:SgIAwpLkH2Vt7KDZZ7tAsAufyI9XRy9YdgMUJVAw1o0= +github.com/PassTheMayo/mcstatus/v4 v4.0.0/go.mod h1:62Iagw0yGRyufsvEqPI4T8fJpIf6UUh14junH8ZR2LY= +github.com/andybalholm/brotli v1.0.4 h1:V7DdXeJtZscaqfNuAdSRuRFzuiKlHSC/Zh3zl9qY3JY= +github.com/andybalholm/brotli v1.0.4/go.mod h1:fO7iG3H7G2nSZ7m0zPUDn85XEX2GTukHGRSepvi9Eig= +github.com/buaazp/fasthttprouter v0.1.1 h1:4oAnN0C3xZjylvZJdP35cxfclyn4TYkW6Y+DSvS+h8Q= +github.com/buaazp/fasthttprouter v0.1.1/go.mod h1:h/Ap5oRVLeItGKTVBb+heQPks+HdIUtGmI4H5WCYijM= +github.com/cespare/xxhash/v2 v2.1.2 h1:YRXhKfTDauu4ajMg1TPgFO5jnlC2HCbmLXMcTG5cbYE= +github.com/cespare/xxhash/v2 v2.1.2/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w= +github.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk= +github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4= +github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI= +github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= +github.com/go-yaml/yaml v2.1.0+incompatible h1:RYi2hDdss1u4YE7GwixGzWwVo47T8UQwnTLB6vQiq+o= +github.com/go-yaml/yaml v2.1.0+incompatible/go.mod h1:w2MrLa16VYP0jy6N7M5kHaCkaLENm+P+Tv+MfurjSw0= +github.com/klauspost/compress v1.15.0/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk= +github.com/klauspost/compress v1.15.6 h1:6D9PcO8QWu0JyaQ2zUMmu16T1T+zjjEpP91guRsvDfY= +github.com/klauspost/compress v1.15.6/go.mod h1:PhcZ0MbTNciWF3rruxRgKxI5NkcHHrHUDtV4Yw2GlzU= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= +github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0= +github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/mattn/go-colorable v0.1.9 h1:sqDoxXbdeALODt0DAeJCVp38ps9ZogZEAXjus69YV3U= +github.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc= +github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= +github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y= +github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= +github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= +github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE= +github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE= +github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA= +github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc= +github.com/rogpeppe/go-internal v1.8.1 h1:geMPLpDpQOgVyCg5z5GoRwLHepNdb71NXb67XFkP+Eg= +github.com/rogpeppe/go-internal v1.8.1/go.mod h1:JeRgkft04UBgHMgCIwADu4Pn6Mtm5d4nPKWu0nJ5d+o= +github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw= +github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= +github.com/valyala/fasthttp v1.37.0 h1:7WHCyI7EAkQMVmrfBhWTCOaeROb1aCBiTopx63LkMbE= +github.com/valyala/fasthttp v1.37.0/go.mod h1:t/G+3rLek+CyY9bnIE+YlMRddxVAAGjhxndDB4i4C0I= +github.com/valyala/tcplisten v1.0.0/go.mod h1:T0xQ8SeCZGxckz9qRXTfG43PvQ/mcWh7FwZEA7Ioqkc= +golang.org/x/crypto v0.0.0-20220214200702-86341886e292/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4= +golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.0.0-20220225172249-27dd8689420f/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk= +golang.org/x/net v0.0.0-20220622184535-263ec571b305 h1:dAgbJ2SP4jD6XYfMNLVj0BF21jo2PjChrtGaAvF5M3I= +golang.org/x/net v0.0.0-20220622184535-263ec571b305/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220227234510-4e6760a101f9/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220622161953-175b2fd9d664 h1:wEZYwx+kK+KlZ0hpvP2Ls1Xr4+RWnlzGFwPP0aiDjIU= +golang.org/x/sys v0.0.0-20220622161953-175b2fd9d664/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= +gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= +gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= diff --git a/src/config.go b/src/config.go new file mode 100644 index 0000000..143ae6b --- /dev/null +++ b/src/config.go @@ -0,0 +1,28 @@ +package main + +import ( + "io/ioutil" + "time" + + "github.com/go-yaml/yaml" +) + +type Configuration struct { + Environment string `yaml:"environment"` + Host string `yaml:"host"` + Port uint16 `yaml:"port"` + Redis string `yaml:"redis"` + CacheEnable bool `yaml:"cache_enable"` + StatusCacheTTL time.Duration `yaml:"status_cache_ttl"` + FaviconCacheTTL time.Duration `yaml:"favicon_cache_ttl"` +} + +func (c *Configuration) ReadFile(path string) error { + data, err := ioutil.ReadFile(path) + + if err != nil { + return err + } + + return yaml.Unmarshal(data, c) +} diff --git a/src/default-icon.png b/src/default-icon.png new file mode 100644 index 0000000000000000000000000000000000000000..8da22cba881a19128543fdde9e46f99221eaaa57 GIT binary patch literal 11083 zcmb7KWm6nXv&Eg@4nczkf;$Aa;O@@i&IVmHI3c(!4#5@=F2UV>ac6N0ZsE;Sb^pNq zaHgxKs%vI?db;cMbhNsvJT?Y71{@q5wxWWJ=06$yAJ9<#jeX)4WB&w_t)z-199-QG z%oht}I5;*0MHxwLpOv#5TO1pkg$SI!x)3UjT9P>IL%8=lzWu%L5_o%@3cAznUha_G>w0^=z%2N0f&=TSCg|1i z0#%R=_!xp1>s$kxdVV~}>w4U!e)F*=s?BS&6)*6;dS3b3gG}P|-ghri3{jMz`Rr@E z@89dZu8uFKeRxl9tC(gFy|%=NYu=lnEZ4y6tSs-Y=k+laOkIxFy`+60hqIu&>7S(Z z>}er&M9v;*9I=}lKyc8@E_K%C-@E3lsi42tm@{1u1D`n1?&^l#8}%=(Zrn9_#=~bs zCfAwxY=>t>QV06Lhp3651oH$v^ijJ6{!{*h1y^TqwI;fAF$-eK{#u+8-S2T7TM(O0p*Pf6zp|g;Bqld!AG^b5!_WseE?1 zj``c@|45Y_HPzp;eDL3mBwfPUA${y6EerSV*rl?iqOWCcIUvyZFisn*G`Pc4!vjIk z-^L~15mE1_`Ub1dpO(C)_@>J$`mhxi`{P-l>u~}W!&a@&(7%lFr^T#zcZE^%i#7UN z!vGv^GCbyA8gnrdpV}7tVnjBR=L`}o2HUrPZ^eWZW=Cw&}8z)GA5W{LGy=6bmDDB!J^^&6dR{MeN2EB~q;DihvA#+%-^G|AQbYv?NN`k*{ zWcQapJL=fgSZBuZ`{CL?GqI)(FF&k(wn9+THsMmXkl5c9S=IP0RPYBq&E@f@n*~1N zktrnKRJWfS)}sE!tR8#hXge&NPz(H%JW*lxdOQ*P&oQ)MwZaDR1x)4X6fP~>TbY({ zH^Rvg8vqZkD;j>82)ua^Zo~+nksCb#CJ-M2zk~`1UNT3#3sV$-UY>F}?|eSDX{=YW zq05Z*9_)7SdfETvz1vGW@OYW8gB7%mX{O)JG@O0X>qa@%>T7$NO(LJV0|t?@eY(Wx zCQ$_RGR6TRxPRwz;YZP0{J2n0#)I)_LK8nRV%AN(1$3OzaGkqMC|ws70pGQ&}z7Jz1~ge=$SOq-+}wS zJ6>z{=3w9jX=AySX!TA{DbvVEfBf_GFL*VunFq5xU?)UwgmXerfn797Q_6)E6fjB4 z%be8FQNm&Ok5AO@)luHa^<5QN7eBH_$Mi2h5Ay~f|1_Y7J}@ln1vq9P5pD#lSE;FLE-mJ#S#0B>in;2xB)ToJv5Cd&k&ETs2r& zeup2&2b`hn9CvA_j7U$G<3vWz53s7ZK@|S((UyZ;|F7rH!5{8#BLq6OzflSg5SeP> zR1j+p?L6Dt!ZFHcQPIVA%Fe#v)KTTqR--PrG__-$AA8+-@qP|OI1bk~uJ>xxia{GKQQ zW@}QTQoK&M_GtE36!YE>oJi78hj-D(M13oh1c&8N$H8*{WwRY=_4p9=N-uFwj#JNg z$s14h>Y^3g?-YT6U4fdA4cN*HBgf3?o`ATgX#c)IF>0L0L5?>El3b%(;Ltmb#F)IN z`BMJ>+@9os6q?X&GQq@3H7SVJ?;LZu7eRw0oJT@nYus(Eu9ca9ktR5Hq)eVXE&?l{ z%f3L#K+@wxc$2n{WQ{s%e|CgVGY!AYK zmo&Y>lwreqRpm`7?-in^0$@B@KR~Beu~w&E-?G$15J@)|u^`wJkkm4iNUszH+-OKH z%tyAFP4>0wSbc|crp9N0EzR*#ge@<`kSn#5}-hwVRC~{x{OX1}y-! zB?8^opDLs*U$dq}AFU)2KAwco&!yNg$kFbh!3bW@40QX8E<-&-5;Q_%FoGZ9UiV8s zCqTexIK3q2n}kYrOW&mZ8b#p_d87y2HSVgh(7mxTRG|+%Ll+ngO9DB-)-%Fp8QT4H08fh z-+KCM?5lpT|4cUTAN!9|(dtB{kE^KxA9Z`ROr}dy;q_Gn{q&dDM4!~nneiSf4T@Xs z@?|;8+I1WA(aT%(URY#eCzA726Z;FUAetys>)S>kCD-SoD~S6aA2+1x}4s@GjD(JYr2^6Q6m z5NHiTo*s#Qs{$m8D&)8-3FJZ5&QouexPPG|{eheXmgK$LF~k-5*Fw7J1xnkW(I&<6 z4a=;2#*8+8Jr2u@|JoD9%|$^fQexnRO0jb@X2iCk_SpXZ{u-c3mOl`o@~dn;fV(Nn zwbK_~Cf`M`{2f(ddr8hD5eN~0ID*9f^{}7RbU~Ww(8-1N?c_3~794o|wcY+I@4!+t zBHKnBZRr6%o|;k&#v7&K_Czxx{<4kpNs%d=L{T9>YIFJG>7?e3n#;78y2j@-2a$SY zXkpdPy^L}9S}mQ%e9fD&dXZz}IFnSGB;l1GhU^Qj(oc4tK0)r}NeXu4x)7@w@9y7p zlZl!EsHvWn;ougJ55Ll|o=2FKSSJ*J_WYOKZGcw%$27AS78-G@GrCV;UO`jzx~k_} zHy7D@FV%T;3nUd2eIpajQsJ^Ay2H6-eWa>>P4ZzvFxcFXMT-ZhE-b~ZtF!2zcdkP?#%YNso3QO7+4e{1*K-BUzk zJuIwmIO{8?LvWT{9>fUQv(1$a-q;-l)DZ%hW2W`GvrDcI3KKJyAM5i^Ce!9|QDFDt zAL?%5MC}KB$jT^WK~>%Qd?VUbah!m))`ivc_%UQRyCgcZfi46#=Ksg9^C>8EOX1V~ za$!0d&9{W54-cY;KKwGw$wdu#hhzyFm24|~zuTlzalr~Y2FjI(s}JG_oa%dnqp2gv zAOtDB4k)yXCSx9HVU{!Tr$nl!N8T192n)dN>lzhAjHGN?mho61<-nAdXCA6_KtB;Q z*OkYb^A7h%BsvTo#8nQ1QC~1>2}RQZqG%y{{U@JmfZg9RNR|fWbL;sV3cK8cLTrk{ zsxx#cGwYeq248qKC7w5}Qw-m~Fm0y*PWuWqff7Di6Diu2ei_J)1;_R(;{t6*U$8vs z9={n;5n(CgtdUfv&4*ZsJtJ++aXvMFoC&8&p5^c?>DuU2aI` z9!AYf`vOuzqX!K>*sw}Z+Wf?)j_&bop(fC$ZJ>3k{=v8RTjIqPbYhg{SltRMo9Ry8 zhg+FAKo!3X2?P8-hGTg;h)~S@`!8a?F_^&%L@xJufs^iwqM^PX0e!>Hfth=+LKo_@tS-W(=H@r0FUM07FZ{i zD-%KSw?+QqG1SGLq%0uBLy30^ezd?VH#Vj*95xc$vqISC{>b6_Qko_92c-m$Q7RFQ zb*75Soq2s{-Ec@cUUl88QKZ1;JagDg8Y#-lFSZM4Qb4kHVP0k~8j3kTS$9lC$0}0V z&Df!kUa@gIPxamZTpwyddI|TIQ;l(mC@gCXbTlpu^ax-0$uk>)_>H$Tn!58J|F{Uw zve1@ok0)aKx%GunGjD}b%DCIR8eQYKK?A>K)JTYzYqFBtA<*KsLy;}U=R*wsj55uK zY{4NAmLXBG-mzUrS3a&hI$jZpdP$eTG&l;LyTqq+`5dpJC z3ZaK!e`GB7(BiJG@S<{Q?uNudytzuwIoJ0;3n7JNpXPccOdh!vKVW`KsRaW!Y0uwmS~Rz8!8fbD@&X5_;4jb#c9)+qWyY?!5o)NgwsreM`6;& ziP0d#LMt*XI&Nh)$LCFtL3n1b+2RlG?QSU1zOxC$8M{=(-iEn~AS#46sqUC5VJlP- zQ^+q@i8YYGd)}9!D7CyXN*~P_X)-)o+_>wtYpebv70W`Tqyxs z73OV8(BX@bivDSu3)2qczu)k`;FvARgwp)*4N`XKt0lH$cE?)hA04u;Tgg_h4;GZm zS$XQQPn>V=xdLqQvPh@bE{4udo-~(8WSLc3-)LyHzT>Q^T0Sh)LiTs*+l)GbNnMg5 zc>LfD#`61^6!O~nk78eBD;TO9;WClHW#d)PzmB-#MvcIv@I&BF7IivQuWMoPNP5tb zkGgxThRBl2HpC+(^kL^@eu2}d^qoANv^-@eBA@CHPE7;I=*m zI`?dkV`OiT^*gF`3!k5__|d9Dfd%s^>fBX^+?t=7#7Ej;q0kLq#jL~LnpqX1%`r0i zw+0vbh>jpwKnHgTGIjescex!y+x*NW-`cd`Hysfrf@ssNVW7IVvok&J!n2+8Ps~9> z`71HqQ-9G~UZMh3ohZMa%z^~2IMhpG8LS_fiZR;cM`+6x- zH!)r&#VgS2t?78!got<6V*8(V6i8oIzo(rIG{H|+QEguN2|_vlaw`#qysY5ZX!t|>If1d8JV3fF2OCb$u)ctS}P-ap(=f|l#V-) zV57pUli(*f0%azgbDI}V)1r9p;yrlT?C)e`3~}8(jOsB-&NGb7qnRjz+L+3}fRCq$ zR>S6N`E+CRZcJ)G22FXL!!Wz`+yj@SvTuwq^UryV$;`Scm(xM{BjuPaA#Ci3Y%wY3 zA2ML%yZ1}F1osx-5ri)a)<35bLy`>OtCYC-`+UloPPwkb+sq~B1qyVU>dj*eCp>iI zd&HDUnR)|xzZ?EYB@WGLGA9qCG%Zuj_q+HwX}8==GfMh$y0={yCtXgT=GM!=ZzXjP zemrr|GMo>!ThO4;U9_|>YfY56erH^;NHDy%C^_n5TBA~z$}kb&VQPp|K8GO+x51r_ z<-Dj3=ZpHWE!3K;n}cb`Jtt^drwsG5kWbOMvs`HlwdQnw7T%<}sir zH+E8pFoBz6W!^o|@FEK{n#~kYO~!6&NPM%)e=n)mKsQ~f3#1jmj5J^>rKf@7dwVw>Sl*H}Tb+@Tgx z?4Mq&<>VNwr8XL+36P=G#dvBJL89nbiv&iXWwc3#7d=Q{{qBR4JQvW7EsUX-u5J8? za+G({z>3Y7fYBzz&qVy1`o_Ze`@`ZliSXatr_O>0z2hUrE!jvKl7@Xn-rSLVYX!`_D+m_< z20WemHY?K6MMa9j2Qv7*{sm6)7iipS64s2(*{(TW(*eP3c00%fJ>U4jj66p3<^E0G zvkO(tsyr%%81Jaq?w34(19gWRvgS9M*9Mqlj~Nv$BgLW6K|`cL4)R^X+rx55RHty> zZ?nwUEp44$%bbOY{juOrNN3yF)9Ml^qKb8XTBkkpkhB zUF?L&4vf8miTq|hF3lrD4n2tUHrX&#QuYL;o4AwxHd0MM7Y?2X6hAfkJv^T}CNKu! z5l}R;JykM#(Ml>-bWT~1Vt(DWPg3dCm!DvFYM4l?2jC9UUr8F+_I0@L!V zy{hoVS>I{mFU>X~a}Q!1vsQa+5>@Vp5aI_wdcT%$_K;23 z8D6UC6&TQEcU;}^Tb%yIZ;>Qql6HaSI>Qw<`f$f|Wy)&b?zH=|ZmAB6(ql48g<{YY7eFd>V};kZQUe1>jYf*m&%_)x zF0-I6u)9hlCoX5U;&c?e%C!>P)yy%pOi3D{`U$dDc!sbYo!~M|-Sx8MGDZ!f+D9W` zwWVrx`ZPBM;}4Ey#CjA*u-%koKW5;+<&>U(Sy3BHV7 z^;#pTqhc@7)2TpR9D#=ThXbCHoFmJ$Mnz>}BlakS5hd!7{lXYQgZ3{?1SJ4mrbQ9E zvQ*Vo%03awrTnl0Jr{-ALIdrYpA`+aUR9Z{pQ+p<>FRM(EkCP#s=4^~WtuUPT$3-y z=+%rm)I`EazpzLVz1!F6^Hd-u4o^#=Z6??qQ70^E+&|I6kVbfJ*sCx0v#~iutty4F zv>{Pe>zRbce;o*=`m^KQy9dF`G!i|U=-$WXYWuaY=|@1N@EcHnsG&o*n6Pv#d}dl4 zEW>3qxan|ifW_)kmM6(rVGq~_OzTJ&bH8#www*V-0+ze&Nm%VByoRhrVlJ z5{PofoF1m8gKqQa^4=nZ^n*-M65~%A=LSygANZSB!`5*GC3`v3ze75mZ#7qQD$+4S zow_C^vSjH?k5lXdvu1i%c7UCv`$=C0kjkjf94GrQW<4VxPWs551dAVZP9Y~N3X-9E z^TY+KOx+8*Mk{a{JcUGcAe0a&k82&IqS`fxr1hj-PX@CjYf68ndfwCYqZSXtMPLqU zW}kL*6&r+Qj_1!FjqxMnSBU5Q?}?h`pdbR4iCjF-C@#YF%Dqr!OdbW^t9EoEI&+;b zU+SG7>{N;rOqf#nkii)n6+1JjfbXGCIy$y3O9#Z<@6QL^eznHRsSjXuG&~NFfUCI0 zGn9zv`#b`jK+^3f{$8h~M0-B-2%n0Ql$fUuMC*ON|5sJXNA%e-}tOrZXl~%ztMLr|BS&jdrr6UOc^F>C)J1- zt{V@%es5g}(95hdzOc+}PsFHRKW8dg08V^8B70LKo(1gQhbm z)lT8saiu<19rw%r>wNl#8>_QAEqAqky&HIAVt~bbi;{uWE*3$e1%Q?lm+cx|6O@4Yq(s^sNrEO{=p3J(d?^PsV$K^QrSTI4* zV!{F5gn(Asi3LIru97e#7J`wFVfkKbh7)~O6~AVcTL;tU{nPbU4yhRG-7!1 zqY~BvB@)3H^CEH?|6&?rHY#~jZ38k#cF{!BN})@EysTH&hlUv`ME(g1C+f{iXRWC< z2IIM8-U69ePFbdXbFr4TglB+-0&Vdn*-P=jcpv|A|52d@#Y1 zT$&}d)^;lH3yPuC2*?7iA*F*@{W4hw=S%vw2fopx5AOOxmRa&Uw%s#UNjG5JGfeSTU6f zZN$cu0fvCHf2?`cGgPK(j{vhwbxG~Il`k{^Wq;L4YSk=xx={0=e0|FJ5y0~aZ&)p0 zhahtF6@&z?z&Gi?I238oq|l{OLny@i%%rg%pKD!o9W~yMO#>FpdjwI+yR$x5o;f~c z9_&8~pIMmOx=;4?ffY0kM87hIKLChZrIx1T61cfo&I1mz<74ILwD?QU!`=Fy%R_aa zdK5Ux1*M}0eg~a7ni-LEyE!vvw}^WdA3+f`Umwj;UOAPJ^Z3SvQd6wqvhIq_1`}Fd z@o(xMvhd^INH0Q`Im<_Ifj$jXW6HnE#B1JHDMF)GZ-jNn{We-_^b0Mk=^8lziQ?nP zA?!fMn$R|jEL3K3cc++h77vPMyJ8JTe9>mz7}L&;FZI#D{G-n-o$8^-L|)@NsO${> ztZY<`hhXL+^CJ}k#z~|F`g@T zz|}SgR9a`Ama^_W#hj|H^Kd*NNOqKl?FO8@VZwt%hCwumRX6q~ zQhAHf!`BuGqHc+xn%{RTOvR?qFU(1V!U5Q8VmhFY(?26v*8V#yDB{_u6O>=R9pA5syQFPMTadrY5w(dH!M zBdqE`pp;DO(QhlW-^a;AW+9jf+u=KIQ>V@X8wP*;))YRpqTHc5=Pa--%Azj)zI zDEx|8gU*-w1(muv>lhRW1SR$};NqvjtCpURR(}>}+bzQPS)Hy0vU!BW(7fd|8)Di? z%3g#q4`*QY+dY>vq@3;sQvQT@EO`8=QhY4K$G0(Ji0BWTxO;oz=`dw0Fv47#EX_H1 zVg=-O8$#>vB%FSYSN=WHniA9Nn^|2ZZF`rYDn~#H@c$!h5Ft_HOcQ|VpE^4{!FHWWfZqeqvg-mK<#ni$?tXe7kH-~ zbGJN{1LfAXZ*X?@(+e-F%WcKCCl;EIGvX;CH*vh)rwfEn+o%x7ZF5;qt6Jb$9Nh_> zQ}xr+FqslJEy)$5GMP&*Nw$-F2>9A_3eipFzr8)RH5Tv%f{LXEeEgZmdx;0S9*Qun zNtScYMoy$%g>UzsHYvRMqRF#v0-!+kGGqNQHeH zMlY)~SeMUeN^N?d^hC8rHQn7&Q9Cp<>ZCXG{80SBylJgs@AUPKg&7ocoIL*vptQTo z?(a|^;gL({kY?y_^`UU6qDCdqDa20CvOPp~W|>ZntOlKheLj18esL$gOvn5B{^{y~ z)!3g=YQbu*jx7lo&4!MgO1wwN@Rb7$f&CtpnDnLQKU-SGD~>LVB`EK&K`K+N8LEtG z+9{{2tElTKYLYl=#Bgg{Vd{0JDa{>4`Py(q;cEBRiK1|Poe^BI`;j?Yziga0%nsS% zHl=CE9q~)dDU4U!@{qwnRVK5|J^YEBb-{M<>Z^GA;7E!Q#-BFf6}0lT{j2NDrh3+| z;`=?~bQc>&nOY&Agi?&c()m>%ki@qWV3ZrvBLR3K_a-_y$^R!{l#zVu{YC;|HKtBVvmWU!smeN|xNm|*2HmB7eD({*Lq4BS#h=Vqh z;-e*)l`Yo7MI|~mqoyK|e>=gdIC)o7(F$E5%FPfI%KN#XGo9wpSZ~V)ginJ{0^AYz z9DCLn=!EhQKn^|1`c)h9^M#Xxt0${=WrOXy;K)ka3}9doXoY^5ZW5i< z=ymO+X{FVtlC701ow(lXs#=`&935a#?R~qTN6p%D$OE!1JsSyeriWG*#ta^339|`n ze2|)PW+$i$h2d!{FIwpV?185Y#_)!#{YEdeGbN;E?RP74x~r6qkYwR(K$T^&8GGna zUzwtl;;MFRDlSm&EL74F?}Mvf6Hlr_Obv>?Lwtj3CpJ2-@3G&xL-}O2X>RA`(P&lR zxmbp;mYZ;m(1*lbO>R}a%M=FL7ZCFT)|l{lXuoARICgG#X-V2-z`0h3uaP*#p^NU` z}yr@W{@{4RJ zui_U>Pb^raGb8#vI8&EL?Fc&MsODTl2w^b2<{`XMM0}=QTC9n6U6n6Z%WAAP1i`;{ zrAZSAX6F-UPrkIGPxGq z7xLJXIsUL0)+jh79RxI}7*!EYsO9+q1GBgd4A(!FusD%gbXp&OD4iUR7`f*yUa5AXqB-evQX-ZG5{d zK+>%*GxT|g+mn(J&z46)Bt#iiUo0BmUVLeV-@R>={W#J&BiUN zq-;)CDBI7WtXK0;1+eqjvCoyUobuWDSF;oY##+JojPXq|4odAudF3266Pu{Ux^6_e0RHMnmOEp<-L%_`UWY1K$tYD>- zw_%l|JRyY3aK9{@YR3chh$ypdY2%}qL&}@{c#!=1h68=?p+FMA0{GV*3#TZnDpM_G G9{NAYCegJ3 literal 0 HcmV?d00001 diff --git a/src/main.go b/src/main.go new file mode 100644 index 0000000..3673efa --- /dev/null +++ b/src/main.go @@ -0,0 +1,125 @@ +package main + +import ( + "fmt" + "log" + "net/http" + "os" + "os/signal" + "strconv" + "sync" + "syscall" + + "github.com/buaazp/fasthttprouter" + "github.com/valyala/fasthttp" +) + +var ( + config *Configuration = &Configuration{} + r *Redis = &Redis{} + blockedServers []string = nil + blockedServersMutex *sync.Mutex = &sync.Mutex{} +) + +func init() { + if err := config.ReadFile("config.yml"); err != nil { + log.Fatal(err) + } + + if err := r.Connect(config.Redis); err != nil { + log.Fatal(err) + } + + log.Println("Successfully connected to Redis") + + if Contains(os.Args, "--flush-cache") { + keys, err := r.Keys("java:*") + + if err != nil { + log.Fatal(err) + } + + if err = r.Delete(keys...); err != nil { + log.Fatal(err) + } + + keys, err = r.Keys("bedrock:*") + + if err != nil { + log.Fatal(err) + } + + if err = r.Delete(keys...); err != nil { + log.Fatal(err) + } + + keys, err = r.Keys("favicon:*") + + if err != nil { + log.Fatal(err) + } + + if err = r.Delete(keys...); err != nil { + log.Fatal(err) + } + + log.Println("Successfully flushed all cache keys") + + os.Exit(0) + } + + if instanceID := os.Getenv("INSTANCE_ID"); len(instanceID) > 0 { + value, err := strconv.ParseUint(instanceID, 10, 16) + + if err != nil { + log.Fatal(err) + } + + config.Port += uint16(value) + } + + if err := GetBlockedServerList(); err != nil { + log.Fatal(err) + } +} + +func middleware(next fasthttp.RequestHandler) fasthttp.RequestHandler { + return func(ctx *fasthttp.RequestCtx) { + if config.Environment == "development" { + log.Printf("GET %s - %s \"%s\"", ctx.Request.URI().Path(), ctx.RemoteAddr(), ctx.UserAgent()) + } + + ctx.Response.Header.Set("Access-Control-Allow-Headers", "*") + ctx.Response.Header.Set("Access-Control-Allow-Methods", "HEAD,GET,POST,OPTIONS") + ctx.Response.Header.Set("Access-Control-Allow-Origin", "*") + ctx.Response.Header.Set("Access-Control-Expose-Headers", "X-Cache-Hit,X-Cache-Time-Remaining,X-Server-Status,Content-Disposition") + + next(ctx) + } +} + +func main() { + defer r.Close() + + router := fasthttprouter.New() + router.GET("/ping", PingHandler) + router.GET("/status/java/:address", JavaStatusHandler) + router.GET("/status/bedrock/:address", BedrockStatusHandler) + router.GET("/favicon/:address", FaviconNoExtensionHandler) + router.GET("/favicon/:address/*filename", FaviconHandler) + + router.PanicHandler = func(rc *fasthttp.RequestCtx, i interface{}) { + log.Println(i) + } + + router.NotFound = func(ctx *fasthttp.RequestCtx) { + WriteError(ctx, nil, http.StatusNotFound) + } + + log.Printf("Listening on %s:%d\n", config.Host, config.Port) + log.Fatal(fasthttp.ListenAndServe(fmt.Sprintf("%s:%d", config.Host, config.Port), middleware(router.Handler))) + + s := make(chan os.Signal) + signal.Notify(s, os.Interrupt, syscall.SIGTERM) + <-s +} diff --git a/src/redis.go b/src/redis.go new file mode 100644 index 0000000..40b2f7d --- /dev/null +++ b/src/redis.go @@ -0,0 +1,155 @@ +package main + +import ( + "context" + "errors" + "time" + + "github.com/go-redis/redis/v8" +) + +type Redis struct { + Conn *redis.Client +} + +func (r *Redis) Connect(uri string) error { + opts, err := redis.ParseURL(uri) + + if err != nil { + return err + } + + conn := redis.NewClient(opts) + + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + if err = conn.Ping(ctx).Err(); err != nil { + return err + } + + r.Conn = conn + + return nil +} + +func (r *Redis) TTL(key string) (time.Duration, error) { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + val, err := r.Conn.TTL(ctx, key).Result() + + if err != nil { + if errors.Is(err, redis.Nil) { + return 0, nil + } + + return 0, err + } + + return val, nil +} + +func (r *Redis) Exists(key string) (bool, error) { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + val, err := r.Conn.Exists(ctx, key).Result() + + return val == 1, err +} + +func (r *Redis) Get(key string) (string, error) { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + val, err := r.Conn.Get(ctx, key).Result() + + if err != nil { + if errors.Is(err, redis.Nil) { + return "", nil + } + + return "", err + } + + return val, nil +} + +func (r *Redis) GetBytes(key string) ([]byte, error) { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + result := r.Conn.Get(ctx, key) + + if err := result.Err(); err != nil { + if errors.Is(err, redis.Nil) { + return nil, nil + } + + return nil, err + } + + return result.Bytes() +} + +func (r *Redis) Set(key string, value interface{}, ttl time.Duration) error { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + return r.Conn.Set(ctx, key, value, ttl).Err() +} + +func (r *Redis) GetValueAndTTL(key string) (bool, string, time.Duration, error) { + exists, err := r.Exists(key) + + if err != nil { + return false, "", 0, err + } + + if !exists { + return false, "", 0, nil + } + + value, err := r.Get(key) + + if err != nil { + return false, "", 0, err + } + + ttl, err := r.TTL(key) + + return true, value, ttl, err +} + +func (r *Redis) Keys(pattern string) ([]string, error) { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + res := r.Conn.Keys(ctx, pattern) + + if err := res.Err(); err != nil { + return nil, err + } + + return res.Result() +} + +func (r *Redis) Delete(keys ...string) error { + ctx, cancel := context.WithTimeout(context.Background(), time.Second*5) + + defer cancel() + + return r.Conn.Del(ctx, keys...).Err() +} + +func (r *Redis) Close() error { + return r.Conn.Close() +} diff --git a/src/routes.go b/src/routes.go new file mode 100644 index 0000000..6ade466 --- /dev/null +++ b/src/routes.go @@ -0,0 +1,308 @@ +package main + +import ( + "encoding/base64" + "encoding/json" + "fmt" + "net/http" + "strconv" + "strings" + + "github.com/valyala/fasthttp" +) + +func PingHandler(ctx *fasthttp.RequestCtx) { + ctx.SetBodyString(http.StatusText(http.StatusOK)) +} + +func JavaStatusHandler(ctx *fasthttp.RequestCtx) { + host, port, err := ParseAddress(ctx.UserValue("address").(string), 25565) + + if err != nil { + WriteError(ctx, nil, http.StatusBadRequest, "Invalid/malformed address") + + return + } + + cacheEnabled, cacheKey, err := IsCacheEnabled(ctx, "java", host, port) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if cacheEnabled { + exists, cache, ttl, err := r.GetValueAndTTL(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if exists { + ctx.Response.Header.Set("X-Cache-Hit", "TRUE") + ctx.Response.Header.Set("X-Cache-Time-Remaining", strconv.Itoa(int(ttl.Seconds()))) + ctx.SetContentType("application/json") + ctx.SetBodyString(cache) + + return + } + } + + ctx.Response.Header.Set("X-Cache-Hit", "FALSE") + + status := GetJavaStatus(host, port) + + if status.Online && status.Response.Favicon != nil { + data, err := base64.StdEncoding.DecodeString(strings.Replace(*status.Response.Favicon, "data:image/png;base64,", "", 1)) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if err := r.Set(fmt.Sprintf("favicon:%s-%d", host, port), data, config.FaviconCacheTTL); err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + } + + data, err := json.Marshal(status) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if err = r.Set(cacheKey, data, config.StatusCacheTTL); err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.SetContentType("application/json") + ctx.SetBody(data) +} + +func BedrockStatusHandler(ctx *fasthttp.RequestCtx) { + host, port, err := ParseAddress(ctx.UserValue("address").(string), 19132) + + if err != nil { + WriteError(ctx, nil, http.StatusBadRequest, "Invalid/malformed address") + + return + } + + cacheEnabled, cacheKey, err := IsCacheEnabled(ctx, "bedrock", host, port) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if cacheEnabled { + exists, cache, ttl, err := r.GetValueAndTTL(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if exists { + ctx.Response.Header.Set("X-Cache-Hit", "TRUE") + ctx.Response.Header.Set("X-Cache-Time-Remaining", strconv.Itoa(int(ttl.Seconds()))) + ctx.SetContentType("application/json") + ctx.SetBodyString(cache) + + return + } + } + + ctx.Response.Header.Set("X-Cache-Hit", "FALSE") + + data, err := json.Marshal(GetBedrockStatus(host, port)) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if err = r.Set(cacheKey, data, config.StatusCacheTTL); err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.SetContentType("application/json") + ctx.SetBody(data) +} + +func FaviconNoExtensionHandler(ctx *fasthttp.RequestCtx) { + host, port, err := ParseAddress(ctx.UserValue("address").(string), 25565) + + if err != nil { + WriteError(ctx, nil, http.StatusBadRequest, "Invalid/malformed address") + + return + } + + cacheKey := fmt.Sprintf("favicon:%s-%d", host, port) + + exists, err := r.Exists(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if exists { + value, err := r.GetBytes(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.Response.Header.Set("X-Cache-Hit", "TRUE") + ctx.Response.Header.Set("X-Server-Status", "online-cache") + ctx.SetContentType("image/png") + ctx.SetBody(value) + + return + } + + ctx.Response.Header.Set("X-Cache-Hit", "FALSE") + + status := GetJavaStatus(host, port) + + if !status.Online { + ctx.Response.Header.Set("X-Server-Status", "offline") + ctx.SetContentType("image/png") + ctx.SetBody(defaultIconBytes) + + return + } + + if status.Response.Favicon == nil { + ctx.Response.Header.Set("X-Server-Status", "online-no-icon") + ctx.SetContentType("image/png") + ctx.SetBody(defaultIconBytes) + + return + } + + data, err := base64.StdEncoding.DecodeString(strings.Replace(*status.Response.Favicon, "data:image/png;base64,", "", 1)) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if err := r.Set(cacheKey, data, config.FaviconCacheTTL); err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.Response.Header.Set("X-Server-Status", "online") + ctx.SetContentType("image/png") + ctx.SetBody(data) +} + +func FaviconHandler(ctx *fasthttp.RequestCtx) { + filename := ctx.UserValue("filename").(string) + + if !strings.HasSuffix(filename, ".png") { + WriteError(ctx, nil, http.StatusBadRequest, "Filename must end with .png") + + return + } + + host, port, err := ParseAddress(ctx.UserValue("address").(string), 25565) + + if err != nil { + WriteError(ctx, nil, http.StatusBadRequest, "Invalid/malformed address") + + return + } + + cacheKey := fmt.Sprintf("favicon:%s-%d", host, port) + + exists, err := r.Exists(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if exists { + value, err := r.GetBytes(cacheKey) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.Response.Header.Set("X-Cache-Hit", "TRUE") + ctx.Response.Header.Set("X-Server-Status", "online-cache") + ctx.Response.Header.Set("Content-Disposition", fmt.Sprintf("inline; filename=\"%s\"", filename)) + ctx.SetContentType("image/png") + ctx.SetBody(value) + + return + } + + ctx.Response.Header.Set("X-Cache-Hit", "FALSE") + + status := GetJavaStatus(host, port) + + if !status.Online { + ctx.Response.Header.Set("X-Server-Status", "offline") + ctx.Response.Header.Set("Content-Disposition", fmt.Sprintf("inline; filename=\"%s\"", filename)) + ctx.SetContentType("image/png") + ctx.SetBody(defaultIconBytes) + + return + } + + if status.Response.Favicon == nil { + ctx.Response.Header.Set("X-Server-Status", "online-no-icon") + ctx.Response.Header.Set("Content-Disposition", fmt.Sprintf("inline; filename=\"%s\"", filename)) + ctx.SetContentType("image/png") + ctx.SetBody(defaultIconBytes) + + return + } + + data, err := base64.StdEncoding.DecodeString(strings.Replace(*status.Response.Favicon, "data:image/png;base64,", "", 1)) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + if err := r.Set(cacheKey, data, config.FaviconCacheTTL); err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return + } + + ctx.Response.Header.Set("X-Server-Status", "online") + ctx.Response.Header.Set("Content-Disposition", fmt.Sprintf("inline; filename=\"%s\"", filename)) + ctx.SetContentType("image/png") + ctx.SetBody(data) +} diff --git a/src/status.go b/src/status.go new file mode 100644 index 0000000..6d3865d --- /dev/null +++ b/src/status.go @@ -0,0 +1,251 @@ +package main + +import "github.com/PassTheMayo/mcstatus/v4" + +type StatusResponse[T JavaStatus | BedrockStatus] struct { + Online bool `json:"online"` + Host string `json:"host"` + Port uint16 `json:"port"` + EULABlocked bool `json:"eula_blocked"` + Response *T `json:"response"` +} + +type JavaStatus struct { + Version *Version `json:"version"` + Players Players `json:"players"` + MOTD MOTD `json:"motd"` + Favicon *string `json:"favicon"` + ModInfo *ModInfo `json:"mod_info"` + SRVRecord *SRVRecord `json:"srv_record"` +} + +type Players struct { + Online int `json:"online"` + Max int `json:"max"` + Sample []SamplePlayer `json:"sample"` +} + +type SamplePlayer struct { + ID string `json:"id"` + Name string `json:"name"` + Clean string `json:"clean"` + HTML string `json:"html"` +} + +type ModInfo struct { + Type string `json:"type"` + Mods []Mod `json:"mods"` +} + +type Mod struct { + ID string `json:"id"` + Version string `json:"version"` +} + +type BedrockStatus struct { + ServerGUID int64 `json:"server_guid"` + Edition *string `json:"edition"` + MOTD *MOTD `json:"motd"` + ProtocolVersion *int64 `json:"protocol_version"` + Version *string `json:"version"` + OnlinePlayers *int64 `json:"online_players"` + MaxPlayers *int64 `json:"max_players"` + ServerID *string `json:"server_id"` + Gamemode *string `json:"gamemode"` + GamemodeID *int64 `json:"gamemode_id"` + PortIPv4 *uint16 `json:"port_ipv4"` + PortIPv6 *uint16 `json:"port_ipv6"` + SRVRecord *SRVRecord `json:"srv_record"` +} + +type MOTD struct { + Raw string `json:"raw"` + Clean string `json:"clean"` + HTML string `json:"html"` +} + +type Version struct { + Name string `json:"name"` + Protocol int `json:"protocol"` +} + +type SRVRecord struct { + Host string `json:"host"` + Port uint16 `json:"port"` +} + +func GetJavaStatus(host string, port uint16) (resp StatusResponse[JavaStatus]) { + status, err := mcstatus.Status(host, port) + + if err != nil { + statusLegacy, err := mcstatus.StatusLegacy(host, port) + + if err != nil { + resp = StatusResponse[JavaStatus]{ + Online: false, + Host: host, + Port: port, + EULABlocked: IsBlockedAddress(host), + Response: nil, + } + + return + } + + resp = StatusResponse[JavaStatus]{ + Online: true, + Host: host, + Port: port, + EULABlocked: IsBlockedAddress(host), + Response: &JavaStatus{ + Version: nil, + Players: Players{ + Online: statusLegacy.Players.Online, + Max: statusLegacy.Players.Max, + Sample: make([]SamplePlayer, 0), + }, + MOTD: MOTD{ + Raw: statusLegacy.MOTD.Raw, + Clean: statusLegacy.MOTD.Clean, + HTML: statusLegacy.MOTD.HTML, + }, + Favicon: nil, + ModInfo: nil, + SRVRecord: nil, + }, + } + + if statusLegacy.Version != nil { + resp.Response.Version = &Version{ + Name: statusLegacy.Version.Name, + Protocol: statusLegacy.Version.Protocol, + } + } + + if statusLegacy.SRVResult != nil { + resp.Response.SRVRecord = &SRVRecord{ + Host: statusLegacy.SRVResult.Host, + Port: statusLegacy.SRVResult.Port, + } + } + + return + } + + samplePlayers := make([]SamplePlayer, 0) + + for _, player := range status.Players.Sample { + samplePlayers = append(samplePlayers, SamplePlayer{ + ID: player.ID, + Name: player.Name, + Clean: player.Clean, + HTML: player.HTML, + }) + } + + resp = StatusResponse[JavaStatus]{ + Online: true, + Host: host, + Port: port, + EULABlocked: IsBlockedAddress(host), + Response: &JavaStatus{ + Version: &Version{ + Name: status.Version.Name, + Protocol: status.Version.Protocol, + }, + Players: Players{ + Online: status.Players.Online, + Max: status.Players.Max, + Sample: samplePlayers, + }, + MOTD: MOTD{ + Raw: status.MOTD.Raw, + Clean: status.MOTD.Clean, + HTML: status.MOTD.HTML, + }, + Favicon: status.Favicon, + ModInfo: nil, + SRVRecord: nil, + }, + } + + if status.ModInfo != nil { + mods := make([]Mod, 0) + + for _, mod := range status.ModInfo.Mods { + mods = append(mods, Mod{ + ID: mod.ID, + Version: mod.Version, + }) + } + + resp.Response.ModInfo = &ModInfo{ + Type: status.ModInfo.Type, + Mods: mods, + } + } + + if status.SRVResult != nil { + resp.Response.SRVRecord = &SRVRecord{ + Host: status.SRVResult.Host, + Port: status.SRVResult.Port, + } + } + + return +} + +func GetBedrockStatus(host string, port uint16) (resp StatusResponse[BedrockStatus]) { + status, err := mcstatus.StatusBedrock(host, port) + + if err != nil { + resp = StatusResponse[BedrockStatus]{ + Online: false, + Host: host, + Port: port, + EULABlocked: IsBlockedAddress(host), + Response: nil, + } + + return + } + + resp = StatusResponse[BedrockStatus]{ + Online: true, + Host: host, + Port: port, + EULABlocked: IsBlockedAddress(host), + Response: &BedrockStatus{ + ServerGUID: status.ServerGUID, + Edition: status.Edition, + MOTD: nil, + ProtocolVersion: status.ProtocolVersion, + Version: status.Version, + OnlinePlayers: status.OnlinePlayers, + MaxPlayers: status.MaxPlayers, + ServerID: status.ServerID, + Gamemode: status.Gamemode, + GamemodeID: status.GamemodeID, + PortIPv4: status.PortIPv4, + PortIPv6: status.PortIPv6, + SRVRecord: nil, + }, + } + + if status.MOTD != nil { + resp.Response.MOTD = &MOTD{ + Raw: status.MOTD.Raw, + Clean: status.MOTD.Clean, + HTML: status.MOTD.HTML, + } + } + + if status.SRVResult != nil { + resp.Response.SRVRecord = &SRVRecord{ + Host: status.SRVResult.Host, + Port: status.SRVResult.Port, + } + } + + return +} diff --git a/src/util.go b/src/util.go new file mode 100644 index 0000000..cbe8f5b --- /dev/null +++ b/src/util.go @@ -0,0 +1,161 @@ +package main + +import ( + "crypto/sha1" + _ "embed" + "encoding/hex" + "errors" + "fmt" + "io/ioutil" + "log" + "net/http" + "regexp" + "strconv" + "strings" + + "github.com/valyala/fasthttp" +) + +var ( + //go:embed default-icon.png + defaultIconBytes []byte + ipAddressRegExp = regexp.MustCompile("^\\d{1,3}(\\.\\d{1,3}){3}$") +) + +var ( + ErrNoAddressMatch = errors.New("address does not match any known format") +) + +func GetBlockedServerList() error { + resp, err := http.Get("https://sessionserver.mojang.com/blockedservers") + + if err != nil { + return err + } + + if resp.StatusCode != http.StatusOK { + return fmt.Errorf("unexpected status code: %d", resp.StatusCode) + } + + defer resp.Body.Close() + + body, err := ioutil.ReadAll(resp.Body) + + if err != nil { + return err + } + + blockedServersMutex.Lock() + blockedServers = strings.Split(string(body), "\n") + blockedServersMutex.Unlock() + + return nil +} + +func IsBlockedAddress(address string) bool { + split := strings.Split(strings.ToLower(address), ".") + + isIPAddress := ipAddressRegExp.MatchString(address) + + for k := range split { + newAddress := "" + + switch k { + case 0: + { + newAddress = strings.Join(split, ".") + + break + } + default: + { + if isIPAddress { + newAddress = fmt.Sprintf("%s.*", strings.Join(split[0:len(split)-k], ".")) + } else { + newAddress = fmt.Sprintf("*.%s", strings.Join(split[k:], ".")) + } + + break + } + } + + newAddressBytes := sha1.Sum([]byte(newAddress)) + newAddressHash := hex.EncodeToString(newAddressBytes[:]) + + blockedServersMutex.Lock() + + if Contains(blockedServers, newAddressHash) { + blockedServersMutex.Unlock() + + return true + } + + blockedServersMutex.Unlock() + } + + return false +} + +func ParseAddress(address string, defaultPort uint16) (string, uint16, error) { + result := strings.SplitN(address, ":", 2) + + if len(result) < 1 { + return "", 0, ErrNoAddressMatch + } + + if len(result) < 2 { + return result[0], defaultPort, nil + } + + port, err := strconv.ParseUint(result[1], 10, 16) + + if err != nil { + return "", 0, err + } + + return result[0], uint16(port), nil +} + +func WriteError(ctx *fasthttp.RequestCtx, err error, statusCode int, body ...string) { + ctx.SetStatusCode(statusCode) + + if len(body) > 0 { + ctx.SetBodyString(body[0]) + } else { + ctx.SetBodyString(http.StatusText(statusCode)) + } + + if err != nil { + log.Println(err) + } +} + +func IsCacheEnabled(ctx *fasthttp.RequestCtx, cacheType, host string, port uint16) (bool, string, error) { + key := fmt.Sprintf("%s:%s-%d", cacheType, host, port) + + if authKey := ctx.Request.Header.Peek("Authorization"); len(authKey) > 0 { + exists, err := r.Exists(fmt.Sprintf("auth_key:%s", authKey)) + + if err != nil { + WriteError(ctx, err, http.StatusInternalServerError) + + return config.CacheEnable, key, err + } + + if exists { + return false, key, nil + } + } + + return config.CacheEnable, key, nil +} + +func Contains[T comparable](arr []T, x T) bool { + for _, v := range arr { + if v == x { + return true + } + } + + return false +}