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;
  }

  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;
  }
}

aspect Mappings {
  // --- effectiveMappings ---
  syn java.util.List<MappingDefinition> UpdateDefinition.effectiveMappings();
  eq ReadFromMqttDefinition.effectiveMappings() {
    // if there is a first mapping, check if its input type is byte[].
    //  or if no mappings are specified.
    // then prepend the suitable default mapping
    java.util.List<MappingDefinition> result;
    if (getMappingList().isEmpty() || !getMappingList().get(0).getFromType().isByteArray()) {
      result = new java.util.ArrayList();
      result.add(suitableDefaultMapping());
      result.addAll(getMappingList());
    } else {
      result = getMappingList();
    }
    return result;
  }
  eq WriteToMqttDefinition.effectiveMappings() {
    // if there is a mapping, check if the output type of the last mapping is byte[].
    //  or if no mappings are specified.
    // then append the suitable default mapping
    java.util.List<MappingDefinition> result;
    int numMappings = getMappingList().size();
    if (numMappings == 0 || !getMappingList().get(numMappings - 1).getToType().isByteArray()) {
      result = new java.util.ArrayList();
      result.addAll(getMappingList());
      result.add(suitableDefaultMapping());
    } else {
      result = getMappingList();
    }
    return result;
  }

  // --- isPrimitiveType ---
  syn boolean TokenComponent.isPrimitiveType() = getJavaTypeUse().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 UpdateDefinition.suitableDefaultMapping();
  eq ReadFromMqttDefinition.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 WriteToMqttDefinition.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;
    }
  }

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

  // --- allMappingDefinitions ---
  syn java.util.List<MappingDefinition> RagConnect.allMappingDefinitions() {
    java.util.List<MappingDefinition> result = new java.util.ArrayList<>();
    getMappingDefinitionList().iterator().forEachRemaining(result::add);
    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());
    return result;
  }
}