cherry.sqlman.tool.statement.SqlStatementControllerImpl.java Source code

Introduction

Here is the source code for cherry.sqlman.tool.statement.SqlStatementControllerImpl.java

Source

/*
 * Copyright 2014,2015 agwlvssainokuni
 *
 * 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 cherry.sqlman.tool.statement;

import static cherry.foundation.springmvc.Contract.shouldExist;
import static cherry.sqlman.ParamDef.REQ_ID;
import static cherry.sqlman.ParamDef.REQ_REF;
import static cherry.sqlman.util.ModelAndViewBuilder.redirect;
import static cherry.sqlman.util.ModelAndViewBuilder.withViewname;
import static org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder.fromMethodCall;
import static org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder.on;

import java.util.Locale;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.tuple.Pair;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.mobile.device.site.SitePreference;
import org.springframework.security.core.Authentication;
import org.springframework.stereotype.Controller;
import org.springframework.validation.BindingResult;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.util.UriComponents;
import org.springframework.web.util.UriComponentsBuilder;

import cherry.foundation.logicalerror.LogicalErrorUtil;
import cherry.goods.paginate.PageSet;
import cherry.sqlman.LogicError;
import cherry.sqlman.tool.metadata.MetadataService;
import cherry.sqlman.tool.metadata.SqlMetadataForm;
import cherry.sqlman.tool.shared.DataSourceDef;
import cherry.sqlman.tool.shared.ResultSet;
import cherry.sqlman.tool.util.Util;
import cherry.sqlman.util.ViewNameUtil;

@Controller
public class SqlStatementControllerImpl extends SqlStatementSupport implements SqlStatementController {

    @Autowired
    private MetadataService metadataService;

    @Autowired
    private StatementService statementService;

    @Autowired
    private DataSourceDef dataSourceDef;

    private final String viewnameOfStart = ViewNameUtil
            .fromMethodCall(on(SqlStatementController.class).start(null, null, null, null, null, null, null));

    @Override
    public ModelAndView init(String redirTo, Integer ref, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request) {
        return redirect(redirectOnInit(redirTo, ref)).build();
    }

    @Override
    public ModelAndView start(Integer ref, SqlStatementForm form, BindingResult binding, Authentication auth,
            Locale locale, SitePreference sitePref, NativeWebRequest request) {
        initializeForm(form, ref, auth);
        return withViewname(viewnameOfStart).build();
    }

    @Override
    public ModelAndView execute(SqlStatementForm form, BindingResult binding, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request) {

        if (hasErrors(form, binding)) {
            return withViewname(viewnameOfStart).build();
        }

        try {
            Pair<PageSet, ResultSet> pair = search(form);
            return withViewname(viewnameOfStart).addObject(pair.getLeft()).addObject(pair.getRight()).build();
        } catch (BadSqlGrammarException ex) {
            LogicalErrorUtil.reject(binding, LogicError.BadSqlGrammer, Util.getRootCause(ex).getMessage());
            return withViewname(viewnameOfStart).build();
        }
    }

    @Override
    public ModelAndView download(SqlStatementForm form, BindingResult binding, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request, HttpServletResponse response) {

        if (hasErrors(form, binding)) {
            return withViewname(viewnameOfStart).build();
        }

        try {
            download(form, response);
            return null;
        } catch (BadSqlGrammarException ex) {
            LogicalErrorUtil.reject(binding, LogicError.BadSqlGrammer, Util.getRootCause(ex).getMessage());
            return withViewname(viewnameOfStart).build();
        }
    }

    @Override
    public ModelAndView create(SqlStatementForm form, BindingResult binding, Authentication auth, Locale locale,
            SitePreference sitePref, NativeWebRequest request) {

        if (hasErrors(form, binding)) {
            return withViewname(viewnameOfStart).build();
        }

        int id = statementService.create(form, auth.getName());
        return redirect(redirectOnCreate(id)).build();
    }

    private UriComponents redirectOnInit(String redirTo, Integer ref) {
        if (StringUtils.isNotEmpty(redirTo)) {
            return UriComponentsBuilder.fromPath(redirTo).build();
        } else {
            UriComponentsBuilder ucb = fromMethodCall(
                    on(SqlStatementController.class).start(null, null, null, null, null, null, null));
            if (ref == null) {
                return ucb.replaceQuery(StringUtils.EMPTY).build();
            } else {
                return ucb.replaceQueryParam(REQ_REF, ref).build();
            }
        }
    }

    private UriComponents redirectOnCreate(int id) {
        return fromMethodCall(on(SqlStatementIdController.class).init(null, 0, null, null, null, null))
                .replaceQueryParam(REQ_ID, id).build();
    }

    private void initializeForm(SqlStatementForm form, Integer ref, Authentication auth) {
        if (ref == null) {
            form.setDatabaseName(dataSourceDef.getDefaultName());
            return;
        }
        SqlMetadataForm mdForm = metadataService.findById(ref, auth.getName());
        if (mdForm != null) {
            SqlStatementForm f = statementService.findById(ref);
            shouldExist(f, SqlStatementForm.class, ref);
            form.setDatabaseName(f.getDatabaseName());
            form.setSql(f.getSql());
            form.setParamMap(f.getParamMap());
            form.setLockVersion(f.getLockVersion());
        }
    }

    private boolean hasErrors(SqlStatementForm form, BindingResult binding) {

        // ??
        if (binding.hasErrors()) {
            return true;
        }

        // ?

        // ??

        return false;
    }

}