Hunter0x7c7
2022-08-11 a82f9cb69f63aaeba40c024960deda7d75b9fece
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package policy
 
import (
    "context"
 
    "github.com/v2fly/v2ray-core/v5/common"
    "github.com/v2fly/v2ray-core/v5/features/policy"
)
 
// Instance is an instance of Policy manager.
type Instance struct {
    levels map[uint32]*Policy
    system *SystemPolicy
}
 
// New creates new Policy manager instance.
func New(ctx context.Context, config *Config) (*Instance, error) {
    m := &Instance{
        levels: make(map[uint32]*Policy),
        system: config.System,
    }
    if len(config.Level) > 0 {
        for lv, p := range config.Level {
            pp := defaultPolicy()
            pp.overrideWith(p)
            m.levels[lv] = pp
        }
    }
 
    return m, nil
}
 
// Type implements common.HasType.
func (*Instance) Type() interface{} {
    return policy.ManagerType()
}
 
// ForLevel implements policy.Manager.
func (m *Instance) ForLevel(level uint32) policy.Session {
    if p, ok := m.levels[level]; ok {
        return p.ToCorePolicy()
    }
    return policy.SessionDefault()
}
 
// ForSystem implements policy.Manager.
func (m *Instance) ForSystem() policy.System {
    if m.system == nil {
        return policy.System{}
    }
    return m.system.ToCorePolicy()
}
 
// Start implements common.Runnable.Start().
func (m *Instance) Start() error {
    return nil
}
 
// Close implements common.Closable.Close().
func (m *Instance) Close() error {
    return nil
}
 
func init() {
    common.Must(common.RegisterConfig((*Config)(nil), func(ctx context.Context, config interface{}) (interface{}, error) {
        return New(ctx, config.(*Config))
    }))
}