OrgNode.java

package com.hdweiss.morgand.data.dao;

import android.text.TextUtils;

import com.hdweiss.morgand.data.OrgCalendarEntry;
import com.hdweiss.morgand.data.OrgNodeUtils;
import com.hdweiss.morgand.settings.PreferenceUtils;
import com.j256.ormlite.dao.ForeignCollection;
import com.j256.ormlite.field.DatabaseField;
import com.j256.ormlite.field.ForeignCollectionField;
import com.j256.ormlite.stmt.Where;
import com.j256.ormlite.table.DatabaseTable;

import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@DatabaseTable(tableName = "OrgNodes")
public class OrgNode {

    public enum Type {
        File, Directory, Date, Headline, Body, Drawer, Checkbox, Setting, Table
    }

    public enum State {
        Clean, Updated, Added, Deleted
    }

    @DatabaseField(generatedId = true)
    public int Id;

    public static final String PARENT_FIELD_NAME = "parent";
    @DatabaseField(foreign = true, columnName = PARENT_FIELD_NAME, foreignAutoRefresh=true, canBeNull=true, maxForeignAutoRefreshLevel=3)
    public OrgNode parent;

    @ForeignCollectionField(eager = false, foreignFieldName = PARENT_FIELD_NAME)
    public ForeignCollection<OrgNode> children;

    @DatabaseField
    public Type type;

    public static final String LEVEL_FIELD_NAME = "level";
    @DatabaseField(columnName = LEVEL_FIELD_NAME)
    public int level = 0;

    public static final String LINENUMBER_FIELD_NAME = "lineNumber";
    @DatabaseField(columnName =  LINENUMBER_FIELD_NAME)
    public int lineNumber = -1;

    public static final String STATE_FIELD_NAME = "state";
    @DatabaseField(columnName = STATE_FIELD_NAME)
    public State state = State.Clean;

    public static final String FILE_FIELD_NAME = "file";
    @DatabaseField(foreign =  true, columnName = FILE_FIELD_NAME)
    public OrgFile file;


    public static final String TITLE_FIELD_NAME = "title";
    @DatabaseField(columnName = TITLE_FIELD_NAME)
    public String title = "";

    @DatabaseField
    public String tags = "";

    @DatabaseField
    public String inheritedTags = "";


    public String toString() {
        if (type != Type.Headline)
            return title;

        StringBuilder builder = new StringBuilder();
        for (int i = 0; i < level; i++)
            builder.append("*");
        builder.append(" ");

        if (TextUtils.isEmpty(title) == false) {
            builder.append(title.trim());
            builder.append("\t" + tags);
        }
        return builder.toString();
    }

    public String toStringRecursively() {
        StringBuilder builder = new StringBuilder(toString());

        for(OrgNode child : children)
            builder.append("\n").append(child.toStringRecursively());

        return builder.toString();
    }


    public boolean isEditable() {
        if (type == Type.File || type == Type.Directory)
            return false;

        if (file.isEditable() == false)
            return false;

        return true;
    }

    public String getProperty(String propertyName) {
        if (type == Type.Headline) {
            Pattern propertyPattern = Pattern.compile(":" + propertyName + ":\\s+(.+)");
            for(OrgNode child: children) {
                if (child.type == Type.Drawer) {
                    Matcher matcher = propertyPattern.matcher(child.title);
                    if (matcher.find()) {
                        return matcher.group(1).trim();
                    }
                }
            }

            return null;
        }

        if (parent != null)
            return parent.getProperty(propertyName);
        return null;
    }

    public String getTitle() {
        if (type != Type.Headline) {
            if (parent != null) {
                return parent.getTitle();
            }
            else {
                return "";
            }
        }

        return title;
    }

    public String getTodo() {
        if (type == Type.Headline) {

            Matcher matcher = OrgNodeUtils.headingPattern.matcher(title);
            if (matcher.find())
                return matcher.group(1);
        }
        return "";
    }

