From 586f135d80784df762725a6add215cf84015f08e Mon Sep 17 00:00:00 2001
From: tbphp <admin@tbphp.net>
Date: Sun, 13 Jul 2025 04:16:17 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=9E=84=E5=BB=BA=E7=89=88=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .github/workflows/docker-build.yml | 2 ++
 Dockerfile                         | 6 ++++--
 internal/version/version.go        | 3 +++
 3 files changed, 9 insertions(+), 2 deletions(-)
 create mode 100644 internal/version/version.go

diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
index c8da614..e82b162 100644
--- a/.github/workflows/docker-build.yml
+++ b/.github/workflows/docker-build.yml
@@ -56,6 +56,8 @@ jobs:
           labels: ${{ steps.meta.outputs.labels }}
           cache-from: type=gha
           cache-to: type=gha,mode=max
+          build-args: |
+            VERSION=${{ github.ref_name }}
 
       - name: Generate artifact attestation
         if: github.event_name != 'pull_request' && steps.build.outputs.digest != ''
diff --git a/Dockerfile b/Dockerfile
index 568a645..8b93dd9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,13 +1,15 @@
 FROM node:20-alpine AS builder
 
+ARG VERSION=1.0.0
 WORKDIR /build
 COPY ./web .
 RUN npm install
-RUN npm run build
+RUN VITE_VERSION=${VERSION} npm run build
 
 
 FROM golang:alpine AS builder2
 
+ARG VERSION=1.0.0
 ENV GO111MODULE=on \
     CGO_ENABLED=0 \
     GOOS=linux
@@ -19,7 +21,7 @@ RUN go mod download
 
 COPY . .
 COPY --from=builder /build/dist ./web/dist
-RUN go build -ldflags "-s -w " -o gpt-load
+RUN go build -ldflags "-s -w -X gpt-load/internal/version.Version=${VERSION}" -o gpt-load
 
 
 FROM alpine
diff --git a/internal/version/version.go b/internal/version/version.go
new file mode 100644
index 0000000..8773a74
--- /dev/null
+++ b/internal/version/version.go
@@ -0,0 +1,3 @@
+package version
+
+var Version = "1.0.0"