Added JsonRpcController::addMethod() for adding functions after instanciation

Author: wasinger

Controller/JsonRpcController.php

@@ -140,6 +140,39 @@ public function execute(Request $httprequest)
         }
     }
 
+    /**
+     * Add a new function that can be called by RPC
+     *
+     * @param string $alias The function name used in the RPC call
+     * @param string $service The service name of the method to call
+     * @param string $method The method of $service
+     * @param bool $overwrite Whether to overwrite an existing function
+     * @throws \InvalidArgumentException
+     */
+    public function addMethod($alias, $service, $method, $overwrite = false)
+    {
+        if (!isset($this->config['functions'])) $this->config['functions'] = array();
+        if (isset($this->config['functions'][$alias]) && !$overwrite) {
+            throw new \InvalidArgumentException('JsonRpcController: The function "' . $alias . '" already exists.');
+        }
+        $this->config['functions'][$alias] = array(
+            'service' => $service,
+            'method' => $method
+        );
+    }
+
+    /**
+     * Remove a method definition
+     *
+     * @param string $alias
+     */
+    public function removeMethod($alias)
+    {
+        if (isset($this->config['functions'][$alias])) {
+            unset($this->config['functions'][$alias]);
+        }
+    }
+
     protected function getError($code)
     {
         $message = '';

Tests/JsonRpcControllerTest.php

@@ -61,6 +61,30 @@ public function testHello()
         $this->assertEquals(-32602, $response['error']['code']);
     }
 
+    public function testAddMethod()
+    {
+        $requestdata = array(
+            'jsonrpc' => '2.0',
+            'id' => 'test',
+            'method' => 'testhi',
+            'params' => array('name' => 'Tom')
+        );
+        // this request will fail because there is no such method "testhi"
+        $response = $this->makeRequest($requestdata);
+        $this->assertArrayHasKey('error', $response);
+        $this->assertArrayNotHasKey('result', $response);
+        $this->assertEquals(-32601, $response['error']['code']);
+
+        // add the method definition for "testhi"
+        $this->controller->addMethod('testhi', 'wa72_jsonrpc.testservice', 'hi');
+
+        // now the request should succeed
+        $response = $this->makeRequest($requestdata);
+        $this->assertArrayHasKey('result', $response);
+        $this->assertArrayNotHasKey('error', $response);
+        $this->assertEquals('Hi Tom!', $response['result']);
+    }
+
     private function makeRequest($requestdata)
     {
         return json_decode($this->controller->execute(

Tests/Testservice.php

@@ -2,7 +2,13 @@
 namespace Wa72\JsonRpcBundle\Tests;
 
 class Testservice {
-    public function hello($name) {
+    public function hello($name)
+    {
         return 'Hello ' . $name . '!';
     }
+
+    public function hi($name)
+    {
+        return 'Hi ' . $name . '!';
+    }
 }