enhance hook for context

Signed-off-by: wzhfang <wzhfang@wzhfang.(none)>