package driver import ( "errors" "fmt" e "git.xdrm.io/go/aicra/err" "git.xdrm.io/go/aicra/response" "net/http" "plugin" "strings" ) // Name returns the driver name func (d *Import) Name() string { return "import" } // RegisterController registers a new controller func (d *Import) RegisterController(_path string, _controller Controller) error { // 1. init controllers if not already if d.Controllers == nil { d.Controllers = make(map[string]Controller) } // 2. fail if no controller if _controller == nil { return errors.New("nil controller") } // 3. Fail on invalid path if len(_path) < 1 || _path[0] != '/' { return errors.New("invalid controller path") } // 4. Store controller d.Controllers[_path] = _controller return nil } // RegisterMiddleware registers a new controller func (d *Import) RegisterMiddlware(_path string, _middleware Middleware) error { // 1. init imddlewares if not already if d.Middlewares == nil { d.Middlewares = make(map[string]Middleware) } // 2. fail if no imddleware if _middleware == nil { return errors.New("nil imddleware") } // 3. Fail on invalid path if len(_path) < 1 || _path[0] != '/' { return errors.New("invalid imddleware path") } // 4. Store imddleware d.Middlewares[_path] = _middleware return nil } // RunController implements the Driver interface func (d *Import) RunController(_path []string, _method string) (func(response.Arguments) response.Response, err.Error) { /* (1) Build controller path */ path := strings.Join(_path, "-") /* (2) Check if controller exists */ controller, ok := d.Controllers[path] if !ok { return nil, e.UncallableController } /* (3) Format method */ method := strings.ToLower(_method) /* (4) Return method according to method */ switch method { case "GET": return controller.Get, e.Success case "POST": return controller.Post, e.Success case "PUT": return controller.Put, e.Success case "DELETE": return controller.Delete, e.Success default: return nil, e.UncallableMethod } return nil, e.UncallableController } // LoadMiddleware returns a new middleware function; it must be a // valid and existing folder/filename file with the .so extension func (d *Import) LoadMiddleware(_path string) (func(http.Request, *[]string), error) { /* (1) Check plugin name */ if len(_path) < 1 { return nil, errors.New("middleware name must not be empty") } /* (2) Check if middleware exists */ middleware, ok := d.Middlewares[_path] if !ok { return nil, errors.New("middleware not found") } /* (3) Return middleware */ return mware, nil }