package eu.siacs.conversations.xmpp.forms;

import android.os.Bundle;
import eu.siacs.conversations.xml.Element;
import eu.siacs.conversations.xml.Namespace;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;

public class Data extends Element {

    public static final String FORM_TYPE = "FORM_TYPE";

    public Data() {
        super("x", Namespace.DATA);
    }

    public List<Field> getFields() {
        ArrayList<Field> fields = new ArrayList<Field>();
        for (Element child : getChildren()) {
            if (child.getName().equals("field") && !FORM_TYPE.equals(child.getAttribute("var"))) {
                fields.add(Field.parse(child));
            }
        }
        return fields;
    }

    public Field getFieldByName(String needle) {
        for (Element child : getChildren()) {
            if (child.getName().equals("field") && needle.equals(child.getAttribute("var"))) {
                return Field.parse(child);
            }
        }
        return null;
    }

    public Field put(String name, String value) {
        Field field = getFieldByName(name);
        if (field == null) {
            field = new Field(name);
            this.addChild(field);
        }
        field.setValue(value);
        return field;
    }

    public void put(String name, Collection<String> values) {
        Field field = getFieldByName(name);
        if (field == null) {
            field = new Field(name);
            this.addChild(field);
        }
        field.setValues(values);
    }

    public void submit(final Bundle options) {
        for (final Field field : getFields()) {
            if (options.containsKey(field.getFieldName())) {
                field.setValue(options.getString(field.getFieldName()));
            }
        }
        submit();
    }

    private void submit() {
        this.setAttribute("type", "submit");
        removeUnnecessaryChildren();
        for (Field field : getFields()) {
            field.removeNonValueChildren();
        }
    }

    private void removeUnnecessaryChildren() {
        for (Iterator<Element> iterator = this.children.iterator(); iterator.hasNext(); ) {
            Element element = iterator.next();
            if (!element.getName().equals("field") && !element.getName().equals("title")) {
                iterator.remove();
            }
        }
    }

    public static Data parse(final Element element) {
        final var data = new Data();
        data.setAttributes(new HashMap<>(element.getAttributes()));
        data.setChildren(element.getChildren());
        return data;
    }

    public void setFormType(final String formType) {
        final var field = this.put(FORM_TYPE, formType);
        field.setAttribute("type", "hidden");
    }

    public String getValue(String name) {
        Field field = this.getFieldByName(name);
        return field == null ? null : field.getValue();
    }

    public String getTitle() {
        return findChildContent("title");
    }

    public static Data create(String type, Bundle bundle) {
        Data data = new Data();
        data.setFormType(type);
        data.setAttribute("type", "submit");
        for (String key : bundle.keySet()) {
            data.put(key, bundle.getString(key));
        }
        return data;
    }
}
