🎉 Initial Commit

Signed-off-by: Teakowa <27560638+Teakowa@users.noreply.github.com>

Author: Teakowa

.github/workflows/ci.yml

@@ -0,0 +1,17 @@
+name: Continuous Integration
+on:
+  push:
+    branches:
+      - '**'
+  pull_request:
+    branches:
+      - '**'
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-go@v3
+        with:
+          go-version: 1.20
+      - uses: golangci/golangci-lint-action@v3

.github/workflows/image.yml

@@ -0,0 +1,66 @@
+name: Images
+
+on:
+  schedule:
+    - cron: '0 10 * * *' # everyday at 10am UTC
+  release:
+    types: [ published ]
+  push:
+    branches:
+      - main
+    tags:
+      - 'v*'
+    paths-ignore:
+      - '.github'
+  workflow_dispatch:
+
+jobs:
+  docker:
+    name: "Build & Push"
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v3
+      -
+        name: Set up QEMU
+        uses: docker/setup-qemu-action@v2
+      -
+        name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v2
+
+      - name: Split repo name
+        uses: jungwinter/split@v2
+        id: split_reponame
+        with:
+          separator: '/'
+          msg: ${{ github.repository }}
+
+      - name: Setup repo name to lowercase
+        id: reponame
+        uses: ASzc/change-string-case-action@v5
+        with:
+          string: ${{ steps.split_reponame.outputs._1 }}
+
+      - name: Log in to the Container registry
+        uses: docker/login-action@v2
+        with:
+          registry: ${{ secrets.REGISTRY }}
+          username: ${{ secrets.AWS_ACCESS_KEY_ID }}
+          password: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
+
+      - name: Extract metadata (tags, labels) for Docker
+        id: meta
+        uses: docker/metadata-action@v4
+        with:
+          images: ${{ secrets.REGISTRY }}/${{ steps.reponame.outputs.lowercase }}
+
+      - name: Build and push Docker image
+        uses: docker/build-push-action@v4
+        with:
+          context: .
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

.github/workflows/release.yml

@@ -0,0 +1,36 @@
+name: Release
+
+on:
+  push:
+    branches:
+      - main
+      - rc
+      - beta
+      - alpha
+  workflow_dispatch:
+
+env:
+  GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+permissions:
+  contents: write
+
+jobs:
+  release:
+    name: Release
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+
+      - uses: actions/setup-go@v3
+        with:
+          go-version: 1.19
+
+      - uses: go-semantic-release/action@v1
+        with:
+          hooks: goreleaser
+        env:
+          GITHUB_TOKEN: ${{ secrets.ACTIONS_BOT_TOKEN }}

Dockerfile

@@ -0,0 +1,31 @@
+FROM golang:1.20-alpine as builder
+
+ENV CGO_ENABLED 0
+ENV GO111MODULE=on
+
+WORKDIR /app
+
+RUN apk update --no-cache && apk add --no-cache tzdata
+
+COPY . ./
+
+RUN go mod download
+
+RUN go build -o main .
+
+FROM alpine
+
+WORKDIR /app
+
+COPY --from=builder /usr/share/zoneinfo/Asia/Shanghai /usr/share/zoneinfo/Asia/Shanghai
+COPY --from=builder /usr/share/zoneinfo/Asia/Shanghai /etc/localtime
+COPY --from=builder /app/main /app/main
+
+RUN apk update --no-cache && apk add --no-cache ca-certificates
+RUN set -ex; \
+    apk add --no-cache supervisor && \
+    echo "Asia/Shanghai" > /etc/timezone && \
+    rm -rf /var/cache/apk/*
+
+EXPOSE 8080
+ENTRYPOINT ["/app/main"]

go.mod

@@ -0,0 +1,3 @@
+module proxy
+
+go 1.20

main.go

@@ -0,0 +1,60 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httputil"
+	"net/url"
+)
+
+func main() {
+	http.HandleFunc("/ping", func(w http.ResponseWriter, r *http.Request) {
+		_, err := w.Write([]byte("pong"))
+		if err != nil {
+			return
+		}
+	})
+
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		targetUrl := "https://api.openai.com"
+		target, err := url.Parse(targetUrl)
+		if err != nil {
+			http.Error(w, "Internal Server Error", http.StatusInternalServerError)
+			return
+		}
+		r.Header.Set("Authorization", r.Header.Get("Authorization"))
+		r.Header.Set("Content-Type", "application/json")
+		r.Host = r.URL.Host
+
+		proxy := httputil.NewSingleHostReverseProxy(target)
+
+		if r.Header.Get("Accept") == "text/event-stream" {
+			handleSSE(w, r, proxy)
+		} else {
+			proxy.ServeHTTP(w, r)
+		}
+	})
+
+	err := http.ListenAndServe(":8080", nil)
+	if err != nil {
+		return
+	}
+}
+
+func handleSSE(w http.ResponseWriter, r *http.Request, proxy *httputil.ReverseProxy) {
+	w.Header().Set("Content-Type", "text/event-stream")
+	w.Header().Set("Cache-Control", "no-cache")
+	w.Header().Set("Connection", "keep-alive")
+
+	proxy.Transport = &http.Transport{
+		Proxy: http.ProxyFromEnvironment,
+	}
+	// Start proxying SSE request to OpenAI API
+	done := make(chan bool)
+	go func() {
+		proxy.ServeHTTP(w, r)
+		done <- true
+	}()
+
+	// Wait for SSE request to finish
+	<-done
+}