@@ 20-125 (lines=106) @@ | ||
17 | ||
18 | use Kenjis\MonkeyPatch\Patcher\MethodPatcher\NodeVisitor; |
|
19 | ||
20 | class MethodPatcher extends AbstractPatcher |
|
21 | { |
|
22 | const CODE = <<<'EOL' |
|
23 | if (($__ret__ = \__PatchManager__::getReturn(__CLASS__, __FUNCTION__, func_get_args())) !== __GO_TO_ORIG__) return $__ret__; |
|
24 | EOL; |
|
25 | const CODENORET = <<<'EOL' |
|
26 | if (($__ret__ = \__PatchManager__::getReturn(__CLASS__, __FUNCTION__, func_get_args())) !== __GO_TO_ORIG__) return; |
|
27 | EOL; |
|
28 | ||
29 | public static $replacement; |
|
30 | ||
31 | public function __construct() |
|
32 | { |
|
33 | $this->node_visitor = new NodeVisitor(); |
|
34 | } |
|
35 | ||
36 | protected static function generateNewSource($source) |
|
37 | { |
|
38 | $tokens = token_get_all($source); |
|
39 | $new_source = ''; |
|
40 | $i = -1; |
|
41 | ||
42 | ksort(self::$replacement); |
|
43 | reset(self::$replacement); |
|
44 | $replacement['key'] = key(self::$replacement); |
|
45 | $replacement['value'] = current(self::$replacement); |
|
46 | next(self::$replacement); |
|
47 | if ($replacement['key'] === null) |
|
48 | { |
|
49 | $replacement = false; |
|
50 | } |
|
51 | ||
52 | $start_method = false; |
|
53 | ||
54 | foreach ($tokens as $key => $token) |
|
55 | { |
|
56 | $i++; |
|
57 | ||
58 | if ($i == $replacement['key']) |
|
59 | { |
|
60 | $start_method = true; |
|
61 | } |
|
62 | ||
63 | if (is_string($token)) |
|
64 | { |
|
65 | if ($start_method && $token === '{') |
|
66 | { |
|
67 | if(self::isVoidFunction($tokens, $key)){ |
|
68 | $new_source .= '{ ' . self::CODENORET; |
|
69 | } |
|
70 | else{ |
|
71 | $new_source .= '{ ' . self::CODE; |
|
72 | } |
|
73 | $start_method = false; |
|
74 | $replacement['key'] = key(self::$replacement); |
|
75 | $replacement['value'] = current(self::$replacement); |
|
76 | next(self::$replacement); |
|
77 | if ($replacement['key'] === null) |
|
78 | { |
|
79 | $replacement = false; |
|
80 | } |
|
81 | } |
|
82 | else |
|
83 | { |
|
84 | $new_source .= $token; |
|
85 | } |
|
86 | } |
|
87 | else |
|
88 | { |
|
89 | $new_source .= $token[1]; |
|
90 | } |
|
91 | } |
|
92 | ||
93 | return $new_source; |
|
94 | } |
|
95 | ||
96 | /** |
|
97 | * Checks if a function has a void return type |
|
98 | * |
|
99 | * @param $tokens |
|
100 | * @param $key |
|
101 | * @return bool |
|
102 | */ |
|
103 | protected static function isVoidFunction($tokens, $key){ |
|
104 | if($key - 1 <= 0){ |
|
105 | return false; |
|
106 | } |
|
107 | $token = $tokens[$key - 1]; |
|
108 | if(is_array($token)){ |
|
109 | $token = $token[1]; |
|
110 | } |
|
111 | //Loop backwards though the start of the function block till you either find "void" or the end of the |
|
112 | //parameters declaration. |
|
113 | while($token !== ")"){ |
|
114 | if(strpos($token, "void") !== false){ |
|
115 | return true; |
|
116 | } |
|
117 | $token = $tokens[$key - 1]; |
|
118 | if(is_array($token)){ |
|
119 | $token = $token[1]; |
|
120 | } |
|
121 | $key--; |
|
122 | } |
|
123 | return false; |
|
124 | } |
|
125 | } |
|
126 |
@@ 20-125 (lines=106) @@ | ||
17 | ||
18 | use Kenjis\MonkeyPatch\Patcher\MethodPatcher\NodeVisitor; |
|
19 | ||
20 | class MethodPatcher extends AbstractPatcher |
|
21 | { |
|
22 | const CODE = <<<'EOL' |
|
23 | if (($__ret__ = \__PatchManager__::getReturn(__CLASS__, __FUNCTION__, func_get_args())) !== __GO_TO_ORIG__) return $__ret__; |
|
24 | EOL; |
|
25 | const CODENORET = <<<'EOL' |
|
26 | if (($__ret__ = \__PatchManager__::getReturn(__CLASS__, __FUNCTION__, func_get_args())) !== __GO_TO_ORIG__) return; |
|
27 | EOL; |
|
28 | ||
29 | public static $replacement; |
|
30 | ||
31 | public function __construct() |
|
32 | { |
|
33 | $this->node_visitor = new NodeVisitor(); |
|
34 | } |
|
35 | ||
36 | protected static function generateNewSource($source) |
|
37 | { |
|
38 | $tokens = token_get_all($source); |
|
39 | $new_source = ''; |
|
40 | $i = -1; |
|
41 | ||
42 | ksort(self::$replacement); |
|
43 | reset(self::$replacement); |
|
44 | $replacement['key'] = key(self::$replacement); |
|
45 | $replacement['value'] = current(self::$replacement); |
|
46 | next(self::$replacement); |
|
47 | if ($replacement['key'] === null) |
|
48 | { |
|
49 | $replacement = false; |
|
50 | } |
|
51 | ||
52 | $start_method = false; |
|
53 | ||
54 | foreach ($tokens as $key => $token) |
|
55 | { |
|
56 | $i++; |
|
57 | ||
58 | if ($i == $replacement['key']) |
|
59 | { |
|
60 | $start_method = true; |
|
61 | } |
|
62 | ||
63 | if (is_string($token)) |
|
64 | { |
|
65 | if ($start_method && $token === '{') |
|
66 | { |
|
67 | if(self::isVoidFunction($tokens, $key)){ |
|
68 | $new_source .= '{ ' . self::CODENORET; |
|
69 | } |
|
70 | else{ |
|
71 | $new_source .= '{ ' . self::CODE; |
|
72 | } |
|
73 | $start_method = false; |
|
74 | $replacement['key'] = key(self::$replacement); |
|
75 | $replacement['value'] = current(self::$replacement); |
|
76 | next(self::$replacement); |
|
77 | if ($replacement['key'] === null) |
|
78 | { |
|
79 | $replacement = false; |
|
80 | } |
|
81 | } |
|
82 | else |
|
83 | { |
|
84 | $new_source .= $token; |
|
85 | } |
|
86 | } |
|
87 | else |
|
88 | { |
|
89 | $new_source .= $token[1]; |
|
90 | } |
|
91 | } |
|
92 | ||
93 | return $new_source; |
|
94 | } |
|
95 | ||
96 | /** |
|
97 | * Checks if a function has a void return type |
|
98 | * |
|
99 | * @param $tokens |
|
100 | * @param $key |
|
101 | * @return bool |
|
102 | */ |
|
103 | protected static function isVoidFunction($tokens, $key){ |
|
104 | if($key - 1 <= 0){ |
|
105 | return false; |
|
106 | } |
|
107 | $token = $tokens[$key - 1]; |
|
108 | if(is_array($token)){ |
|
109 | $token = $token[1]; |
|
110 | } |
|
111 | //Loop backwards though the start of the function block till you either find "void" or the end of the |
|
112 | //parameters declaration. |
|
113 | while($token !== ")"){ |
|
114 | if(strpos($token, "void") !== false){ |
|
115 | return true; |
|
116 | } |
|
117 | $token = $tokens[$key - 1]; |
|
118 | if(is_array($token)){ |
|
119 | $token = $token[1]; |
|
120 | } |
|
121 | $key--; |
|
122 | } |
|
123 | return false; |
|
124 | } |
|
125 | } |
|
126 |