Examples with EntitySet com.github.davidmoten.odata.client.generator.model.EntitySet used on opensource projects

Example 1 with EntitySet

use of com.github.davidmoten.odata.client.generator.model.EntitySet in project odata-client by davidmoten.

the class Generator method writeContainer.

private void writeContainer(Schema schema, TEntityContainer t) {
    names.getDirectoryContainer(schema).mkdirs();
    String simpleClassName = names.getSimpleClassNameContainer(schema, t.getName());
    Imports imports = new Imports(names.getFullClassNameContainer(schema, t.getName()));
    Indent indent = new Indent();
    StringWriter w = new StringWriter();
    try (PrintWriter p = new PrintWriter(w)) {
        p.format("package %s;\n\n", names.getPackageContainer(schema));
        p.format(IMPORTSHERE);
        final String extension;
        if (t.getExtends() != null) {
            extension = " extends " + imports.add(names.getFullClassNameFromTypeWithNamespace(t.getExtends()));
        } else {
            extension = "";
        }
        p.format("public final class %s%s implements %s {\n\n", simpleClassName, extension, imports.add(HasContext.class));
        // TODO handle container extension
        // write fields
        p.format("%sprivate final %s contextPath;\n\n", indent.right(), imports.add(ContextPath.class));
        // write constructor
        p.format("%spublic %s(%s context) {\n", indent, simpleClassName, imports.add(Context.class));
        p.format("%sthis.contextPath = new %s(context, context.service().getBasePath());\n", indent.right(), imports.add(ContextPath.class));
        p.format("%s}\n", indent.left());
        p.format("\n%s@%s\n", indent, imports.add(Override.class));
        p.format("%spublic %s _context() {\n", indent, imports.add(Context.class));
        p.format("%sreturn contextPath.context();\n", indent.right());
        p.format("%s}\n", indent.left());
        p.format("\n%spublic %s _service() {\n", indent, imports.add(HttpService.class));
        p.format("%sreturn contextPath.context().service();\n", indent.right());
        p.format("%s}\n", indent.left());
        // write static testing method
        p.format("\n%sstatic final class ContainerBuilderImpl extends %s<%s> {\n", indent, imports.add(ContainerBuilder.class), simpleClassName);
        p.format("\n%s@%s\n", indent.right(), imports.add(Override.class));
        p.format("%spublic %s _create(%s context) {\n", indent, simpleClassName, imports.add(Context.class));
        p.format("%sreturn new %s(context);\n", indent.right(), simpleClassName);
        p.format("%s}\n", indent.left());
        p.format("%s}\n", indent.left());
        p.format("\n%spublic static %s<%s<%s>, %s> test() {\n", indent, imports.add(BuilderBase.class), imports.add(ContainerBuilder.class), simpleClassName, simpleClassName);
        p.format("%sreturn new ContainerBuilderImpl();\n", indent.right());
        p.format("%s}\n", indent.left());
        // write get methods from properties
        // 
        Util.filter(t.getEntitySetOrActionImportOrFunctionImport(), TEntitySet.class).forEach(x -> {
            Schema sch = names.getSchema(x.getEntityType());
            boolean addedWithEmptyKeys = false;
            if (names.isEntityWithNamespace(x.getEntityType())) {
                String entityRequestType = names.getFullClassNameEntityRequestFromTypeWithNamespace(sch, x.getEntityType());
                EntityType et = names.getEntityType(x.getEntityType());
                KeyInfo k = getKeyInfo(et, imports);
                p.format("\n%spublic %s %s(%s) {\n", indent, imports.add(entityRequestType), Names.getIdentifier(x.getName()), k.typedParams);
                p.format("%sreturn new %s(contextPath.addSegment(\"%s\")%s, %s.empty());\n", indent.right(), imports.add(entityRequestType), x.getName(), k.addKeys, imports.add(Optional.class));
                p.format("%s}\n", indent.left());
                addedWithEmptyKeys = k.isEmpty();
            }
            if (!addedWithEmptyKeys) {
                // don't add a collection method if there were no keys for the entity
                EntitySet es = new EntitySet(schema, t, x, names);
                p.format("\n%spublic %s %s() {\n", indent, imports.add(es.getFullClassNameEntitySet()), Names.getIdentifier(x.getName()));
                p.format("%sreturn new %s(\n", indent.right(), imports.add(es.getFullClassNameEntitySet()));
                p.format("%scontextPath.addSegment(\"%s\"));\n", indent.right().right().right().right(), x.getName());
                p.format("%s}\n", indent.left().left().left().left().left());
            }
        });
        // 
        Util.filter(t.getEntitySetOrActionImportOrFunctionImport(), // 
        TSingleton.class).forEach(x -> {
            String importedType = toClassName(x, imports);
            p.format("\n%spublic %s %s() {\n", indent, importedType, Names.getIdentifier(x.getName()));
            p.format("%sreturn new %s(contextPath.addSegment(\"%s\"), %s.empty());\n", indent.right(), importedType, x.getName(), imports.add(Optional.class));
            p.format("%s}\n", indent.left());
        });
        // write unbound actions
        Set<String> methodNames = new HashSet<>();
        // 
        Util.types(schema, // 
        TAction.class).filter(x -> !x.isIsBound()).forEach(x -> writeAction(imports, indent, p, new Action(x, names), methodNames));
        // 
        Util.types(schema, // 
        TFunction.class).filter(x -> !x.isIsBound()).forEach(x -> writeFunction(imports, indent, p, new Function(x, names), methodNames));
        p.format("\n}\n");
        File classFile = names.getClassFileContainer(schema, t.getName());
        writeToFile(imports, w, classFile);
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}

Example 2 with EntitySet

use of com.github.davidmoten.odata.client.generator.model.EntitySet in project odata-client by davidmoten.

the class Generator method writeEntityRequest.

private void writeEntityRequest(Schema schema, TEntityType entityType, Map<String, List<Action>> typeActions, Map<String, List<Function>> typeFunctions) {
    EntityType t = new EntityType(entityType, names);
    names.getDirectoryEntityRequest(schema).mkdirs();
    // TODO only write out those requests needed
    String simpleClassName = t.getSimpleClassNameEntityRequest();
    Imports imports = new Imports(t.getFullClassNameEntityRequest());
    Indent indent = new Indent();
    StringWriter w = new StringWriter();
    try (PrintWriter p = new PrintWriter(w)) {
        p.format("package %s;\n\n", t.getPackageEntityRequest());
        p.format(IMPORTSHERE);
        p.format("@%s\n", imports.add(JsonIgnoreType.class));
        // don't make class final because can get extended by EntitySet
        p.format("public class %s extends %s {\n\n", simpleClassName, imports.add(EntityRequest.class) + "<" + imports.add(t.getFullClassNameEntity()) + ">");
        indent.right();
        // add constructor
        // 
        p.format(// 
        "%spublic %s(%s contextPath, %s<%s> value) {\n", // 
        indent, // 
        simpleClassName, // 
        imports.add(ContextPath.class), imports.add(Optional.class), imports.add(Object.class));
        // 
        p.format(// 
        "%ssuper(%s.class, contextPath, value, %s);\n", // 
        indent.right(), // 
        imports.add(t.getFullClassNameEntity()), t.isMediaEntityOrHasStreamProperty());
        p.format("%s}\n", indent.left());
        indent.left();
        if (t.hasStream()) {
            p.format("\n%s/**\n", indent);
            p.format("%s * If returning a stream without using object metadata is not supported then", indent);
            p.format("%s * returns {@code Optional.empty()}. Otherwise, returns a stream provider\n", indent);
            p.format("%s * where the location of the stream is assumed to be the current path + {@code /$value}.\n", indent);
            p.format("%s *\n", indent);
            p.format("%s * @return StreamProvider if suitable metadata found otherwise returns\n", indent);
            p.format("%s *         {@code Optional.empty()}\n", indent);
            p.format("%s */\n", indent);
            p.format("%s@%s\n", indent, imports.add(JsonIgnore.class));
            p.format("%spublic %s<%s> getStreamCurrentPath() {\n", indent, imports.add(Optional.class), imports.add(StreamProvider.class));
            p.format("%sreturn %s.createStream(contextPath, null);\n", indent.right(), imports.add(RequestHelper.class));
            p.format("%s}\n", indent.left());
        }
        // TODO also support navigation properties with complexTypes?
        // 
        t.getNavigationProperties().stream().filter(x -> {
            boolean isEntity = names.isEntityWithNamespace(names.getInnerType(names.getType(x)));
            if (!isEntity) {
                log("Unexpected entity with non-entity navigation property type: " + simpleClassName + "." + x.getName() + ". If you get this message then raise an issue on the github project for odata-client.");
            }
            return isEntity;
        }).forEach(x -> {
            indent.right();
            final String returnClass;
            String y = x.getType().get(0);
            Schema sch = names.getSchema(names.getInnerType(y));
            if (Names.isCollection(y)) {
                returnClass = toClassName(x, imports);
            } else {
                returnClass = imports.add(names.getFullClassNameEntityRequestFromTypeWithNamespace(sch, y));
            }
            // if collection then add with id method (if has ids)
            if (Names.isCollection(y)) {
                // TODO use actual key name from metadata
                String inner = names.getInnerType(y);
                // TODO remove redundant check
                if (names.isEntityWithNamespace(inner)) {
                    String entityRequestType = names.getFullClassNameEntityRequestFromTypeWithNamespace(sch, inner);
                    EntityType et = names.getEntityType(inner);
                    KeyInfo k = getKeyInfo(et, imports);
                    if (!k.isEmpty()) {
                        p.format("\n%spublic %s %s(%s) {\n", indent, imports.add(entityRequestType), Names.getIdentifier(x.getName()), k.typedParams);
                        p.format("%sreturn new %s(contextPath.addSegment(\"%s\")%s, %s.empty());\n", indent.right(), imports.add(entityRequestType), x.getName(), k.addKeys, imports.add(Optional.class));
                        p.format("%s}\n", indent.left());
                    }
                }
            }
            // 
            p.format(// 
            "\n%spublic %s %s() {\n", // 
            indent, // 
            returnClass, Names.getGetterMethodWithoutGet(x.getName()));
            if (isCollection(x)) {
                p.format("%sreturn new %s(\n", indent.right(), toClassName(x, imports));
                p.format("%scontextPath.addSegment(\"%s\"), %s.empty());\n", indent.right().right().right().right(), x.getName(), imports.add(Optional.class));
                indent.left().left().left().left();
            } else {
                p.format("%sreturn new %s(contextPath.addSegment(\"%s\"), %s.empty());\n", indent.right(), returnClass, x.getName(), imports.add(Optional.class));
            }
            p.format("%s}\n", indent.left());
            indent.left();
        });
        indent.right();
        Set<String> methodNames = new HashSet<>();
        writeBoundActionMethods(t, typeActions, imports, indent, p, methodNames);
        writeBoundFunctionMethods(t, typeFunctions, imports, indent, p, methodNames);
        indent.left();
        p.format("\n}\n");
        writeToFile(imports, w, t.getClassFileEntityRequest());
    } catch (IOException e) {
        throw new RuntimeException(e);
    }
}

Example 3 with EntitySet

use of com.github.davidmoten.odata.client.generator.model.EntitySet in project odata-client by davidmoten.

the class Generator method writeEntitySet.

private void writeEntitySet(Schema schema, Pair<TEntityContainer, TEntitySet> pair) {
    EntitySet t = new EntitySet(schema, pair.a, pair.b, names);
    t.getDirectoryEntitySet().mkdirs();
    Imports imports = new Imports(t.getFullClassNameEntitySet());
    Indent indent = new Indent();
    StringWriter w = new StringWriter();
    try (PrintWriter p = new PrintWriter(w)) {
        p.format("package %s;\n\n", t.getPackage());
        p.format(IMPORTSHERE);
        String baseCollectionClassName = t.getBaseCollectionRequestClassName(imports);
        // 
        p.format(// 
        "%spublic final class %s extends %s {\n", // 
        indent, // 
        t.getSimpleClassNameEntitySet(), baseCollectionClassName);
        indent.right();
        // 
        p.format(// 
        "\n%spublic %s(%s contextPath) {\n", // 
        indent, // 
        t.getSimpleClassNameEntitySet(), imports.add(ContextPath.class));
        p.format("%ssuper(contextPath, %s.empty());\n", indent.right(), imports.add(Optional.class));
        p.format("%s}\n", indent.left());
        // write navigation property bindings
        Set<String> duplicateMethodNames = findDuplicateNavigationPropertyBindingMethodNames(pair, t);
        // 
        Util.filter(// 
        pair.b.getNavigationPropertyBindingOrAnnotation(), // 
        TNavigationPropertyBinding.class).forEach(b -> {
            String methodName = t.getMethodName(b);
            if (duplicateMethodNames.contains(methodName)) {
                methodName = t.getLongerMethodName(b);
            }
            Optional<EntitySet> referredEntitySet = t.getReferredEntitySet(b.getTarget());
            if (referredEntitySet.isPresent()) {
                String returnClassName = referredEntitySet.get().getFullClassNameEntitySet();
                p.format("\n%spublic %s %s() {\n", indent, imports.add(returnClassName), methodName);
                // 
                p.format(// 
                "%sreturn new %s(contextPath.addSegment(\"%s\"));\n", // 
                indent.right(), imports.add(// 
                referredEntitySet.get().getFullClassNameEntitySet()), b.getPath());
                p.format("%s}\n", indent.left());
            } else {
                log(// 
                "WARN: EntitySet '" + b.getTarget() + // 
                "' not found for navigation property binding " + t.getSimpleClassNameEntitySet() + "." + b.getPath());
            }
        });
        p.format("%s}\n", indent.left());
        writeToFile(imports, w, t.getClassFile());
    } catch (IOException e) {
        throw new UncheckedIOException(e);
    }
}