2015-12-05 16:55:45 -05:00
|
|
|
package app
|
|
|
|
|
2016-05-18 11:12:04 -04:00
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
|
2016-08-20 14:55:45 -04:00
|
|
|
"v2ray.com/core/common"
|
2016-05-18 11:12:04 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2016-06-27 02:53:35 -04:00
|
|
|
ErrMissingApplication = errors.New("App: Failed to found one or more applications.")
|
2016-05-18 11:12:04 -04:00
|
|
|
)
|
|
|
|
|
2016-01-31 11:01:28 -05:00
|
|
|
type ID int
|
|
|
|
|
2015-12-11 06:01:20 -05:00
|
|
|
// Context of a function call from proxy to app.
|
2015-12-10 17:55:39 -05:00
|
|
|
type Context interface {
|
|
|
|
CallerTag() string
|
|
|
|
}
|
|
|
|
|
2016-05-18 02:05:52 -04:00
|
|
|
type Caller interface {
|
|
|
|
Tag() string
|
|
|
|
}
|
|
|
|
|
2016-05-18 11:12:04 -04:00
|
|
|
type Application interface {
|
|
|
|
common.Releasable
|
|
|
|
}
|
|
|
|
|
|
|
|
type ApplicationInitializer func() error
|
|
|
|
|
2015-12-11 06:01:20 -05:00
|
|
|
// A Space contains all apps that may be available in a V2Ray runtime.
|
|
|
|
// Caller must check the availability of an app by calling HasXXX before getting its instance.
|
|
|
|
type Space interface {
|
2016-05-18 11:12:04 -04:00
|
|
|
Initialize() error
|
|
|
|
InitializeApplication(ApplicationInitializer)
|
|
|
|
|
2016-01-31 11:01:28 -05:00
|
|
|
HasApp(ID) bool
|
2016-05-18 11:12:04 -04:00
|
|
|
GetApp(ID) Application
|
|
|
|
BindApp(ID, Application)
|
2016-01-31 11:01:28 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
type spaceImpl struct {
|
2016-05-22 13:32:28 -04:00
|
|
|
cache map[ID]Application
|
|
|
|
appInit []ApplicationInitializer
|
2016-01-31 11:01:28 -05:00
|
|
|
}
|
2015-12-11 09:56:10 -05:00
|
|
|
|
2016-05-18 02:05:52 -04:00
|
|
|
func NewSpace() Space {
|
|
|
|
return &spaceImpl{
|
2016-05-22 13:32:28 -04:00
|
|
|
cache: make(map[ID]Application),
|
|
|
|
appInit: make([]ApplicationInitializer, 0, 32),
|
2016-05-18 11:12:04 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *spaceImpl) InitializeApplication(f ApplicationInitializer) {
|
2016-05-22 13:32:28 -04:00
|
|
|
this.appInit = append(this.appInit, f)
|
2016-05-18 11:12:04 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func (this *spaceImpl) Initialize() error {
|
2016-05-22 13:32:28 -04:00
|
|
|
for _, f := range this.appInit {
|
|
|
|
err := f()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-01-31 11:01:28 -05:00
|
|
|
}
|
2016-05-18 11:12:04 -04:00
|
|
|
return nil
|
2016-01-31 11:01:28 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
func (this *spaceImpl) HasApp(id ID) bool {
|
|
|
|
_, found := this.cache[id]
|
|
|
|
return found
|
|
|
|
}
|
|
|
|
|
2016-05-18 11:12:04 -04:00
|
|
|
func (this *spaceImpl) GetApp(id ID) Application {
|
2016-01-31 11:01:28 -05:00
|
|
|
obj, found := this.cache[id]
|
|
|
|
if !found {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
return obj
|
|
|
|
}
|
|
|
|
|
2016-05-18 11:12:04 -04:00
|
|
|
func (this *spaceImpl) BindApp(id ID, application Application) {
|
|
|
|
this.cache[id] = application
|
2015-12-05 16:55:45 -05:00
|
|
|
}
|