package notebook3;
public class Notebook3 implements Data {

    public static void main(String[] args) throws IOException, ParserConfigurationException, TransformerException {

        File file = new File(FILE_NAME);
        if (!file.exists()) {
            DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
            DocumentBuilder docBuilder = docFactory.newDocumentBuilder();
            Document xml_doc = docBuilder.newDocument();
            Element rootElement = xml_doc.createElement("notebook");
            xml_doc.appendChild(rootElement);
            XmlUtil.TransformFactory(xml_doc);
        }


        String menu = "\n<-- Press: -->\n1 - to add Contact\n2 - to display all contacts\n3 - to find Contact by name\n4 - add e-mail to contact\n5 - add phone to contact\n6 - to remove contact\n7 - exit\n";

        System.out.println(menu);

        Scanner input = new Scanner(System.in);
        String command = "";
        String error = "";
        Helper helper = new Helper();

        while (!(command = input.nextLine()).equals("7")) {
            try {
                switch (command) {
                    case "1": {
                        error = "";
                        System.out.print("Enter contact name: ");
                        String name = input.nextLine();
                        System.out.print("Enter phone number: ");
                        int phoneNumber = input.nextInt();
                        input.nextLine();
                        System.out.print("Enter e-mail: ");
                        String email = input.nextLine();
                        Contact contact = new Contact(new Name(name), new PhoneNumber(phoneNumber), new Email(email));
                        helper.addContactToXml(contact);
                        System.out.println("Note " + name + " was created..");
                    }
                    break;
                    case "2": {
                        error = "";
                        String result = helper.displayAllContacts();
                        System.out.println(result);
                    }
                    break;
                    case "3": {
                        error = "";
                        System.out.print("Enter contact name: ");
                        String name = input.nextLine();
                        String result = helper.findContactByName(name);
                        System.out.println(result);
                    }
                    break;
                    case "4": {
                        error = "";
                        System.out.print("Enter contact name: ");
                        String name = input.nextLine();
                        System.out.print("Enter e-mail: ");
                        String email = input.nextLine();
                        Contact contact = new Contact(new Name(name), new Email(email));
                        helper.addEmailToContact(contact);
                        System.out.print("e-mail added to " + name);
                    }
                    break;
                    case "5": {
                        error = "";
                        System.out.print("Enter contact name: ");
                        String name = input.nextLine();
                        System.out.print("Enter phone number: ");
                        int phoneNumber = input.nextInt();
                        Contact contact = new Contact(new Name(name), new PhoneNumber(phoneNumber));
                        helper.addPhoneNumberToContact(contact);
                        input.nextLine();
                        System.out.print("phone added to " + name);
                    }
                    break;
                    case "6": {
                        error = "";
                        System.out.print("Enter contact name: ");
                        String name = input.nextLine();
                        helper.removeContact(name);
                        System.out.println("contact " + name + " removed");
                    }
                    break;
                    case "7": {
                        error = "";
                    }
                    break;
                    default:
                        error = "Your command is not valid..";
                        break;
                }
            } catch (Exception e) {
                System.out.println("Error occured.. Please, try again..");
            }
            System.out.println(error + "" + menu);
        };

        System.out.println("\nExit..");

    }
}

package logic;
public class Contact {

    private Name name;
    private PhoneNumber phoneNumber;
    private Email email;
    
    /**
     * Class constructor Contact allows to create object Contact for create 
     * a new note in Notebook.xml
     * @param name contact name
     * @param phoneNumber contact number of telephone
     * @param email contact e-mail
     */
    public Contact(Name name, PhoneNumber phoneNumber, Email email) {
        this.name = name;
        this.phoneNumber = phoneNumber;
        this.email = email;
    }

    /**
     * Class constructor Contact allows to add a new telephone number to exist contact
     * @param name contact name
     * @param phoneNumber contact number of telephone
     */
    public Contact(Name name, PhoneNumber phoneNumber) {
        this.name = name;
        this.phoneNumber = phoneNumber;
    }

