org.obiba.opal.web.magma.view.ViewDtos.java Source code

Introduction

Here is the source code for org.obiba.opal.web.magma.view.ViewDtos.java

Source

/*******************************************************************************
 * Copyright 2008(c) The OBiBa Consortium. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/
package org.obiba.opal.web.magma.view;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;

import javax.validation.constraints.NotNull;

import org.obiba.magma.MagmaEngine;
import org.obiba.magma.ValueTable;
import org.obiba.magma.js.views.JavascriptClause;
import org.obiba.magma.views.View;
import org.obiba.magma.views.WhereClause;
import org.obiba.opal.web.model.Magma.TableDto;
import org.obiba.opal.web.model.Magma.ViewDto;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import com.google.common.collect.ImmutableSet;

/**
 * Utilities for handling View Dtos. This class is not static like others because it is extensible through a set of
 * {@code ViewDtoExtension} present in the context.
 */
@Component
public final class ViewDtos {

    private Set<ViewDtoExtension> extensions;

    @Autowired
    public void setExtensions(Collection<ViewDtoExtension> extensions) {
        this.extensions = ImmutableSet.copyOf(extensions);
    }

    @NotNull
    public View fromDto(@NotNull ViewDto viewDto) {
        List<ValueTable> fromTables = getFromTables(viewDto);
        View.Builder builder = View.Builder.newView(viewDto.getName(),
                (ValueTable[]) fromTables.toArray(new ValueTable[fromTables.size()]));

        if (viewDto.hasWhere()) {
            WhereClause whereClause = new JavascriptClause(viewDto.getWhere());
            builder.where(whereClause);
        }

        for (ViewDtoExtension extension : extensions) {
            if (extension.isExtensionOf(viewDto)) {
                return extension.fromDto(viewDto, builder);
            }
        }
        throw new IllegalStateException("Unknown view type");
    }

    @NotNull
    public TableDto asTableDto(@NotNull ViewDto viewDto) {
        TableDto.Builder builder = TableDto.newBuilder().setName(viewDto.getName());
        for (ViewDtoExtension extension : extensions) {
            if (extension.isExtensionOf(viewDto)) {
                return extension.asTableDto(viewDto, builder);
            }
        }
        throw new IllegalStateException("Unknown view type");
    }

    @NotNull
    public ViewDto asDto(@NotNull View view) {
        for (ViewDtoExtension extension : extensions) {
            if (extension.isDtoOf(view)) {
                return extension.asDto(view);
            }
        }
        throw new IllegalStateException("Unknown view type");
    }

    @NotNull
    private List<ValueTable> getFromTables(@NotNull ViewDto viewDto) {
        List<ValueTable> fromTables = new ArrayList<>();
        for (int i = 0; i < viewDto.getFromCount(); i++) {
            String fromTable = viewDto.getFrom(i);
            fromTables.add(MagmaEngine.get().createReference(fromTable));
        }
        return fromTables;
    }

}