rokojori-godot/cpp-class-member.js

class CppParameter
{
  name = "";
  type = "";
}

class CppClassMember
{
  name = "";
  isMethod  = false;
  isVirtual = false;
  isStatic  = false;
  isSignal  = false;
  type = "void";
  parameters = [];

  constructor( nameDefinition, body )
  {
    if ( nameDefinition.endsWith( "()" ) )
    {
      this.isMethod = true;
      nameDefinition = nameDefinition.replace( /\(\)$/, "" );
    }

    let names = nameDefinition.split( /\s+/ );
    this.name = names[ names.length -1 ];

    for ( let i = 0; i < names.length - 1; i++ )
    {
      if ( "virtual" === names[ i ] )
      {
        this.isVirtual = true;
      }

      if ( "static" === names[ i ] )
      {
        this.isStatic = true;
      }

      if ( "signal" === names[ i ] )
      {
        this.isSignal = true;
      }

    }

    if ( this.isMethod )
    {
      this.parseMethodBody( body );
    }
  }

  parseMethodBody( body )
  {
    if ( ! body )
    {
      return;
    }

    this.type = body.type || "void";

    let parameters = body.parameters;

    if ( ! parameters )
    {
      return;
    }

    for ( let i = 0; i < parameters.length; i++ )
    {
      let pm = parameters[ i ];
      
      let cppParameter = new CppParameter();

      for ( let it in pm )
      {
        if ( ! pm.hasOwnProperty( it ) )
        {
          continue;
        }
        
        cppParameter.name = it;
        cppParameter.type = pm[ it ];
      }

      this.parameters.push( cppParameter );
    }
  }

  getMethodBinding( className )
  {
    if ( this.isSignal )
    {  
      //ADD_SIGNAL(MethodInfo("session_supported", PropertyInfo(Variant::STRING, "session_mode"), PropertyInfo(Variant::BOOL, "supported")));
      let parameterInfos = "";

      for ( let i = 0; i < this.parameters.length; i++ )
      {
        let type = stringToVariantType( this.parameters[ i ].type );
        let name = this.parameters[ i ].name;

        let pi = `, PropertyInfo(Variant::${type}, "${name}")`;
        
        parameterInfos += pi;
      }

      return `ADD_SIGNAL (MethodInfo( "${this.name}" ${parameterInfos}) );`;
    }
    else if ( this.isVirtual )
    {
      return `GDVIRTUAL_BIND( ${this.name} );`;
    }
    else
    {
      return `ClassDB::bind_method( D_METHOD( "${this.name}" ) , &${className}::${this.name} );`
    }
  }

  getHeaderDefinition()
  {
    if ( this.isSignal )
    {
      return `/* signal ${this.name} */`;
    }
    else if ( this.isMethod )
    {
      return this.getMethodHeaderDefinition();
    }
    else
    {
      return this.getVariableHeaderDefinition();
    }
  } 

  getParametersDefinition()
  {
    return this.parameters.map( p => p.type + " " + p.name ).join( ", " );
  }

  getMethodHeaderDefinition()
  {
    if ( this.isVirtual )
    {
      let numParameters = this.parameters.length;
      let methodReturnType = this.type === "void" ? "" : "R";
      let returnTypeDefinition = this.type === "void" ? "" : ( this.type + ", " ) ;

      let parametersDefinition = "";

      if ( numParameters > 0 )
      {
        parametersDefinition = ", " + this.parameters.map( p => p.type ).join( ", " );
      }


      return `GDVIRTUAL${numParameters}${methodReturnType}( ${returnTypeDefinition}${this.name}${parametersDefinition} );`
    }

    return `${this.type} ${this.name}(${this.parametersDefinition});`
  }

  getVariableHeaderDefinition()
  {
    return `${this.type} ${this.name};`
  }

}

module.exports = 
{
  CppParameter,
  CppClassMember  
}


function stringToVariantType( stringType )
{
  switch( stringType )
  {
    case "bool": return "BOOL";
    case "int": return "INT";
    case "float": return "FLOAT";
    case "String": return "STRING";

    // Math types.
    case "Vector2": return "VECTOR2";
    case "Vector2i": return "VECTOR2I";
    case "Rect2": return "RECT2";
    case "Rect2i": return "RECT2I";
    case "Transform2D": return "TRANSFORM2D";
    case "Vector3": return "VECTOR3";
    case "Vector3i": return "VECTOR3I";
    case "Vector4": return "VECTOR4";
    case "Vector4i": return "VECTOR4I";
    case "Plane": return "PLANE";
    case "AABB": return "AABB";
    case "Quaternion": return "QUATERNION";
    case "Basis": return "BASIS";
    case "Transform3D": return "TRANSFORM3D";
    case "Projection": return "PROJECTION";

    // Miscellaneous types.
    case "Color": return "COLOR";
    case "RID": return "RID";
    case "Object": return "OBJECT";
    case "Callable": return "CALLABLE";
    case "Signal": return "SIGNAL";
    case "StringName": return "STRING_NAME";
    case "NodePath": return "NODE_PATH";
    case "Dictionary": return "DICTIONARY";
    case "Array": return "ARRAY";

    // Arrays.
    case "PackedByteArray": return "PACKED_BYTE_ARRAY";
    case "PackedInt32Array": return "PACKED_INT32_ARRAY";
    case "PackedInt64Array": return "PACKED_INT64_ARRAY";
    case "PackedFloat32Array": return "PACKED_FLOAT32_ARRAY";
    case "PackedFloat64Array": return "PACKED_FLOAT64_ARRAY";
    case "PackedStringArray": return "PACKED_STRING_ARRAY";
    case "PackedVector2Array": return "PACKED_VECTOR2_ARRAY";
    case "PackedVector3Array": return "PACKED_VECTOR3_ARRAY";
    case "PackedColorArray": return "PACKED_COLOR_ARRAY";
  }

  return null;
}