Mappings.jrag

aspect DefaultMappings {

  private DefaultMappingDefinition RagConnect.baseDefaultMappingDefinitionFromBytes(String typeName) {
    DefaultMappingDefinition result = new DefaultMappingDefinition();
    result.setID("_DefaultBytesTo" + Character.toUpperCase(typeName.charAt(0)) + typeName.substring(1) + "Mapping");
    result.setFromType(new JavaArrayMappingDefinitionType(new SimpleJavaTypeUse("byte")));
    result.setFromVariableName("bytes");
    result.setToType(new JavaMappingDefinitionType(new SimpleJavaTypeUse(typeName)));
    return result;
  }

  private DefaultMappingDefinition RagConnect.baseDefaultMappingDefinitionToBytes(String typeName) {
    DefaultMappingDefinition result = new DefaultMappingDefinition();
    result.setID("_Default" + Character.toUpperCase(typeName.charAt(0)) + typeName.substring(1) + "ToBytesMapping");
    result.setFromType(new JavaMappingDefinitionType(new SimpleJavaTypeUse(typeName)));
    result.setFromVariableName("input");
    result.setToType(new JavaArrayMappingDefinitionType(new SimpleJavaTypeUse("byte")));
    return result;
  }

  private String RagConnect.baseDefaultMappingTypeNamePart(String typeName) {
    return Character.toUpperCase(typeName.charAt(0)) + typeName.substring(1).replace("[]", "s");
  }

  private MappingDefinitionType RagConnect.baseDefaultMappingTypeFromName(String typeName) {
    return typeName.endsWith("[]") ?
        new JavaArrayMappingDefinitionType(new SimpleJavaTypeUse(typeName.replace("[]", ""))) :
        new JavaMappingDefinitionType(new SimpleJavaTypeUse(typeName));
  }

  private DefaultMappingDefinition RagConnect.baseDefaultMappingDefinition(String fromTypeName, String toTypeName, String content) {
    DefaultMappingDefinition result = new DefaultMappingDefinition();
    result.setID("_Default" + baseDefaultMappingTypeNamePart(fromTypeName) + "To" + baseDefaultMappingTypeNamePart(toTypeName) + "Mapping");
    result.setFromType(baseDefaultMappingTypeFromName(fromTypeName));
    result.setFromVariableName("input");
    result.setToType(baseDefaultMappingTypeFromName(toTypeName));
    result.setContent(content);
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToIntMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("int");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getInt();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToShortMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("short");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getShort();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToLongMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("long");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getLong();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToFloatMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("float");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getFloat();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToDoubleMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("double");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getDouble();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToCharMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("char");
    result.setContent("return java.nio.ByteBuffer.wrap(bytes).getChar();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultBytesToStringMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionFromBytes("String");
    result.setContent("return new String(bytes);");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultIntToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("int");
    result.setContent("return java.nio.ByteBuffer.allocate(4).putInt(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultShortToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("short");
    result.setContent("return java.nio.ByteBuffer.allocate(2).putShort(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultLongToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("long");
    result.setContent("return java.nio.ByteBuffer.allocate(8).putLong(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultFloatToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("float");
    result.setContent("return java.nio.ByteBuffer.allocate(4).putFloat(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultDoubleToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("double");
    result.setContent("return java.nio.ByteBuffer.allocate(8).putDouble(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultCharToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("char");
    result.setContent("return java.nio.ByteBuffer.allocate(2).putChar(input).array();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultStringToBytesMapping() {
    DefaultMappingDefinition result = baseDefaultMappingDefinitionToBytes("String");
    result.setContent("return input.getBytes();");
    return result;
  }

  syn nta DefaultMappingDefinition RagConnect.defaultStringToIntMapping() = baseDefaultMappingDefinition("String", "int", "return Integer.parseInteger(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultStringToShortMapping() = baseDefaultMappingDefinition("String", "short", "return Short.parseShort(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultStringToLongMapping() = baseDefaultMappingDefinition("String", "long", "return Long.parseLong(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultStringToFloatMapping() = baseDefaultMappingDefinition("String", "float", "return Float.parseFloat(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultStringToDoubleMapping() = baseDefaultMappingDefinition("String", "double", "return Double.parseDouble(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultStringToCharMapping() = baseDefaultMappingDefinition("String", "char", "return input.charAt(0);");

  syn nta DefaultMappingDefinition RagConnect.defaultIntToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultShortToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultLongToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultFloatToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultDoubleToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
  syn nta DefaultMappingDefinition RagConnect.defaultCharToStringMapping() = baseDefaultMappingDefinition("int", "String", "return String.valueOf(input);");
}

aspect Mappings {
  // --- effectiveMappings ---
  syn java.util.List<MappingDefinition> EndpointDefinition.effectiveMappings();
  eq ReceiveTokenEndpointDefinition.effectiveMappings() {
    // if there is a first mapping, check if it is suitable.
    //  or if no mappings are specified.
    // then prepend the suitable default mapping
    java.util.List<MappingDefinition> result;
    if (getMappingList().isEmpty() || !hasSuitableEdgeMapping()) {
      result = new java.util.ArrayList();
      result.add(suitableDefaultMapping());
      result.addAll(getMappingList());
    } else {
      result = getMappingList();
    }
    return result;
  }
  eq SendTokenEndpointDefinition.effectiveMappings() {
    // if there is a mapping, check if it is suitable.
    //  or if no mappings are specified.
    // then append the suitable default mapping
    java.util.List<MappingDefinition> result;
    if (getMappingList().isEmpty() || !hasSuitableEdgeMapping()) {
      result = new java.util.ArrayList();
      result.addAll(getMappingList());
      result.add(suitableDefaultMapping());
    } else {
      result = getMappingList();
    }
    return result;
  }

  // --- hasSuitableEdgeMapping ---
  syn boolean TokenEndpointDefinition.hasSuitableEdgeMapping();
  eq ReceiveTokenEndpointDefinition.hasSuitableEdgeMapping() = isSuitableEdgeMapping(getMappingList().get(0));
  eq SendTokenEndpointDefinition.hasSuitableEdgeMapping() = isSuitableEdgeMapping(getMappingList().get(getMappingList().size() - 1));

  // --- isSuitableEdgeMapping(def) ---
  syn boolean TokenEndpointDefinition.isSuitableEdgeMapping(MappingDefinition def);
  eq ReceiveFromMqttDefinition.isSuitableEdgeMapping(MappingDefinition def) = def.getFromType().isByteArray();
  eq SendToMqttDefinition.isSuitableEdgeMapping(MappingDefinition def) = def.getToType().isByteArray();
  eq ReceiveFromRestDefinition.isSuitableEdgeMapping(MappingDefinition def) = def.getFromType().isString();
  eq SendToRestDefinition.isSuitableEdgeMapping(MappingDefinition def) = def.getToType().isString();

  // --- isPrimitiveType ---
  syn boolean TokenComponent.isPrimitiveType() = effectiveJavaTypeUse().isPrimitiveType();
  syn boolean JavaTypeUse.isPrimitiveType() = false;
  eq SimpleJavaTypeUse.isPrimitiveType() {
    switch(getName()) {
      case "int":
      case "short":
      case "long":
      case "float":
      case "double":
      case "char":
      case "byte": return true;
      default: return false;
    }
  }
  syn boolean MappingDefinitionType.isPrimitiveType() = false;
  eq JavaMappingDefinitionType.isPrimitiveType() = getType().isPrimitiveType();

  // --- isArray ---
  syn boolean MappingDefinitionType.isArray() = false;
  eq JavaArrayMappingDefinitionType.isArray() = true;

  // --- suitableDefaultMapping ---
  syn DefaultMappingDefinition EndpointDefinition.suitableDefaultMapping();
  eq ReceiveFromMqttDefinition.suitableDefaultMapping() {
    String typeName = getMappingList().isEmpty() ?
        getToken().getJavaTypeUse().getName() :
        getMappingList().get(0).getFromType().prettyPrint();
    switch(typeName) {
      case "int":
      case "Integer": return ragconnect().defaultBytesToIntMapping();
      case "short":
      case "Short": return ragconnect().defaultBytesToShortMapping();
      case "long":
      case "Long": return ragconnect().defaultBytesToLongMapping();
      case "float":
      case "Float": return ragconnect().defaultBytesToFloatMapping();
      case "double":
      case "Double": return ragconnect().defaultBytesToDoubleMapping();
      case "char":
      case "Character": return ragconnect().defaultBytesToCharMapping();
      case "String": return ragconnect().defaultBytesToStringMapping();
      default: return null;
    }
  }
  eq SendToMqttDefinition.suitableDefaultMapping() {
    String typeName = getMappingList().isEmpty() ?
        getToken().getJavaTypeUse().getName() :
        getMappingList().get(getMappingList().size() - 1).getFromType().prettyPrint();
    switch(typeName) {
      case "int":
      case "Integer": return ragconnect().defaultIntToBytesMapping();
      case "short":
      case "Short": return ragconnect().defaultShortToBytesMapping();
      case "long":
      case "Long": return ragconnect().defaultLongToBytesMapping();
      case "float":
      case "Float": return ragconnect().defaultFloatToBytesMapping();
      case "double":
      case "Double": return ragconnect().defaultDoubleToBytesMapping();
      case "char":
      case "Character": return ragconnect().defaultCharToBytesMapping();
      case "String": return ragconnect().defaultStringToBytesMapping();
      default: return null;
    }
  }
  eq ReceiveFromRestDefinition.suitableDefaultMapping() {
    String typeName = getMappingList().isEmpty() ?
        getToken().getJavaTypeUse().getName() :
        getMappingList().get(0).getFromType().prettyPrint();
    switch(typeName) {
      case "int":
      case "Integer": return ragconnect().defaultStringToIntMapping();
      case "short":
      case "Short": return ragconnect().defaultStringToShortMapping();
      case "long":
      case "Long": return ragconnect().defaultStringToLongMapping();
      case "float":
      case "Float": return ragconnect().defaultStringToFloatMapping();
      case "double":
      case "Double": return ragconnect().defaultStringToDoubleMapping();
      case "char":
      case "Character": return ragconnect().defaultStringToCharMapping();
      default: return null;
    }
  }
  eq SendToRestDefinition.suitableDefaultMapping() {
    String typeName = getMappingList().isEmpty() ?
        getToken().getJavaTypeUse().getName() :
        getMappingList().get(getMappingList().size() - 1).getFromType().prettyPrint();
    switch(typeName) {
      case "int":
      case "Integer": return ragconnect().defaultIntToStringMapping();
      case "short":
      case "Short": return ragconnect().defaultShortToStringMapping();
      case "long":
      case "Long": return ragconnect().defaultLongToStringMapping();
      case "float":
      case "Float": return ragconnect().defaultFloatToStringMapping();
      case "double":
      case "Double": return ragconnect().defaultDoubleToStringMapping();
      case "char":
      case "Character": return ragconnect().defaultCharToStringMapping();
      default: return null;
    }
  }

  // --- isByteArray ---
  syn boolean MappingDefinitionType.isByteArray() = false;
  eq JavaArrayMappingDefinitionType.isByteArray() = getType().getName().equals("byte");

  // --- isString ---
  syn boolean MappingDefinitionType.isString() = false;
  eq JavaMappingDefinitionType.isString() = getType().getName().equals("String");

  // --- allMappingDefinitions ---
  syn java.util.List<MappingDefinition> RagConnect.allMappingDefinitions() {
    java.util.List<MappingDefinition> result = new java.util.ArrayList<>();
    getMappingDefinitionList().iterator().forEachRemaining(result::add);
    // byte[] converstion
    result.add(defaultBytesToIntMapping());
    result.add(defaultBytesToShortMapping());
    result.add(defaultBytesToLongMapping());
    result.add(defaultBytesToFloatMapping());
    result.add(defaultBytesToDoubleMapping());
    result.add(defaultBytesToCharMapping());
    result.add(defaultBytesToStringMapping());
    result.add(defaultIntToBytesMapping());
    result.add(defaultShortToBytesMapping());
    result.add(defaultLongToBytesMapping());
    result.add(defaultFloatToBytesMapping());
    result.add(defaultDoubleToBytesMapping());
    result.add(defaultCharToBytesMapping());
    result.add(defaultStringToBytesMapping());
    // string converstion
    result.add(defaultStringToIntMapping());
    result.add(defaultStringToShortMapping());
    result.add(defaultStringToLongMapping());
    result.add(defaultStringToFloatMapping());
    result.add(defaultStringToDoubleMapping());
    result.add(defaultStringToCharMapping());
    result.add(defaultIntToStringMapping());
    result.add(defaultShortToStringMapping());
    result.add(defaultLongToStringMapping());
    result.add(defaultFloatToStringMapping());
    result.add(defaultDoubleToStringMapping());
    result.add(defaultCharToStringMapping());
    return result;
  }
}