    /**
     * Class constructor Contact allows to add a new e-mail to exist contact
     * @param name contact name
     * @param email contact e-mail
     */
    public Contact(Name name, Email email) {
        this.name = name;
        this.email = email;
    }
    public Name getName() {
        return name;
    }
    public void setName(Name name) {
        this.name = name;
    }
 
    public PhoneNumber getPhoneNumber() {
        return phoneNumber;
    }
    public void setPhoneNumber(PhoneNumber phoneNumber) {
        this.phoneNumber = phoneNumber;
    }
    public Email getEmail() {
        return email;
    }
    public void setEmail(Email email) {
        this.email = email;
    }
}

package logic;
public interface Data {
    public static final String FILE_NAME = "Notebook.xml";
}

package logic;
public class Email {
    private String email;

    public Email(String email) {
        this.email = email;
    }
    public String getEmail() {
        return email;
    }
    public void setEmail(String email) {
        this.email = email;
    }
}

package logic;
public class Name {

    private String name;

    public Name(String name) {
        this.name = name;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
}

package logic;

public class PhoneNumber {
    private int phoneNumber;

    public PhoneNumber(int phoneNumber) {
        this.phoneNumber = phoneNumber;
    }

    public int getPhoneNumber() {
        return phoneNumber;
    }
    public void setPhoneNumber(int phoneNumber) {
        this.phoneNumber = phoneNumber;
    }
}

package logic;

public class Helper implements Data {
    private Document doc;
    private Element name, phone, email;   
    private NodeList contacts, contactFields, names;
    private Node contact;
 
    /**
     * This method adds a new contact to the XML-document
     * @param contact name of contact, witch necessary to add
     */
    public void addContactToXml(Contact contact) throws ParserConfigurationException, IOException, SAXException, TransformerException {

        doc = XmlUtil.getDocument();
        
        Node notebook = doc.getFirstChild();
        Element newContact = doc.createElement("contact");
        notebook.appendChild(newContact);
        name = doc.createElement("name");
        phone = doc.createElement("phone");
        email = doc.createElement("email");
        newContact.appendChild(name);
        newContact.appendChild(phone);
        newContact.appendChild(email);
        name.appendChild(doc.createTextNode(contact.getName().getName()));
        phone.appendChild(doc.createTextNode(String.valueOf(contact.getPhoneNumber().getPhoneNumber())));
        email.appendChild(doc.createTextNode(contact.getEmail().getEmail()));

        XmlUtil.TransformFactory(doc);
    }
    
    /**
     * This method adds the email to an existing contact
     * @param c contact, which necessary to add a new e-mail
     */
    public void addEmailToContact(Contact c) throws TransformerException, ParserConfigurationException, SAXException, IOException {

        doc = XmlUtil.getDocument();
        contacts = doc.getElementsByTagName("contact");
        for (int i = 0; i < contacts.getLength(); i++) {
            contact = doc.getElementsByTagName("contact").item(i);
            contactFields = contact.getChildNodes();

            for (int j = 0; j < contactFields.getLength(); j++) {
                if ((c.getName().getName()).equalsIgnoreCase(contactFields.item(j).getTextContent())) {
                    email = doc.createElement("email");
                    contact.appendChild(email);
                    email.appendChild(doc.createTextNode(c.getEmail().getEmail()));
                }
            }
        }

        XmlUtil.TransformFactory(doc);
    }
    
