RecordTypeWriter.java

/*
 * Copyright 2010 Outerthought bvba
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.lilyproject.tools.import_.json;

import java.util.Map;

import org.codehaus.jackson.node.ArrayNode;
import org.codehaus.jackson.node.JsonNodeFactory;
import org.codehaus.jackson.node.ObjectNode;
import org.lilyproject.repository.api.FieldTypeEntry;
import org.lilyproject.repository.api.LRepository;
import org.lilyproject.repository.api.RecordType;
import org.lilyproject.repository.api.SchemaId;

public class RecordTypeWriter implements EntityWriter<RecordType> {
    public static final EntityWriter<RecordType> INSTANCE = new RecordTypeWriter();

    @Override
    public ObjectNode toJson(RecordType recordType, WriteOptions options, LRepository repository) {
        Namespaces namespaces = new NamespacesImpl(options != null ? options.getUseNamespacePrefixes() :
                        NamespacesImpl.DEFAULT_USE_PREFIXES);

        ObjectNode rtNode = toJson(recordType, options, namespaces, repository);

        if (namespaces.usePrefixes()) {
            rtNode.put("namespaces", NamespacesConverter.toJson(namespaces));
        }

        return rtNode;
    }

    @Override
    public ObjectNode toJson(RecordType recordType, WriteOptions options, Namespaces namespaces, LRepository repository) {
        return toJson(recordType, options, namespaces, true);
    }

    public static ObjectNode toJson(RecordType recordType, WriteOptions options, Namespaces namespaces,
            boolean includeName) {
        ObjectNode rtNode = JsonNodeFactory.instance.objectNode();

        rtNode.put("id", recordType.getId().toString());

        if (includeName) {
            rtNode.put("name", QNameConverter.toJson(recordType.getName(), namespaces));
        }

        ArrayNode fieldsNode = rtNode.putArray("fields");
        for (FieldTypeEntry entry : recordType.getFieldTypeEntries()) {
            ObjectNode entryNode = fieldsNode.addObject();
            entryNode.put("id", entry.getFieldTypeId().toString());
            entryNode.put("mandatory", entry.isMandatory());
        }

        rtNode.put("version", recordType.getVersion());


        ArrayNode supertypesNode = rtNode.putArray("supertypes");
        for (Map.Entry<SchemaId, Long> supertype : recordType.getSupertypes().entrySet()) {
            ObjectNode entryNode = supertypesNode.addObject();
            entryNode.put("id", supertype.getKey().toString());
            entryNode.put("version", supertype.getValue());
        }

        return rtNode;
    }

}