    public String getBody() {
        if (type == Type.Headline) {
            StringBuilder body = new StringBuilder();
            for(OrgNode child: children) {
                if (child.type == Type.Body)
                    body.append(child.title);
            }
            return body.toString();
        } else if (parent != null)
            return  parent.getBody();

        return title;
    }

    public ArrayList<OrgNode> getDisplayChildren() {
        ArrayList<OrgNode> nodes = new ArrayList<OrgNode>();
        boolean showSettings = PreferenceUtils.showSettings();
        boolean showDrawers = PreferenceUtils.showDrawers();
        for(OrgNode node: children) {
            if (node.state == State.Deleted)
                continue;

            switch (node.type) {
                case Drawer:
                    if (showDrawers)
                        nodes.add(node);
                    break;

                case Setting:
                    if (showSettings)
                        nodes.add(node);
                    break;

                default:
                    nodes.add(node);
            }
        }
        return nodes;
    }

    public OrgNode addChild(Type type) {
        OrgNode node = new OrgNode();
        node.parent = this;
        node.file = this.file;
        node.type = type;
        node.level = this.level + 1;
        node.inheritedTags = OrgNodeUtils.combineTags(tags, inheritedTags, PreferenceUtils.getExcludedTags());
        return node;
    }

    public OrgNode addChild(Type type, String title) {
        OrgNode node = addChild(type);
        node.title = title;

        OrgNodeRepository.create(node);
        return node;
    }

    public int getNextNodeLineNumber() {
        try {
            Where<OrgNode, Integer> where = OrgNodeRepository.queryBuilder().orderBy(LINENUMBER_FIELD_NAME, true).where();
            where.eq(FILE_FIELD_NAME, file).and().gt(LINENUMBER_FIELD_NAME, lineNumber);
            OrgNode node = where.queryForFirst();
            if (node != null)
                return node.lineNumber;
            else
                return Integer.MAX_VALUE; // Current node seems to be last in file
        } catch (SQLException e) {
            e.printStackTrace();
        }

        return -1;
    }

    /**
     * @return line number of next node on same or higher level than current node.
     */
    public int getSiblingLineNumber() {
        try {
            Where<OrgNode, Integer> where = OrgNodeRepository.queryBuilder().orderBy(LINENUMBER_FIELD_NAME, true).where();
            where.eq(FILE_FIELD_NAME, file).and().gt(LINENUMBER_FIELD_NAME, lineNumber).and().le(LEVEL_FIELD_NAME, level);
            OrgNode node = where.queryForFirst();
            if (node != null) {
                if (node.type != Type.File)
                    return node.lineNumber;
            }
            else
                return Integer.MAX_VALUE; // Current node seems to be last in file
        } catch (SQLException e) {
            e.printStackTrace();
        }

        return -1;
    }

    public boolean isNodeWritable() {
        if (parent != null) {
            if (parent.state == State.Added)
                return false;

            if (parent.state == State.Deleted)
                return false;

            return parent.isNodeWritable();
        }

        return true;
    }

    public List<OrgCalendarEntry> getOrgNodeDates() {
        ArrayList<OrgCalendarEntry> dates = new ArrayList<OrgCalendarEntry>();

        Matcher matcher = OrgNodeUtils.dateMatcher.matcher(title);
        while(matcher.find()) {
            String type = matcher.group(1);
            String startDate = matcher.group(2);
            String endDate = matcher.group(3); // TODO Support day ranges

            try {
                OrgCalendarEntry orgCalendarEntry = new OrgCalendarEntry(startDate);
                orgCalendarEntry.type = type != null ? type : "";

                orgCalendarEntry.setTitle(getTitle());
                dates.add(orgCalendarEntry);
            } catch (IllegalArgumentException ex) {}
        }

        return dates;
    }


    public static class OrgNodeCompare implements Comparator<OrgNode> {
        @Override
        public int compare(OrgNode node1, OrgNode node2) {
            return node1.title.compareTo(node2.title);
        }
    }
}