fixed base class search

Author: ovfireandvoid

README.md

@@ -190,7 +190,7 @@ function Command(cmd) {
 }
 
 Command.prototype = Object.create(Argument.prototype);
-Command.prototype.constructor = Argument;
+Command.prototype.constructor = Command;
 
 /*!
  * @brief Event Class
@@ -200,7 +200,7 @@ function Event(event) {
 }
 
 Event.prototype = Object.create(Argument.prototype);
-Event.prototype.constructor = Argument;
+Event.prototype.constructor = Event;
 ```
 
 Output pseudo code:
@@ -263,7 +263,7 @@ Here some things to notice:
 *   Base class is determined by next code:
 
     ```javascript
-    Event.prototype.constructor = Argument;
+    Event.prototype = Object.create(Argument.prototype);
     ```
 
     Here `Argument` will be used as base class of `Event`.

__tests__/extractToken.test.js

@@ -6,7 +6,7 @@ describe('extractToken', () => {
     const function_line = 'function foo() {}';
     const class_constructor_line = 'function Class(args) {}';
     const class_method_line = 'Class.prototype.foo = function(arg) {}';
-    const base_class_line = 'Class.prototype.constructor = BaseClass';
+    const base_class_line = 'Class.prototype = Object.create(BaseClass.prototype)';
 
     it('extracts global variable', () => {
       expect(extractToken.extractVariable(variable_line)).toEqual({

src/extractToken.js

@@ -49,7 +49,7 @@ const extractClass = line => {
 };
 
 const extractBaseClass = line => {
-  const base_class_expr = /^\s*(\w+)\.prototype\.constructor\s*\=\s*(\w+)/;
+  const base_class_expr = /^\s*(\w+)\.prototype\s*\=\s*Object\.create\((\w+)\.prototype\)/;
   const base_class_match = line.match(base_class_expr);
 
   if (base_class_match) {