    /**
     * This method adds the phone number to an existing contact
     * @param c contact, which necessary add a new number of phone
     */
    public void addPhoneNumberToContact(Contact c) throws ParserConfigurationException, SAXException, IOException, TransformerException {
        doc = XmlUtil.getDocument();

        contacts = doc.getElementsByTagName("contact");
        for (int i = 0; i < contacts.getLength(); i++) {
            contact = doc.getElementsByTagName("contact").item(i);
            contactFields = contact.getChildNodes();

            for (int j = 0; j < contactFields.getLength(); j++) {
                if ((c.getName().getName()).equalsIgnoreCase(contactFields.item(j).getTextContent())) {
                    phone = doc.createElement("phone");
                    contact.appendChild(phone);
                    phone.appendChild(doc.createTextNode(String.valueOf(c.getPhoneNumber().getPhoneNumber())));
                }
            }
        }
        XmlUtil.TransformFactory(doc);
    }
    
    /**
     * This method displays all contacts
     * @return list of all contacts in Notebook.xml
     */
    public String displayAllContacts() throws ParserConfigurationException, SAXException, IOException {
        doc = XmlUtil.getDocument();
        StringBuilder result = new StringBuilder();

        contacts = doc.getElementsByTagName("contact");
        for (int i = 0; i < contacts.getLength(); i++) {
            contact = contacts.item(i);
            contactFields = contact.getChildNodes();
            result.append("\n");
            for (int j = 0; j < contactFields.getLength(); j++) {
                String title = contactFields.item(j).getNodeName().trim();
                result.append(title).append(": ").append(contactFields.item(j).getTextContent()).append("\n");
            }
        }
        return result.toString();
    }
    
    /**
     * This method searchs a contact by name
     * @param name name of contact, which necessary to find
     * @return contact, which you find
     */
    public String findContactByName(String name) throws ParserConfigurationException, SAXException, IOException {
        Node findName;      
        StringBuilder result = new StringBuilder();
        doc = XmlUtil.getDocument();

        names = doc.getElementsByTagName("name");
        for (int i = 0; i < names.getLength(); i++) {
            findName = names.item(i);
            if (name.equalsIgnoreCase(findName.getTextContent())) {
                contact = findName.getParentNode();
                contactFields = contact.getChildNodes();
                for (int j = 0; j < contactFields.getLength(); j++) {
                    String title = contactFields.item(j).getNodeName().trim();
                    result.append(title).append(": ").append(contactFields.item(j).getTextContent()).append("\n");
                }
            }
        }

        return result.toString();
    }

    /**
     * This method removes a contact by name
     * @param name contact, which necessary to remove
     */
    public void removeContact(String name) throws ParserConfigurationException, SAXException, IOException, TransformerException {      
        doc = XmlUtil.getDocument();
        names = doc.getElementsByTagName("name");

        for (int i = 0; i < names.getLength(); i++) {
            Node contactName = names.item(i);

            if (name.equalsIgnoreCase(contactName.getTextContent())) {
                contact = contactName.getParentNode();
                NodeList nodesForRemove = contact.getChildNodes();
                while (nodesForRemove.getLength() > 0) {
                    Node removingNode = nodesForRemove.item(0);
                    removingNode.getParentNode().removeChild(removingNode);
                }
            }
        }

        contacts = doc.getElementsByTagName("contact");
        for (int i = 0; i < contacts.getLength(); i++) {
            contact = contacts.item(i);
            if (!contact.hasChildNodes()) {
                contact.getParentNode().removeChild(contact);
            }
        }

        XmlUtil.TransformFactory(doc);

    }
}

package logic;

public class XmlUtil implements Data{
    public static Document getDocument() throws  SAXException, IOException, ParserConfigurationException{
        DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
        DocumentBuilder docBuilder = docFactory.newDocumentBuilder();
        Document doc = docBuilder.parse(FILE_NAME);
        return doc;
    }
    
    public static void TransformFactory(Document doc) throws TransformerException {
        TransformerFactory transformerFactory = TransformerFactory.newInstance();
        Transformer transformer = transformerFactory.newTransformer();
        DOMSource source = new DOMSource(doc);
        StreamResult result = new StreamResult(new File(FILE_NAME));
        transformer.transform(source, result);
    }
}