DefinitionLookupRequestImpl.java

/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates.
 *
 * 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.kie.workbench.common.stunner.core.lookup.definition;

import java.util.Set;

import org.jboss.errai.common.client.api.annotations.MapsTo;
import org.jboss.errai.common.client.api.annotations.NonPortable;
import org.jboss.errai.common.client.api.annotations.Portable;
import org.kie.workbench.common.stunner.core.lookup.AbstractLookupRequest;
import org.kie.workbench.common.stunner.core.lookup.AbstractLookupRequestBuilder;

@Portable
public final class DefinitionLookupRequestImpl extends AbstractLookupRequest implements DefinitionLookupRequest {

    private final String definitionSetId;

    public DefinitionLookupRequestImpl(final @MapsTo("criteria") String criteria,
                                       final @MapsTo("page") int page,
                                       final @MapsTo("pageSize") int pageSize,
                                       final @MapsTo("definitionSetId") String definitionSetId) {
        super(criteria,
              page,
              pageSize);
        this.definitionSetId = definitionSetId;
    }

    @Override
    public String getDefinitionSetId() {
        return definitionSetId;
    }

    @NonPortable
    public static class Builder extends AbstractLookupRequestBuilder<Builder> {

        enum Type {
            NODE,
            EDGE;
        }

        private String defSetId;
        private final StringBuilder criteria = new StringBuilder();

        public Builder definitionSetId(final String defSetId) {
            this.defSetId = defSetId;
            return this;
        }

        public Builder id(final String id) {
            criteria.append("id=").append(id).append(";");
            return this;
        }

        public Builder type(final Type type) {
            criteria.append("type=").append(type.name().toLowerCase()).append(";");
            return this;
        }

        public Builder labels(final Set<String> labels) {
            criteria.append("labels=").append(fromSet(labels)).append(";");
            return this;
        }

        public DefinitionLookupRequest build() {
            return new DefinitionLookupRequestImpl(criteria.toString(),
                                                   page,
                                                   pageSize,
                                                   defSetId);
        }
    }
}