/*
 * SoapUI, Copyright (C) 2004-2016 SmartBear Software 
 *
 * Licensed under the EUPL, Version 1.1 or - as soon as they will be approved by the European Commission - subsequent 
 * versions of the EUPL (the "Licence"); 
 * You may not use this work except in compliance with the Licence. 
 * You may obtain a copy of the Licence at: 
 * 
 * http://ec.europa.eu/idabc/eupl 
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the Licence is 
 * distributed on an "AS IS" basis, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 
 * express or implied. See the Licence for the specific language governing permissions and limitations 
 * under the Licence. 
 */

package com.eviware.soapui.impl.rest.panels.request.views.content;

import com.eviware.soapui.impl.rest.RestRepresentation;
import com.eviware.soapui.impl.rest.RestRequestInterface;
import com.eviware.soapui.impl.rest.panels.resource.RestParamsTable;
import com.eviware.soapui.impl.rest.panels.resource.RestParamsTableModel;
import com.eviware.soapui.impl.rest.support.RestUtils;
import com.eviware.soapui.impl.support.http.HttpRequestContentView;
import com.eviware.soapui.impl.support.panels.AbstractHttpXmlRequestDesktopPanel.HttpRequestMessageEditor;
import com.eviware.soapui.impl.wsdl.support.xsd.SampleXmlUtil;
import com.eviware.soapui.model.iface.Request;
import com.eviware.soapui.support.UISupport;
import com.eviware.soapui.support.components.JXToolBar;
import com.eviware.soapui.support.types.StringList;
import com.eviware.soapui.support.types.TupleList;
import org.apache.xmlbeans.SchemaType;

import javax.swing.AbstractAction;
import javax.swing.Action;
import javax.swing.JButton;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.event.ActionEvent;
import java.beans.PropertyChangeEvent;

import static com.eviware.soapui.impl.rest.actions.support.NewRestResourceActionBase.ParamLocation;

public class RestRequestContentView extends HttpRequestContentView {
    private RestRequestInterface restRequest;
    private JButton recreateButton;

    @SuppressWarnings("unchecked")
    public RestRequestContentView(HttpRequestMessageEditor restRequestMessageEditor, RestRequestInterface restRequest) {
        super(restRequestMessageEditor, restRequest);
        this.restRequest = restRequest;
    }


    protected RestParamsTable buildParamsTable() {
        RestParamsTableModel model = new RestParamsTableModel(restRequest.getParams()) {
            @Override
            public String getColumnName(int columnIndex) {
                if (columnIndex == 1) {
                    return "Value";
                }

                return super.getColumnName(columnIndex);
            }
        };
        return new RestParamsTable(restRequest.getParams(), true, model, ParamLocation.RESOURCE, true, true);
    }

    public RestParamsTable getParamsTable() {
        return paramsTable;
    }

    protected Component buildToolbar() {
        JXToolBar toolbar = UISupport.createToolbar();

        addMediaTypeCombo(toolbar);
        toolbar.addSeparator();

        recreateButton = UISupport.createActionButton(new CreateDefaultRepresentationAction(), true);
        recreateButton.setEnabled(canRecreate());
        toolbar.addFixed(recreateButton);

        toolbar.addSeparator();

        addPostQueryCheckBox(toolbar);

        toolbar.setMinimumSize(new Dimension(50, 20));

        return toolbar;
    }

    public void propertyChange(PropertyChangeEvent evt) {
        if (evt.getPropertyName().equals(Request.MEDIA_TYPE) && recreateButton != null) {
            recreateButton.setEnabled(canRecreate());
        } else if (evt.getPropertyName().equals("restMethod")) {
            paramsTable.setParams(restRequest.getParams());
        }

        super.propertyChange(evt);
    }

    protected Object[] getRequestMediaTypes() {
        StringList result = new StringList(super.getRequestMediaTypes());

        for (RestRepresentation representation : restRequest.getRepresentations(RestRepresentation.Type.REQUEST, null)) {
            if (!result.contains(representation.getMediaType())) {
                result.add(representation.getMediaType());
            }
        }

        return result.toStringArray();
    }

    private boolean canRecreate() {
        for (RestRepresentation representation : restRequest.getRepresentations(RestRepresentation.Type.REQUEST,
                restRequest.getMediaType())) {
            if (representation.getSchemaType() != null) {
                return true;
            }
        }
        return false;
    }

    private class UpdateRestParamsAction extends AbstractAction {
        private UpdateRestParamsAction() {
            putValue(Action.SMALL_ICON, UISupport.createImageIcon("/update-request-parameters-from-url.png"));
            putValue(Action.SHORT_DESCRIPTION, "Updates this Requests params from a specified URL");
        }

        public void actionPerformed(ActionEvent e) {
            String str = UISupport.prompt("Enter new url below", "Extract Params", "");
            if (str == null) {
                return;
            }

            try {
                restRequest.getParams().resetValues();
                RestUtils.extractParams(str, restRequest.getParams(), false);
                paramsTable.refresh();
            } catch (Exception e1) {
                UISupport.showErrorMessage(e1);
            }
        }
    }

    private class CreateDefaultRepresentationAction extends AbstractAction {
        private CreateDefaultRepresentationAction() {
            putValue(Action.SMALL_ICON, UISupport.createImageIcon("/recreate_request.gif"));
            putValue(Action.SHORT_DESCRIPTION, "Recreates a default representation from the schema");
        }

        @SuppressWarnings("unchecked")
        public void actionPerformed(ActionEvent e) {
            TupleList<RestRepresentation, SchemaType> list = new TupleList<RestRepresentation, SchemaType>() {
                protected String toStringHandler(Tuple tuple) {
                    return tuple.getValue2().getDocumentElementName().toString();
                }
            };

            for (RestRepresentation representation : ((RestRequestInterface) restRequest).getRepresentations(
                    RestRepresentation.Type.REQUEST, restRequest.getMediaType())) {
                SchemaType schemaType = representation.getSchemaType();
                if (schemaType != null) {
                    list.add(representation, schemaType);
                }
            }

            if (list.isEmpty()) {
                UISupport.showErrorMessage("Missing recreatable representations for this method");
                return;
            }

            TupleList<RestRepresentation, SchemaType>.Tuple result = (TupleList.Tuple) UISupport.prompt(
                    "Select element to create", "Create default content", list.toArray());
            if (result == null) {
                return;
            }

            restRequest.setRequestContent(SampleXmlUtil.createSampleForType(result.getValue2()));
        }
    }
}