Add cross domain plugin

cmd/proxy/proxy.go

@@ -60,6 +60,8 @@ var (
 
  // internal plugin configuration file
  jwtCfg = flag.String("jwt", "", "Plugin(JWT): jwt plugin configuration file, json format")
+ // crossCfg
+ crossCfg = flag.String("cross", "", "Plugin(CROSS): cross plugin configuration file, json format")
 
  // metric
  metricJob = flag.String("metric-job", "", "prometheus job name")
@@ -170,6 +172,7 @@ func getCfg() *proxy.Cfg {
  cfg.Option.LimitTimeoutWrite = time.Second * time.Duration(*limitTimeoutWriteSec)
  cfg.Option.LimitIntervalHeathCheck = time.Second * time.Duration(*limitIntervalHeathCheckSec)
  cfg.Option.JWTCfgFile = *jwtCfg
+ cfg.Option.CrossCfgFile = *crossCfg
  cfg.Option.EnableWebSocket = *enableWebSocket
  cfg.Option.EnableJSPlugin = *enableJSPlugin
  cfg.Option.DisableHeaderNameNormalizing = *disableHeaderNameNormalizing

pkg/proxy/cfg.go

@@ -24,7 +24,8 @@ type Option struct {
  LimitBytesBody int
  LimitBytesCaching uint64
 
- JWTCfgFile string
+ JWTCfgFile string
+ CrossCfgFile string
 
  EnableWebSocket bool
  EnableJSPlugin bool

pkg/proxy/factory.go

@@ -38,6 +38,8 @@ const (
  FilterCaching = "CACHING"
  // FilterJWT jwt filter
  FilterJWT = "JWT"
+ // FilterCross cross filter
+ FilterCross = "CROSS"
  // FilterJSPlugin js plugin engine
  FilterJSPlugin = "JS-ENGINE"
 )
@@ -74,6 +76,8 @@ func (p *Proxy) newFilter(filterSpec *FilterSpec) (filter.Filter, error) {
  return newCachingFilter(p.cfg.Option.LimitBytesCaching, p.dispatcher.tw), nil
  case FilterJWT:
  return newJWTFilter(p.cfg.Option.JWTCfgFile)
+ case FilterCross:
+ return newCrossDomainFilter(p.cfg.Option.CrossCfgFile)
  case FilterJSPlugin:
  return p.jsEngine, nil
  default:

pkg/proxy/filter_cross_domain.go

@@ -0,0 +1,76 @@
+package proxy
+
+import (
+ "bytes"
+ "encoding/json"
+ "io/ioutil"
+
+ "github.com/fagongzi/gateway/pkg/filter"
+ "github.com/valyala/fasthttp"
+)
+
+var (
+ options = []byte("OPTIONS")
+)
+
+// CrossCfg cross cfg
+type CrossCfg struct {
+ Headers []CrossHeader `json:"headers"`
+}
+
+// CrossHeader cross header
+type CrossHeader struct {
+ Name string `json:"name"`
+ Value string `json:"value"`
+}
+
+// CrossDomainFilter cross domain
+type CrossDomainFilter struct {
+ filter.BaseFilter
+ cfg CrossCfg
+}
+
+func newCrossDomainFilter(file string) (filter.Filter, error) {
+ f := &CrossDomainFilter{}
+
+ err := f.parseCfg(file)
+ if err != nil {
+ return nil, err
+ }
+
+ return f, nil
+}
+
+func (f *CrossDomainFilter) parseCfg(file string) error {
+ data, err := ioutil.ReadFile(file)
+ if err != nil {
+ return err
+ }
+
+ err = json.Unmarshal(data, &f.cfg)
+ if err != nil {
+ return err
+ }
+
+ return nil
+}
+
+// Name return name of this filter
+func (f *CrossDomainFilter) Name() string {
+ return FilterCross
+}
+
+// Pre execute before proxy
+func (f *CrossDomainFilter) Pre(c filter.Context) (statusCode int, err error) {
+ if bytes.Compare(c.OriginRequest().Method(), options) != 0 {
+ return f.BaseFilter.Pre(c)
+ }
+
+ resp := fasthttp.AcquireResponse()
+ for _, h := range f.cfg.Headers {
+ resp.Header.Add(h.Name, h.Value)
+ }
+
+ c.SetAttr(filter.AttrUsingResponse, resp)
+ return filter.BreakFilterChainCode, nil
+}