mirror of
https://github.com/eolinker/apinto
synced 2025-10-20 07:34:36 +08:00
83 lines
1.6 KiB
Go
83 lines
1.6 KiB
Go
package ip_restriction
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/eolinker/eosc"
|
|
http_service "github.com/eolinker/eosc/http-service"
|
|
)
|
|
|
|
type IPHandler struct {
|
|
*Driver
|
|
id string
|
|
responseType string
|
|
filter IPFilter
|
|
}
|
|
|
|
func (I *IPHandler) doRestriction(ctx http_service.IHttpContext) error {
|
|
realIP := ctx.Request().ReadIP()
|
|
if I.filter != nil {
|
|
ok, err := I.filter(realIP)
|
|
if !ok {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (I *IPHandler) Id() string {
|
|
return I.id
|
|
}
|
|
|
|
func (I *IPHandler) Start() error {
|
|
return nil
|
|
}
|
|
|
|
func (I *IPHandler) Reset(conf interface{}, workers map[eosc.RequireId]interface{}) error {
|
|
confObj, err := I.check(conf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
I.filter = confObj.genFilter()
|
|
I.responseType = confObj.ResponseType
|
|
return nil
|
|
}
|
|
|
|
func (I *IPHandler) Stop() error {
|
|
return nil
|
|
}
|
|
|
|
func (I *IPHandler) CheckSkill(skill string) bool {
|
|
return http_service.FilterSkillName == skill
|
|
}
|
|
|
|
func (I *IPHandler) responseEncode(origin string, statusCode int) string {
|
|
if I.responseType == "json" {
|
|
tmp := map[string]interface{}{
|
|
"message": origin,
|
|
"status_code": statusCode,
|
|
}
|
|
newInfo, _ := json.Marshal(tmp)
|
|
return string(newInfo)
|
|
}
|
|
return origin
|
|
}
|
|
func (I *IPHandler) Destroy() {
|
|
I.filter = nil
|
|
I.responseType = ""
|
|
}
|
|
|
|
func (I *IPHandler) DoFilter(ctx http_service.IHttpContext, next http_service.IChain) error {
|
|
err := I.doRestriction(ctx)
|
|
if err != nil {
|
|
resp := ctx.Response()
|
|
info := I.responseEncode(err.Error(), 403)
|
|
resp.SetStatus(403, "403")
|
|
resp.SetBody([]byte(info))
|
|
return err
|
|
}
|
|
if next != nil {
|
|
return next.DoChain(ctx)
|
|
}
|
|
return nil
|
|
}
|