Merge pull request #1 from steemax/paa.init

v0.1

Author: steemax

.traefik.yml

@@ -1,12 +1,20 @@
-displayName: Demo Plugin
+# The name of your plugin as displayed in the Plugins Catalog web UI.
+displayName: Rate Limit
+
+# For now, `middleware` is the only type available.
 type: middleware
-iconPath: .assets/icon.png
 
-import: github.com/traefik/plugindemo
+# The import path of your plugin.
+import: github.com/kav789/traefik-ratelimit
+
+# A brief description of what your plugin is doing.
+summary: Rate Limit
 
-summary: '[Demo] Add Request Header'
+# Medias associated to the plugin (optional)
+iconPath:  .assets/icon.png
 
+# Configuration data for your plugin.
+# This is mandatory,
+# and Plugins Catalog will try to execute the plugin with the data you provide as part of its startup validity tests.
 testData:
-  Headers:
-    X-Demo: test
-    X-URL: '{{URL}}'
+  rate: 100

demo.go

@@ -1,3 +1,3 @@
-module github.com/traefik/plugindemo
+module github.com/kav789/traefik-ratelimit
 
-go 1.19
+go 1.18

demo_test.go

@@ -0,0 +1,54 @@
+package traefik_ratelimit
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"os"
+)
+
+func CreateConfig() *Config {
+	return &Config{}
+}
+
+type Config struct {
+	Rate int `json:"rate,omitempty"`
+}
+
+type RateLimit struct {
+	name   string
+	next   http.Handler
+	rate   int
+	config *Config
+}
+
+func New(ctx context.Context, next http.Handler, config *Config, name string) (http.Handler, error) {
+	mlog(fmt.Sprintf("config %v", config))
+	return &RateLimit{
+		name:   name,
+		next:   next,
+		config: config,
+	}, nil
+}
+
+func (r *RateLimit) Allow(ctx context.Context, req *http.Request, rw http.ResponseWriter) bool {
+	return true
+}
+
+func (r *RateLimit) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	encoder := json.NewEncoder(rw)
+	reqCtx := req.Context()
+	if r.Allow(reqCtx, req, rw) {
+		r.next.ServeHTTP(rw, req)
+		return
+	}
+	rw.Header().Set("Content-Type", "application/json")
+	rw.WriteHeader(http.StatusTooManyRequests)
+	encoder.Encode(map[string]any{"status_code": http.StatusTooManyRequests, "message": "rate limit exceeded, try again later"})
+	return
+}
+
+func mlog(args ...any) {
+	os.Stdout.WriteString(fmt.Sprintf("[rate-limit-middleware-plugin] %s\n", fmt.Sprint(args...)))
+}

go.mod

@@ -0,0 +1,8 @@
+package traefik_ratelimit
+
+import (
+	"testing"
+)
+
+func TestMain(t *testing.T) {
+}