Compare commits
8 Commits
Author | SHA1 | Date | |
---|---|---|---|
838051c880
![]() |
|||
5ff262d5da
|
|||
2e2f9f7fee
|
|||
ff66877192
![]() |
|||
aa1699243a
|
|||
7c236a49b6
|
|||
df992caf1e
|
|||
88ef0c23a0
|
3
.dockerignore
Normal file
3
.dockerignore
Normal file
@@ -0,0 +1,3 @@
|
||||
.idea/
|
||||
|
||||
top-wallpaper
|
42
.gitea/workflows/build.yml
Normal file
42
.gitea/workflows/build.yml
Normal file
@@ -0,0 +1,42 @@
|
||||
on:
|
||||
workflow_dispatch:
|
||||
inputs:
|
||||
tag:
|
||||
description: 'Tag to build and push'
|
||||
required: false
|
||||
default: 'latest'
|
||||
push:
|
||||
tags:
|
||||
- "v*"
|
||||
|
||||
name: 🏗️✨ Build Workflow
|
||||
|
||||
jobs:
|
||||
Build:
|
||||
name: 🖥️ 🔨 Build
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: 📖 🔍 Checkout Repository Code
|
||||
uses: actions/checkout@v4
|
||||
with:
|
||||
fetch-depth: 1
|
||||
submodules: 'true'
|
||||
|
||||
- name: 🔑 🔐 Login to Docker Hub
|
||||
uses: docker/login-action@v3
|
||||
with:
|
||||
username: ${{ secrets.DOCKER_USERNAME }}
|
||||
password: ${{ secrets.DOCKER_PASSWORD }}
|
||||
|
||||
- name: 🏗️ 🔧 Set up Docker Buildx
|
||||
uses: docker/setup-buildx-action@v3
|
||||
|
||||
- name: 🐳 🔨 Build Container
|
||||
uses: docker/build-push-action@v6
|
||||
with:
|
||||
push: true
|
||||
sbom: true
|
||||
provenance: true
|
||||
context: .
|
||||
dockerfile: Dockerfile
|
||||
tags: siteworxpro/top-wallpaper:${{ gitea.ref_name }}
|
54
.gitea/workflows/lint.yml
Normal file
54
.gitea/workflows/lint.yml
Normal file
@@ -0,0 +1,54 @@
|
||||
on:
|
||||
push:
|
||||
branches:
|
||||
- "*"
|
||||
|
||||
name: 🚨 Test Code Base
|
||||
run-name: ${{ github.workflow }}-{{ github.ref_name }}-{{ github.run_number }}
|
||||
|
||||
env:
|
||||
GO_VERSION: '1.25.0'
|
||||
|
||||
jobs:
|
||||
test-go:
|
||||
name: 🔍 🐹 Go Tests
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
|
||||
- name: ⚙️ 🐹 Set up Go Environment
|
||||
uses: actions/setup-go@v2
|
||||
with:
|
||||
go-version: ${{ env.GO_VERSION }}
|
||||
cache: true
|
||||
|
||||
- name: 📖 🔍 Checkout Repository Code
|
||||
uses: actions/checkout@v2
|
||||
with:
|
||||
fetch-depth: 1
|
||||
|
||||
- name: 📦 📥 Install Dependencies
|
||||
run: |
|
||||
go mod download
|
||||
|
||||
- name: ✅ 🔍 Run Go Tests
|
||||
run: |
|
||||
go test -v ./... -coverprofile=coverage.out
|
||||
|
||||
golangci-lint:
|
||||
name: 🧹 Lint
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- name: 📥 Checkout code
|
||||
uses: actions/checkout@v3
|
||||
|
||||
- name: 🛠️ Set up Go
|
||||
uses: actions/setup-go@v4
|
||||
with:
|
||||
go-version: ${{ env.GO_VERSION }}
|
||||
|
||||
- name: 📦 📥 Install Dependencies
|
||||
run: |
|
||||
go mod download
|
||||
|
||||
- name: ✅ 🔍 Run Go Lint
|
||||
uses: golangci/golangci-lint-action@v8.0.0
|
61
.golangci.yml
Normal file
61
.golangci.yml
Normal file
@@ -0,0 +1,61 @@
|
||||
version: "2"
|
||||
linters:
|
||||
default: none
|
||||
enable:
|
||||
- bodyclose
|
||||
- gocritic
|
||||
- govet
|
||||
- ineffassign
|
||||
- staticcheck
|
||||
- testifylint
|
||||
- unused
|
||||
settings:
|
||||
govet:
|
||||
disable:
|
||||
- fieldalignment
|
||||
enable-all: true
|
||||
exclusions:
|
||||
generated: lax
|
||||
presets:
|
||||
- comments
|
||||
- common-false-positives
|
||||
- legacy
|
||||
- std-error-handling
|
||||
rules:
|
||||
- linters:
|
||||
- bodyclose
|
||||
path: .*_test.go
|
||||
- linters:
|
||||
- bodyclose
|
||||
path: router/timeout_response.go
|
||||
# False positives on CGO generated code
|
||||
- linters:
|
||||
- staticcheck
|
||||
path: vips/.*
|
||||
text: 'SA4000:'
|
||||
# False positives on CGO generated code
|
||||
- linters:
|
||||
- gocritic
|
||||
path: vips/.*
|
||||
text: dupSubExpr
|
||||
# False positives on CGO generated code
|
||||
- linters:
|
||||
- staticcheck
|
||||
text: 'ST1005:'
|
||||
paths:
|
||||
- .tmp
|
||||
- vendor
|
||||
- third_party$
|
||||
- builtin$
|
||||
- examples$
|
||||
formatters:
|
||||
enable:
|
||||
- goimports
|
||||
exclusions:
|
||||
generated: lax
|
||||
paths:
|
||||
- .tmp
|
||||
- vendor
|
||||
- third_party$
|
||||
- builtin$
|
||||
- examples$
|
@@ -1,10 +1,10 @@
|
||||
FROM siteworxpro/golang:1.24.0 AS build
|
||||
FROM siteworxpro/golang:1.25.0 AS build
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
ADD . .
|
||||
|
||||
ENV GOPRIVATE=git.s.int
|
||||
ENV GOPRIVATE=gitea.siteworxpro.com
|
||||
ENV GOPROXY=direct
|
||||
ENV CGO_ENABLED=0
|
||||
|
||||
|
15
README.md
15
README.md
@@ -30,7 +30,20 @@ PORT: Port to serve the image (default: 8080)
|
||||
|
||||
## docker
|
||||
```shell
|
||||
docker run --rm -d -p 8080:8080 --name top-wallpaper -e REDIS_URL=redis -e REDIS_PORT=6379 -e REDIS_PASSWORD=pass -e REDIS_DB=0 -e ALLOWED_ORIGINS=http://localhost:8080 -e PATH=/wallpaper -e PORT=8080 siteworxpro/top-wallpaper:latest
|
||||
docker run --rm -d -p 8080:8080 --name top-wallpaper -e REDIS_URL=redis -e REDIS_PORT=6379 -e REDIS_PASSWORD=pass -e REDIS_DB=0 -e ALLOWED_ORIGINS=http://localhost:8080 -e PATH=/wallpaper -e PORT=8080 siteworxpro/top-wallpaper:v1.3.1
|
||||
```
|
||||
|
||||
## Docker Compose
|
||||
```shell
|
||||
curl -L https://gitea.siteworxpro.com/siteworxpro/top-wallpaper/raw/tag/v1.3.1/docker-compose.yml | docker compose -f - up -d
|
||||
````
|
||||
Access the app at [https://localhost](https://localhost).
|
||||
|
||||
## Embedding the Image
|
||||
You can embed the image in your HTML using the following code:
|
||||
|
||||
```html
|
||||
<img src="https://localhost" alt="Top Wallpaper" />
|
||||
```
|
||||
|
||||
## License
|
||||
|
50
docker-compose.yml
Normal file
50
docker-compose.yml
Normal file
@@ -0,0 +1,50 @@
|
||||
volumes:
|
||||
redis_data:
|
||||
|
||||
services:
|
||||
traefik:
|
||||
image: traefik:latest
|
||||
container_name: traefik
|
||||
ports:
|
||||
- "80:80"
|
||||
- "443:443"
|
||||
volumes:
|
||||
- "/var/run/docker.sock:/var/run/docker.sock"
|
||||
restart: always
|
||||
command:
|
||||
- "--providers.docker=true"
|
||||
- "--providers.docker.exposedByDefault=false"
|
||||
- "--entrypoints.web.address=:80"
|
||||
- "--entrypoints.web-secure.address=:443"
|
||||
- "--accesslog=true"
|
||||
- "--entrypoints.web.http.redirections.entryPoint.to=web-secure"
|
||||
- "--entrypoints.web.http.redirections.entryPoint.scheme=https"
|
||||
- "--entrypoints.web.http.redirections.entrypoint.permanent=true"
|
||||
|
||||
redis:
|
||||
image: redis:latest
|
||||
container_name: redis
|
||||
restart: always
|
||||
healthcheck:
|
||||
test: ["CMD", "redis-cli", "ping"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
volumes:
|
||||
- redis_data:/data
|
||||
|
||||
api:
|
||||
labels:
|
||||
- "traefik.enable=true"
|
||||
- "traefik.http.routers.api.entrypoints=web-secure"
|
||||
- "traefik.http.routers.api.rule=Host(`localhost`) || Host(`127.0.0.1`)"
|
||||
- "traefik.http.routers.api.tls=true"
|
||||
- "traefik.http.services.api.loadbalancer.server.port=8080"
|
||||
image: siteworxpro/top-wallpaper:v1.3.1
|
||||
container_name: top-wallpaper
|
||||
environment:
|
||||
REDIS_URL: "redis"
|
||||
CORS_ORIGINS: "https://localhost"
|
||||
depends_on:
|
||||
redis:
|
||||
condition: service_healthy
|
20
go.mod
20
go.mod
@@ -1,24 +1,24 @@
|
||||
module github.com/siteworxpro/top-wallpaper
|
||||
|
||||
go 1.24.0
|
||||
go 1.25.0
|
||||
|
||||
require (
|
||||
github.com/labstack/echo/v4 v4.13.3
|
||||
github.com/labstack/echo/v4 v4.13.4
|
||||
github.com/labstack/gommon v0.4.2
|
||||
github.com/nfnt/resize v0.0.0-20180221191011-83c6a9932646
|
||||
github.com/redis/go-redis/v9 v9.7.0
|
||||
github.com/redis/go-redis/v9 v9.12.1
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/cespare/xxhash/v2 v2.2.0 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||
github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
|
||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
||||
github.com/valyala/fasttemplate v1.2.2 // indirect
|
||||
golang.org/x/crypto v0.31.0 // indirect
|
||||
golang.org/x/net v0.33.0 // indirect
|
||||
golang.org/x/sys v0.28.0 // indirect
|
||||
golang.org/x/text v0.21.0 // indirect
|
||||
golang.org/x/time v0.8.0 // indirect
|
||||
golang.org/x/crypto v0.41.0 // indirect
|
||||
golang.org/x/net v0.43.0 // indirect
|
||||
golang.org/x/sys v0.35.0 // indirect
|
||||
golang.org/x/text v0.28.0 // indirect
|
||||
golang.org/x/time v0.12.0 // indirect
|
||||
)
|
||||
|
18
go.sum
18
go.sum
@@ -4,16 +4,22 @@ github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA=
|
||||
github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0=
|
||||
github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
|
||||
github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=
|
||||
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
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/labstack/echo/v4 v4.13.3 h1:pwhpCPrTl5qry5HRdM5FwdXnhXSLSY+WE+YQSeCaafY=
|
||||
github.com/labstack/echo/v4 v4.13.3/go.mod h1:o90YNEeQWjDozo584l7AwhJMHN0bOC4tAfg+Xox9q5g=
|
||||
github.com/labstack/echo/v4 v4.13.4 h1:oTZZW+T3s9gAu5L8vmzihV7/lkXGZuITzTQkTEhcXEA=
|
||||
github.com/labstack/echo/v4 v4.13.4/go.mod h1:g63b33BZ5vZzcIUF8AtRH40DrTlXnx4UMC8rBdndmjQ=
|
||||
github.com/labstack/gommon v0.4.2 h1:F8qTUNXgG1+6WQmqoUWnz8WiEU60mXVVw0P4ht1WRA0=
|
||||
github.com/labstack/gommon v0.4.2/go.mod h1:QlUFxVM+SNXhDL/Z7YhocGIBYOiwB0mXm1+1bAPHPyU=
|
||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
@@ -23,6 +29,8 @@ github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZb
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/redis/go-redis/v9 v9.7.0 h1:HhLSs+B6O021gwzl+locl0zEDnyNkxMtf/Z3NNBMa9E=
|
||||
github.com/redis/go-redis/v9 v9.7.0/go.mod h1:f6zhXITC7JUJIlPEiBOTXxJgPLdZcA93GewI7inzyWw=
|
||||
github.com/redis/go-redis/v9 v9.12.1 h1:k5iquqv27aBtnTm2tIkROUDp8JBXhXZIVu1InSgvovg=
|
||||
github.com/redis/go-redis/v9 v9.12.1/go.mod h1:huWgSWd8mW6+m0VPhJjSSQ+d6Nh1VICQ6Q5lHuCH/Iw=
|
||||
github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
|
||||
github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
|
||||
github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw=
|
||||
@@ -31,15 +39,25 @@ github.com/valyala/fasttemplate v1.2.2 h1:lxLXG0uE3Qnshl9QyaK6XJxMXlQZELvChBOCmQ
|
||||
github.com/valyala/fasttemplate v1.2.2/go.mod h1:KHLXt3tVN2HBp8eijSv/kGJopbvo7S+qRAEEKiv+SiQ=
|
||||
golang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=
|
||||
golang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=
|
||||
golang.org/x/crypto v0.41.0 h1:WKYxWedPGCTVVl5+WHSSrOBT0O8lx32+zxmHxijgXp4=
|
||||
golang.org/x/crypto v0.41.0/go.mod h1:pO5AFd7FA68rFak7rOAGVuygIISepHftHnr8dr6+sUc=
|
||||
golang.org/x/net v0.33.0 h1:74SYHlV8BIgHIFC/LrYkOGIwL19eTYXQ5wc6TBuO36I=
|
||||
golang.org/x/net v0.33.0/go.mod h1:HXLR5J+9DxmrqMwG9qjGCxZ+zKXxBru04zlTvWlWuN4=
|
||||
golang.org/x/net v0.43.0 h1:lat02VYK2j4aLzMzecihNvTlJNQUq316m2Mr9rnM6YE=
|
||||
golang.org/x/net v0.43.0/go.mod h1:vhO1fvI4dGsIjh73sWfUVjj3N7CA9WkKJNQm2svM6Jg=
|
||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=
|
||||
golang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
||||
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
||||
golang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=
|
||||
golang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=
|
||||
golang.org/x/text v0.28.0 h1:rhazDwis8INMIwQ4tpjLDzUhx6RlXqZNPEM0huQojng=
|
||||
golang.org/x/text v0.28.0/go.mod h1:U8nCwOR8jO/marOQ0QbDiOngZVEBB7MAiitBuMjXiNU=
|
||||
golang.org/x/time v0.8.0 h1:9i3RxcPv3PZnitoVGMPDKZSq1xW1gK1Xy3ArNOGZfEg=
|
||||
golang.org/x/time v0.8.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
||||
golang.org/x/time v0.12.0 h1:ScB/8o8olJvc+CQPWrK3fPZNfh7qgwCrY0zJmoEQLSE=
|
||||
golang.org/x/time v0.12.0/go.mod h1:CDIdPxbZBQxdj6cxyCIdrNogrJKMJ7pr37NYpMcMDSg=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
104
http/handler.go
104
http/handler.go
@@ -1,110 +1,26 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/labstack/echo/v4"
|
||||
client "github.com/siteworxpro/top-wallpaper/reddit"
|
||||
"github.com/siteworxpro/top-wallpaper/resize"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/siteworxpro/top-wallpaper/redis"
|
||||
)
|
||||
|
||||
func Get(c echo.Context) error {
|
||||
cc := c.(*CustomContext)
|
||||
|
||||
var size int
|
||||
var err error
|
||||
|
||||
if sizeS, ok := os.LookupEnv("MAX_SIZE"); ok {
|
||||
size, err = strconv.Atoi(sizeS)
|
||||
if err != nil {
|
||||
size = 1200
|
||||
}
|
||||
} else {
|
||||
size = 1200
|
||||
}
|
||||
|
||||
if size < 100 {
|
||||
size = 1200
|
||||
}
|
||||
|
||||
var latestImageVal string
|
||||
if cc.redis != nil {
|
||||
latestImage := cc.redis.Get(context.TODO(), "latestImage")
|
||||
latestImageVal, err = latestImage.Result()
|
||||
}
|
||||
|
||||
if err != nil || latestImageVal == "" {
|
||||
c.Logger().Info("Fetching latest image")
|
||||
latestImageVal, err = client.GetLatestImage(func(url string) (*http.Response, error) {
|
||||
return http.Get(url)
|
||||
})
|
||||
if err != nil {
|
||||
return c.String(http.StatusInternalServerError, "Error fetching latest image")
|
||||
}
|
||||
|
||||
if cc.redis != nil {
|
||||
cmd := cc.redis.Set(context.TODO(), "latestImage", latestImageVal, 600*time.Second)
|
||||
if cmd.Err() != nil {
|
||||
c.Logger().Warn("could not cache image")
|
||||
}
|
||||
}
|
||||
} else {
|
||||
c.Logger().Info("Image name fetched from cache")
|
||||
}
|
||||
|
||||
var imageData string
|
||||
if cc.redis != nil {
|
||||
latestImageBin := cc.redis.Get(context.TODO(), "latestImage:bin:"+latestImageVal)
|
||||
imageData = latestImageBin.Val()
|
||||
}
|
||||
|
||||
if imageData != "" {
|
||||
c.Logger().Info("Image data fetched from cache")
|
||||
|
||||
return c.Blob(http.StatusOK, "image/jpeg", []byte(imageData))
|
||||
}
|
||||
|
||||
response, err := http.Get(latestImageVal)
|
||||
rc, err := redis.NewRedis()
|
||||
if err != nil {
|
||||
return c.String(http.StatusInternalServerError, "Error fetching image")
|
||||
return c.String(http.StatusInternalServerError, "Internal Server Error: "+err.Error())
|
||||
}
|
||||
|
||||
if response.StatusCode != http.StatusOK {
|
||||
return c.String(http.StatusInternalServerError, "Error fetching image")
|
||||
val, err := rc.Get(redis.CacheKey)
|
||||
|
||||
if err != nil || val == "" {
|
||||
return c.NoContent(http.StatusNoContent)
|
||||
}
|
||||
|
||||
defer func(Body io.ReadCloser) {
|
||||
_ = Body.Close()
|
||||
}(response.Body)
|
||||
|
||||
imageDataBytes, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
return c.String(http.StatusInternalServerError, "Error fetching image")
|
||||
}
|
||||
|
||||
imageData = string(imageDataBytes)
|
||||
resized, err := resize.Shrink(imageData, uint(size), 70)
|
||||
|
||||
if err != nil {
|
||||
return c.String(http.StatusInternalServerError, "Error resizing image")
|
||||
}
|
||||
|
||||
go func(data string) {
|
||||
if cc.redis == nil {
|
||||
return
|
||||
}
|
||||
|
||||
_, err = cc.redis.Set(context.TODO(), "latestImage:bin:"+latestImageVal, data, 600*time.Second).Result()
|
||||
if err != nil {
|
||||
c.Logger().Warn("could not cache image")
|
||||
}
|
||||
}(resized)
|
||||
|
||||
c.Response().Header().Set("Cache-Control", "public, max-age=600")
|
||||
|
||||
return c.Blob(http.StatusOK, "image/jpeg", []byte(resized))
|
||||
return c.Blob(http.StatusOK, "image/jpeg", []byte(val))
|
||||
}
|
||||
|
@@ -1,59 +0,0 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/redis/go-redis/v9"
|
||||
"os"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type CustomContext struct {
|
||||
echo.Context
|
||||
redis *redis.Client
|
||||
}
|
||||
|
||||
func GetCustomContext(next echo.HandlerFunc) echo.HandlerFunc {
|
||||
return func(c echo.Context) error {
|
||||
redisUrl := os.Getenv("REDIS_URL")
|
||||
if redisUrl == "" {
|
||||
c.Logger().Warn("REDIS_URL not set, skipping redis connection. Use REDIS_URL to cache the image")
|
||||
|
||||
return next(&CustomContext{c, nil})
|
||||
}
|
||||
|
||||
redisPort := os.Getenv("REDIS_PORT")
|
||||
if redisPort == "" {
|
||||
redisPort = "6379"
|
||||
}
|
||||
|
||||
redisDb := os.Getenv("REDIS_DB")
|
||||
if redisDb == "" {
|
||||
redisDb = "0"
|
||||
}
|
||||
|
||||
redisDbInt, err := strconv.ParseInt(redisDb, 10, 64)
|
||||
if err != nil {
|
||||
c.Logger().Warn("REDIS_DB is not a valid integer, skipping redis connection. Use REDIS_DB to cache the image")
|
||||
|
||||
return next(&CustomContext{})
|
||||
}
|
||||
|
||||
cc := &CustomContext{c, redis.NewClient(&redis.Options{
|
||||
Addr: fmt.Sprintf("%s:%s", redisUrl, redisPort),
|
||||
Password: os.Getenv("REDIS_PASSWORD"),
|
||||
DB: int(redisDbInt),
|
||||
})}
|
||||
|
||||
cmd := cc.redis.Ping(context.Background())
|
||||
|
||||
if cmd.Err() != nil {
|
||||
c.Logger().Warn("could not connect to redis")
|
||||
|
||||
cc.redis = nil
|
||||
}
|
||||
|
||||
return next(cc)
|
||||
}
|
||||
}
|
35
main.go
35
main.go
@@ -1,23 +1,39 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/labstack/echo/v4/middleware"
|
||||
"github.com/labstack/gommon/log"
|
||||
tphttp "github.com/siteworxpro/top-wallpaper/http"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
"github.com/siteworxpro/top-wallpaper/reddit"
|
||||
"github.com/siteworxpro/top-wallpaper/redis"
|
||||
)
|
||||
|
||||
func main() {
|
||||
|
||||
e := echo.New()
|
||||
e.Logger.SetLevel(log.INFO)
|
||||
ctx, fn := context.WithCancel(context.Background())
|
||||
defer fn()
|
||||
|
||||
rc, err := redis.NewRedis()
|
||||
|
||||
if err != nil {
|
||||
log.Error("Could not initialize Redis client: ", err)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
e := echo.New()
|
||||
e.AcquireContext().Set("redisClient", rc)
|
||||
e.Logger.SetOutput(os.Stdout)
|
||||
|
||||
e.Logger.SetLevel(log.INFO)
|
||||
e.HideBanner = true
|
||||
|
||||
e.Use(tphttp.GetCustomContext)
|
||||
e.Use(middleware.Logger())
|
||||
e.Use(middleware.RequestID())
|
||||
e.Use(middleware.CORSWithConfig(middleware.CORSConfig{
|
||||
@@ -34,12 +50,17 @@ func main() {
|
||||
e.GET(path, tphttp.Get)
|
||||
|
||||
e.Logger.Info("Starting server at path " + path)
|
||||
// start the server
|
||||
|
||||
port := os.Getenv("PORT")
|
||||
if port == "" {
|
||||
port = "8080"
|
||||
}
|
||||
|
||||
ctx = redis.WithContext(ctx, rc)
|
||||
|
||||
ctx = context.WithValue(ctx, "logger", e.Logger) //nolint:staticcheck
|
||||
|
||||
go reddit.Fetch(ctx)
|
||||
|
||||
e.Logger.Fatal(e.Start(":" + port))
|
||||
}
|
||||
|
@@ -60,15 +60,16 @@ func GetLatestImage(httpGet func(url string) (*http.Response, error)) (string, e
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer func(Body io.ReadCloser) {
|
||||
_ = Body.Close()
|
||||
}(response.Body)
|
||||
defer response.Body.Close()
|
||||
|
||||
if response.StatusCode != http.StatusOK {
|
||||
return "", fmt.Errorf("error fetching reddit data: %s", response.Status)
|
||||
}
|
||||
|
||||
jsonBytes, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
redditData := redditResponse{}
|
||||
err = json.Unmarshal(jsonBytes, &redditData)
|
||||
|
148
reddit/fetcher.go
Normal file
148
reddit/fetcher.go
Normal file
@@ -0,0 +1,148 @@
|
||||
package reddit
|
||||
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/labstack/echo/v4"
|
||||
"github.com/siteworxpro/top-wallpaper/redis"
|
||||
"github.com/siteworxpro/top-wallpaper/resize"
|
||||
)
|
||||
|
||||
func Fetch(ctx context.Context) {
|
||||
const defaultImageSize = 1600
|
||||
const minImageSize = 100
|
||||
|
||||
rc, err := redis.FromContext(ctx)
|
||||
if err != nil {
|
||||
panic("Redis client not found in context: " + err.Error())
|
||||
}
|
||||
|
||||
l := ctx.Value("logger").(echo.Logger)
|
||||
|
||||
var size int
|
||||
if sizeS, ok := os.LookupEnv("MAX_SIZE"); ok {
|
||||
size, err = strconv.Atoi(sizeS)
|
||||
if err != nil {
|
||||
size = defaultImageSize
|
||||
}
|
||||
} else {
|
||||
size = defaultImageSize
|
||||
}
|
||||
|
||||
if size < minImageSize {
|
||||
size = defaultImageSize
|
||||
}
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
|
||||
l.Info("Stopping Reddit fetcher...")
|
||||
_ = rc.Close()
|
||||
|
||||
return
|
||||
default:
|
||||
|
||||
val, err := rc.Get(redis.CacheKey)
|
||||
if err != nil {
|
||||
l.Error("Error fetching from Redis: ", err)
|
||||
break
|
||||
}
|
||||
|
||||
if val != "" {
|
||||
l.Info("Reddit image fetched from cache...")
|
||||
|
||||
break
|
||||
}
|
||||
|
||||
l.Info("Fetching latest image from Reddit...")
|
||||
|
||||
latestImageVal, err := GetLatestImage(func(u string) (*http.Response, error) {
|
||||
|
||||
request := &http.Request{
|
||||
Method: http.MethodGet,
|
||||
Header: http.Header{
|
||||
"User-Agent": []string{"Mozilla/5.0 (compatible; RedditBot/1.0; +https://www.reddit.com/wiki/redditauth)"},
|
||||
},
|
||||
URL: &url.URL{
|
||||
Scheme: "https",
|
||||
Host: "www.reddit.com",
|
||||
Path: strings.TrimPrefix(u, "https://www.reddit.com"),
|
||||
},
|
||||
}
|
||||
|
||||
httpClient := &http.Client{
|
||||
Timeout: 10 * time.Second,
|
||||
}
|
||||
|
||||
response, e := httpClient.Do(request)
|
||||
|
||||
if e != nil {
|
||||
l.Error("Error fetching image from Reddit: ", e)
|
||||
return nil, e
|
||||
}
|
||||
|
||||
if response.StatusCode != http.StatusOK {
|
||||
l.Error("Error fetching image from Reddit: ", response.Status)
|
||||
return nil, &echo.HTTPError{
|
||||
Code: response.StatusCode,
|
||||
Message: "Error fetching image from Reddit",
|
||||
}
|
||||
}
|
||||
|
||||
return response, nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
l.Error("Error getting latest image URL: ", err)
|
||||
break
|
||||
}
|
||||
|
||||
response, err := http.Get(latestImageVal)
|
||||
if err != nil {
|
||||
l.Error("Error fetching image from Reddit: ", err)
|
||||
|
||||
break
|
||||
}
|
||||
|
||||
if response.StatusCode != http.StatusOK {
|
||||
l.Error("Error fetching image from Reddit: ", response.Status)
|
||||
|
||||
break
|
||||
}
|
||||
|
||||
imageDataBytes, err := io.ReadAll(response.Body)
|
||||
if err != nil {
|
||||
l.Error("Error reading image data: ", err)
|
||||
_ = response.Body.Close()
|
||||
break
|
||||
}
|
||||
|
||||
_ = response.Body.Close()
|
||||
|
||||
imageData := string(imageDataBytes)
|
||||
resized, err := resize.Shrink(imageData, uint(size), 70)
|
||||
|
||||
if err != nil {
|
||||
l.Error("Error resizing image: ", err)
|
||||
break
|
||||
}
|
||||
|
||||
err = rc.Set(redis.CacheKey, resized, 10*time.Minute)
|
||||
if err != nil {
|
||||
l.Warn("could not cache image")
|
||||
}
|
||||
|
||||
l.Info("Reddit image fetched and resized successfully, cached for 10 minutes.")
|
||||
}
|
||||
|
||||
time.Sleep(10 * time.Minute)
|
||||
}
|
||||
}
|
119
redis/client.go
Normal file
119
redis/client.go
Normal file
@@ -0,0 +1,119 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/redis/go-redis/v9"
|
||||
)
|
||||
|
||||
type contextKey string
|
||||
|
||||
const redisKey contextKey = "redisClient"
|
||||
|
||||
const CacheKey = "top-wallpaper:latestImage"
|
||||
|
||||
type Cache interface {
|
||||
Get(key string) (string, error)
|
||||
Set(key string, value string, expiration int64) error
|
||||
Close() error
|
||||
}
|
||||
|
||||
type Redis struct {
|
||||
client *redis.Client
|
||||
}
|
||||
|
||||
func NewRedis() (*Redis, error) {
|
||||
|
||||
redisUrl := os.Getenv("REDIS_URL")
|
||||
if redisUrl == "" {
|
||||
return nil, errors.New("REDIS_URL not set, cannot initialize Redis client")
|
||||
}
|
||||
|
||||
redisPort := os.Getenv("REDIS_PORT")
|
||||
if redisPort == "" {
|
||||
redisPort = "6379"
|
||||
}
|
||||
|
||||
redisDb := os.Getenv("REDIS_DB")
|
||||
if redisDb == "" {
|
||||
redisDb = "0"
|
||||
}
|
||||
|
||||
redisDbInt, err := strconv.ParseInt(redisDb, 10, 64)
|
||||
if err != nil {
|
||||
return nil, errors.New("REDIS_DB is not a valid integer, cannot initialize Redis client")
|
||||
}
|
||||
|
||||
rc := redis.NewClient(&redis.Options{
|
||||
Addr: fmt.Sprintf("%s:%s", redisUrl, redisPort),
|
||||
Password: os.Getenv("REDIS_PASSWORD"),
|
||||
DB: int(redisDbInt),
|
||||
})
|
||||
|
||||
cmd := rc.Ping(context.Background())
|
||||
|
||||
if cmd.Err() != nil {
|
||||
return nil, fmt.Errorf("could not connect to Redis: %v", cmd.Err())
|
||||
}
|
||||
if cmd.Val() != "PONG" {
|
||||
return nil, fmt.Errorf("unexpected response from Redis: %s", cmd.Val())
|
||||
}
|
||||
|
||||
return &Redis{
|
||||
client: rc,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (r *Redis) Get(key string) (string, error) {
|
||||
cmd := r.client.Get(context.Background(), key)
|
||||
if cmd.Err() != nil {
|
||||
if errors.Is(cmd.Err(), redis.Nil) {
|
||||
return "", nil // Key does not exist
|
||||
}
|
||||
return "", fmt.Errorf("could not get value for key %s: %v", key, cmd.Err())
|
||||
}
|
||||
return cmd.Val(), nil
|
||||
}
|
||||
|
||||
func (r *Redis) Set(key string, value string, expiration time.Duration) error {
|
||||
cmd := r.client.Set(context.Background(), key, value, expiration)
|
||||
if cmd.Err() != nil {
|
||||
return fmt.Errorf("could not set value for key %s: %v", key, cmd.Err())
|
||||
}
|
||||
if cmd.Val() != "OK" {
|
||||
return fmt.Errorf("unexpected response from Redis when setting key %s: %s", key, cmd.Val())
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Redis) Close() error {
|
||||
if r.client == nil {
|
||||
return nil
|
||||
}
|
||||
err := r.client.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("could not close Redis client: %v", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func WithContext(ctx context.Context, r *Redis) context.Context {
|
||||
return context.WithValue(ctx, redisKey, r)
|
||||
}
|
||||
|
||||
func FromContext(ctx context.Context) (*Redis, error) {
|
||||
r, ok := ctx.Value(redisKey).(*Redis)
|
||||
if !ok {
|
||||
return nil, errors.New("no Redis client found in context")
|
||||
}
|
||||
if r == nil {
|
||||
return nil, errors.New("redis client is nil")
|
||||
}
|
||||
return r, nil
|
||||
}
|
@@ -3,10 +3,11 @@ package resize
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"github.com/nfnt/resize"
|
||||
i "image"
|
||||
"image/jpeg"
|
||||
_ "image/png"
|
||||
|
||||
"github.com/nfnt/resize"
|
||||
)
|
||||
|
||||
func Shrink(image string, maxSize uint, quality int) (string, error) {
|
||||
|
Reference in New Issue
Block a user