fathom.rest.controller.extractors.ParamExtractor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fathom-rest Show documentation
RESTful Routes and Controllers based on Pippo
The newest version!
/*
 * Copyright (C) 2015 the original author or authors.
 *
 * 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 fathom.rest.controller.extractors;

import com.google.common.base.Strings;
import com.google.common.io.Files;
import fathom.rest.Context;
import fathom.rest.controller.Param;
import ro.pippo.core.ParameterValue;

import java.util.Set;

/**
 * @author James Moger
 */
public class ParamExtractor extends DefaultObjectExtractor
        implements NamedExtractor, PatternExtractor, ConfigurableExtractor, SuffixExtractor {

    private String name;

    private String pattern;

    private Set suffixes;

    @Override
    public Class getAnnotationClass() {
        return Param.class;
    }

    @Override
    public void configure(Param param) {
        setName(param.value());
        setPattern(param.pattern());
    }

    @Override
    public String getName() {
        return name;
    }

    @Override
    public void setName(String name) {
        this.name = name;
    }

    @Override
    public String getPattern() {
        return pattern;
    }

    @Override
    public void setPattern(String pattern) {
        this.pattern = pattern;
    }

    @Override
    public void setSuffixes(Set suffixes) {
        this.suffixes = suffixes;
    }

    @Override
    public Set getSuffixes() {
        return this.suffixes;
    }

    @Override
    public Object extract(Context context) {
        ParameterValue pv = context.getParameter(name);
        if (suffixes != null) {
            // automatically strip the suffix, if we can
            String value = pv.toString();
            if (value != null) {
                String ext = Files.getFileExtension(value);
                if (!Strings.isNullOrEmpty(ext) && suffixes.contains(ext.toLowerCase())) {
                    pv = new ParameterValue(Files.getNameWithoutExtension(value));
                }
            }
        }

        if (collectionType == null) {
            Object o = pv.to(objectType, pattern);
            return o;
        }

        Object o = pv.toCollection(collectionType, objectType, pattern);
        return o;
    }
}