All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.igormaznitsa.mindmap.model.Topic Maven / Gradle / Ivy

/*
 * Copyright (C) 2015-2022 Igor A. Maznitsa
 *
 * 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 com.igormaznitsa.mindmap.model;

import static com.igormaznitsa.mindmap.model.MiscUtils.ensureNoNullElement;
import static java.util.Objects.requireNonNull;

import com.igormaznitsa.mindmap.model.parser.MindMapLexer;
import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.EnumMap;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Set;
import java.util.TreeMap;
import java.util.concurrent.atomic.AtomicLong;
import java.util.function.Predicate;
import java.util.regex.Pattern;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;

/**
 * Class describes the main work unit for Mind Map.
 */
public final class Topic implements Serializable, Constants, Iterable {

  private static final long serialVersionUID = -4642569244907433215L;
  private static final AtomicLong LOCAL_UID_GENERATOR = new AtomicLong();
  private final EnumMap> extras =
      new EnumMap<>(Extra.ExtraType.class);
  private final Map attributes =
      new TreeMap<>(Comparator.naturalOrder());
  private final Map codeSnippets =
      new TreeMap<>(Comparator.naturalOrder());
  private final List children = new ArrayList<>();
  private final transient long localUID = LOCAL_UID_GENERATOR.getAndIncrement();
  private final MindMap map;
  private Topic parent;
  private volatile String text;
  private transient Object payload;

  /**
   * Constructor to build topic on base of another topic for another mind map.
   *
   * @param mindMap      mind map to be owner for new topic
   * @param base         base source topic
   * @param copyChildren flag to make copy of children, true if to make copy,
   *                     false otherwise
   */
  public Topic(final MindMap mindMap, final Topic base,
               final boolean copyChildren) {
    this(mindMap, null, base.text);
    this.attributes.putAll(base.attributes);
    this.extras.putAll(base.extras);
    this.codeSnippets.putAll(base.codeSnippets);

    if (copyChildren) {
      for (final Topic t : base.children) {
        final Topic clonedChildren = new Topic(mindMap, t, true);
        clonedChildren.parent = this;
        this.children.add(clonedChildren);
      }
    }
  }

  /**
   * Constructor
   *
   * @param map    parent map, must not be null
   * @param parent parent topic can be null
   * @param text   topic text, must not be null
   * @param extras extras for the topic, must not be null
   */
  public Topic(final MindMap map, final Topic parent, final String text,
               final Extra... extras) {
    this.map = requireNonNull(map);
    this.text = requireNonNull(text);

    for (final Extra e : extras) {
      if (e != null) {
        this.extras.put(e.getType(), e);
      }
    }
    this.parent = parent;

    if (parent != null) {
      if (parent.getMap() != map) {
        throw new IllegalArgumentException("Parent must belong to the same mind map");
      }
      parent.children.add(this);
    }
  }

  public static Topic parse(final MindMap map, final MindMapLexer lexer,
                            final boolean ignoreErrors) {
    Topic topic = null;
    int depth = 0;

    Extra.ExtraType extraType = null;

    String codeSnippet = null;
    StringBuilder codeSnippetBody = null;

    int detectedLevel = -1;

    while (true) {
      final int oldLexerPosition = lexer.getCurrentPosition().getOffset();
      lexer.advance();
      final boolean lexerPositionWasNotChanged =
          oldLexerPosition == lexer.getCurrentPosition().getOffset();

      final MindMapLexer.TokenType token = lexer.getTokenType();
      if (token == null || lexerPositionWasNotChanged) {
        break;
      }

      switch (token) {
        case TOPIC_LEVEL: {
          final String tokenText = lexer.getTokenText();
          detectedLevel = ModelUtils.countPrefixChars('#', tokenText);
        }
        break;
        case TOPIC_TITLE: {
          final String tokenText = ModelUtils.removeISOControls(lexer.getTokenText());
          final String newTopicText = ModelUtils.unescapeMarkdown(tokenText);

          if (detectedLevel == depth + 1) {
            depth = detectedLevel;
            topic = new Topic(map, topic, newTopicText);
          } else if (detectedLevel == depth) {
            topic = new Topic(map, topic == null ? null : topic.getParent(), newTopicText);
          } else if (detectedLevel < depth) {
            if (topic != null) {
              topic = topic.findParentForDepth(depth - detectedLevel);
              topic = new Topic(map, topic, newTopicText);
              depth = detectedLevel;
            }
          }

        }
        break;
        case EXTRA_TYPE: {
          final String extraName = lexer.getTokenText().substring(1).trim();
          try {
            extraType = Extra.ExtraType.valueOf(extraName);
          } catch (IllegalArgumentException ex) {
            extraType = null;
          }
        }
        break;
        case CODE_SNIPPET_START: {
          if (topic != null) {
            codeSnippet = lexer.getTokenText().substring(3);
            codeSnippetBody = new StringBuilder();
          }
        }
        break;
        case CODE_SNIPPET_BODY: {
          codeSnippetBody.append(lexer.getTokenText());
        }
        break;
        case CODE_SNIPPET_END: {
          if (topic != null && codeSnippet != null && codeSnippetBody != null) {
            topic.codeSnippets.put(codeSnippet.trim(), codeSnippetBody.toString());
          }
          codeSnippet = null;
          codeSnippetBody = null;
        }
        break;
        case ATTRIBUTE: {
          if (topic != null) {
            final String text = lexer.getTokenText().trim();
            MindMap.fillMapByAttributes(text, topic.attributes);
          }
          extraType = null;
        }
        break;
        case EXTRA_TEXT: {
          if (topic != null && extraType != null) {
            try {
              final String text = lexer.getTokenText();
              final String groupPre =
                  extraType.preprocessString(text.substring(5, text.length() - 6));
              if (groupPre != null) {
                topic.setExtra(extraType.parseLoaded(groupPre, topic.attributes));
              } else {
                if (!ignoreErrors) {
                  throw new IllegalStateException("Detected invalid extra data " + extraType);
                }
              }
            } catch (Exception ex) {
              throw new Error("Unexpected exception #23241", ex);
            } finally {
              extraType = null;
            }
          }
        }
        break;
        case UNKNOWN_LINE: {
          if (topic != null && extraType != null) {
            extraType = null;
          }
        }
        break;
        default:
          break;
      }
    }
    return topic == null ? null : topic.getRoot();
  }

  public Topic findRoot() {
    Topic result = this;
    while (!result.isRoot()) {
      result = result.getParent();
    }
    return result;
  }

  public boolean containTopic(final Topic topic) {
    boolean result = false;

    if (this == topic) {
      result = true;
    } else {
      for (final Topic t : this.children) {
        if (t.containTopic(topic)) {
          result = true;
          break;
        }
      }
    }
    return result;
  }

  public Topic nextSibling() {
    final int position = this.parent == null ? -1 : this.parent.getChildren().indexOf(this);

    final Topic result;
    if (position < 0) {
      result = null;
    } else {
      final List all = this.parent.getChildren();
      final int nextPosition = position + 1;
      result = all.size() > nextPosition ? all.get(nextPosition) : null;
    }
    return result;
  }

  public Topic prevSibling() {
    final int position = this.parent == null ? -1 : this.parent.getChildren().indexOf(this);

    final Topic result;
    if (position <= 0) {
      result = null;
    } else {
      final List all = this.parent.getChildren();
      result = all.get(position - 1);
    }
    return result;
  }

  public boolean containsPattern(final File baseFolder, final Pattern pattern,
                                 final boolean findInTopicText,
                                 final Set extrasForSearch) {
    boolean result = false;

    if (findInTopicText && pattern.matcher(this.text).find()) {
      result = true;
    } else if (extrasForSearch != null && !extrasForSearch.isEmpty()) {
      for (final Extra e : this.extras.values()) {
        if (extrasForSearch.contains(e.getType()) && e.containsPattern(baseFolder, pattern)) {
          result = true;
          break;
        }
      }
    }
    return result;
  }

  public boolean isRoot() {
    return this.parent == null;
  }

  public Object getPayload() {
    return this.payload;
  }

  public void setPayload(final Object value) {
    this.payload = value;
  }

  private Object readResolve() {
    return new Topic(this.map, this, true);
  }

  public MindMap getMap() {
    return this.map;
  }

  public int getTopicLevel() {
    Topic topic = this.parent;
    int result = 0;
    while (topic != null) {
      topic = topic.parent;
      result++;
    }
    return result;
  }

  public Topic findParentForDepth(int depth) {
    Topic result = this.parent;
    while (depth > 0 && result != null) {
      result = result.parent;
      depth--;
    }
    return result;
  }

  public Topic getRoot() {
    Topic result = this;
    while (true) {
      final Topic prev = result.parent;
      if (prev == null) {
        break;
      }
      result = prev;
    }
    return result;
  }

  public Topic getFirst() {
    return this.children.isEmpty() ? null : this.children.get(0);
  }

  public Topic getLast() {
    return this.children.isEmpty() ? null : this.children.get(this.children.size() - 1);
  }

  public List getChildren() {
    return this.children;
  }

  public boolean isExtrasEmpty() {
    return this.extras.isEmpty();
  }

  public Map> getExtras() {
    return this.extras;
  }

  public Map getAttributes() {
    return this.attributes;
  }

  public Map getCodeSnippets() {
    return this.codeSnippets;
  }

  public boolean putAttribute(final String name, final String value) {
    if (value == null) {
      return this.attributes.remove(name) != null;
    } else {
      return !value.equals(this.attributes.put(name, value));
    }
  }

  public boolean putCodeSnippet(final String language, final String text) {
    if (text == null) {
      return this.codeSnippets.remove(language) != null;
    } else {
      return !text.equals(this.codeSnippets.put(language, text));
    }
  }

  public String getCodeSnippet(final String language) {
    return this.codeSnippets.get(language);
  }

  public String getAttribute(final String name) {
    return this.attributes.get(name);
  }

  public void delete() {
    final Topic theParent = this.parent;
    if (theParent != null) {
      theParent.children.remove(this);
    }
  }

  public Topic getParent() {
    return this.parent;
  }

  public String getText() {
    return this.text;
  }

  public void setText(final String text) {
    this.text = requireNonNull(text);
  }

  public boolean isFirstChild(final Topic t) {
    return !this.children.isEmpty() && this.children.get(0) == t;
  }

  public boolean isLastChild(final Topic t) {
    return !this.children.isEmpty() && this.children.get(this.children.size() - 1) == t;
  }

  public boolean removeExtra(final Extra.ExtraType... types) {
    boolean result = false;
    for (final Extra.ExtraType e : ensureNoNullElement(types)) {
      final Extra removed = this.extras.remove(e);
      if (removed != null) {
        removed.detachedToTopic(this);
      }
      result |= removed != null;
    }
    return result;
  }

  public void setExtra(final Extra... extras) {
    for (final Extra e : ensureNoNullElement(extras)) {
      this.extras.put(e.getType(), e);
      e.attachedToTopic(this);
    }
  }

  public boolean makeFirst() {
    final Topic theParent = this.parent;
    if (theParent != null) {
      int thatIndex = theParent.children.indexOf(this);
      if (thatIndex > 0) {
        theParent.children.remove(thatIndex);
        theParent.children.add(0, this);
        return true;
      }
    }
    return false;
  }

  public boolean hasAncestor(final Topic topic) {
    Topic parent = this.parent;
    while (parent != null) {
      if (parent == topic) {
        return true;
      }
      parent = parent.getParent();
    }
    return false;
  }

  public boolean makeLast() {
    final Topic theParent = this.parent;
    if (theParent != null) {
      int thatIndex = theParent.children.indexOf(this);
      if (thatIndex >= 0 && thatIndex != theParent.children.size() - 1) {
        theParent.children.remove(thatIndex);
        theParent.children.add(this);
        return true;
      }
    }
    return false;
  }

  public void moveBefore(final Topic topic) {
    final Topic theParent = this.parent;
    if (theParent != null) {
      int thatIndex = theParent.children.indexOf(topic);
      final int thisIndex = theParent.children.indexOf(this);

      if (thatIndex > thisIndex) {
        thatIndex--;
      }

      if (thatIndex >= 0 && thisIndex >= 0) {
        theParent.children.remove(this);
        theParent.children.add(thatIndex, this);
      }
    }
  }

  public String findAttributeInAncestors(final String attrName) {
    String result = null;
    Topic current = this.parent;
    while (result == null && current != null) {
      result = current.getAttribute(attrName);
      current = current.parent;
    }
    return result;
  }

  public void moveAfter(final Topic topic) {
    final Topic theParent = this.parent;
    if (theParent != null) {
      int thatIndex = theParent.children.indexOf(topic);
      int thisIndex = theParent.children.indexOf(this);

      if (thatIndex > thisIndex) {
        thatIndex--;
      }

      if (thatIndex >= 0 && thisIndex >= 0) {
        theParent.children.remove(this);
        theParent.children.add(thatIndex + 1, this);
      }
    }
  }

  public void write(final Writer out) throws IOException {
    write(1, out);
  }

  private void write(final int level, final Writer out) throws IOException {
    out.append(NEXT_LINE);
    ModelUtils.repeatChar(out, '#', level);
    out.append(' ').append(ModelUtils.escapeMarkdown(this.text)).append(NEXT_LINE);

    if (!this.attributes.isEmpty() || !this.extras.isEmpty()) {
      final Map attributesToWrite = new HashMap<>(this.attributes);
      for (final Map.Entry> e : this.extras.entrySet()) {
        e.getValue().addAttributesForWrite(attributesToWrite);
      }

      if (!attributesToWrite.isEmpty()) {
        out.append("> ").append(MindMap.allAttributesAsString(attributesToWrite)).append(NEXT_LINE)
            .append(NEXT_LINE);
      }
    }

    if (!this.extras.entrySet().isEmpty()) {
      final List types = new ArrayList<>(this.extras.keySet());
      types.sort(Comparator.comparing(Enum::name));

      for (final Extra.ExtraType e : types) {
        this.extras.get(e).write(out);
        out.append(NEXT_LINE);
      }
    }

    if (!this.codeSnippets.isEmpty()) {
      final List sortedKeys = new ArrayList<>(this.codeSnippets.keySet());
      Collections.sort(sortedKeys);
      for (final String language : sortedKeys) {
        final String body = this.codeSnippets.get(language);
        out.append("```").append(language).append(NEXT_LINE);
        out.append(body);
        if (!body.endsWith("\n")) {
          out.append(NEXT_LINE);
        }
        out.append("```").append(NEXT_LINE);
      }
    }

    for (final Topic t : this.children) {
      t.write(level + 1, out);
    }
  }

  /**
   * Sort child topics by provided comparator.
   *
   * @param topicComparator comparator to be used for sort, must not be null.
   * @param sortChildren    flag if true then child elements also must be sorted
   * @since 1.6.0
   */
  public void sortChildren(final Comparator topicComparator, final boolean sortChildren) {
    this.children.sort(topicComparator);
    if (sortChildren) {
      this.children.forEach(x -> x.sortChildren(topicComparator, true));
    }
  }

  @Override
  public int hashCode() {
    return (int) ((this.localUID >>> 32) ^ (this.localUID & 0xFFFFFFFFL));
  }

  @Override
  public boolean equals(final Object topic) {
    if (this == topic) {
      return true;
    }
    if (topic instanceof Topic) {
      return this.localUID == ((Topic) topic).localUID;
    }
    return false;
  }

  @Override
  public String toString() {
    return "MindMapTopic('" + this.text + ':' + this.getLocalUid() + "')";
  }

  public long getLocalUid() {
    return this.localUID;
  }

  public boolean isEmpty() {
    return this.children.isEmpty();
  }

  boolean removeAllLinksTo(final Topic topic) {
    boolean result = false;
    if (topic != null) {
      final String uid = topic.getAttribute(ExtraTopic.TOPIC_UID_ATTR);
      if (uid != null) {
        final ExtraTopic link = (ExtraTopic) this.getExtras().get(Extra.ExtraType.TOPIC);
        if (link != null && uid.equals(link.getValue())) {
          this.removeExtra(Extra.ExtraType.TOPIC);
          result = true;
        }
      }

      for (final Topic ch : this.children) {
        result |= ch.removeAllLinksTo(topic);
      }
    }

    return result;
  }

  boolean removeTopic(final Topic topic) {
    if (topic == null) {
      return false;
    }
    final Iterator iterator = this.children.iterator();
    while (iterator.hasNext()) {
      final Topic t = iterator.next();
      if (t == topic) {
        iterator.remove();
        return true;
      } else if (t.removeTopic(topic)) {
        return true;
      }
    }
    return false;
  }

  public void removeAllChildren() {
    this.children.clear();
  }

  public boolean moveToNewParent(final Topic newParent) {
    if (newParent == null || this == newParent || this.getParent() == newParent ||
        this.children.contains(newParent)) {
      return false;
    }

    final Topic theParent = this.parent;
    if (theParent != null) {
      theParent.children.remove(this);
    }
    newParent.children.add(this);
    this.parent = newParent;

    return true;
  }

  public Topic makeChild(final String text, final Topic afterTheTopic) {
    final Topic result = new Topic(this.map, this, MiscUtils.ensureNotNull(text, ""));
    if (afterTheTopic != null && this.children.contains(afterTheTopic)) {
      result.moveAfter(afterTheTopic);
    }
    return result;
  }

  public Topic findNext(final Predicate checker) {
    Topic result = null;
    Topic current = this.getParent();
    if (current != null) {
      final int indexThis = current.children.indexOf(this);
      if (indexThis >= 0) {
        for (int i = indexThis + 1; i < current.children.size(); i++) {
          if (checker == null) {
            result = current.children.get(i);
            break;
          } else if (checker.test(current.children.get(i))) {
            result = current.children.get(i);
            break;
          }
        }
      }
    }

    return result;
  }

  public Topic findPrev(final Predicate checker) {
    Topic result = null;
    Topic current = this.getParent();
    if (current != null) {
      final int indexThis = current.children.indexOf(this);
      if (indexThis >= 0) {
        for (int i = indexThis - 1; i >= 0; i--) {
          if (checker.test(current.children.get(i))) {
            result = current.children.get(i);
            break;
          }
        }
      }
    }

    return result;
  }

  public void removeExtras(final Extra... extras) {
    if (extras == null || extras.length == 0) {
      this.extras.clear();
    } else {
      for (final Extra e : extras) {
        if (e != null) {
          this.extras.remove(e.getType());
        }
      }
    }
  }

  /**
   * Find max length of children chain. It doesn't count the root topic.
   *
   * @return max length of child chain, 0 if no children.
   */
  public int findMaxChildPathLength() {
    int len = 0;
    for (final Topic t : this.getChildren()) {
      final int childLen = t.findMaxChildPathLength();
      len = Math.max(len, childLen + 1);
    }
    return len;
  }

  /**
   * Find among subtree first topic contains attibute with value
   *
   * @param attributeName  name of attribute, must not be null
   * @param attributeValue value to be checked, must not be null
   * @return first topic in subtree with such attribute value, null if not found
   */
  public Topic findForAttribute(final String attributeName, String attributeValue) {
    if (attributeValue.equals(this.getAttribute(attributeName))) {
      return this;
    }
    Topic result = null;
    for (final Topic c : this.children) {
      result = c.findForAttribute(attributeName, attributeValue);
      if (result != null) {
        break;
      }
    }
    return result;
  }

  /**
   * Make position path to the topic
   *
   * @return array of indexes in parents, must not be null
   */
  public int[] getPositionPath() {
    final Topic[] path = this.getPath();
    final int[] result = new int[path.length];

    Topic current = path[0];
    int index = 1;
    while (index < path.length) {
      final Topic next = path[index];
      final int theindex = current.children.indexOf(next);
      result[index++] = theindex;
      if (theindex < 0) {
        break;
      }
      current = next;
    }

    return result;
  }

  /**
   * Collect all ancestors of the topic and build path
   *
   * @return path to the topic with all ancestors, must not be null
   */
  public Topic[] getPath() {
    final List list = new ArrayList<>();
    Topic current = this;
    do {
      list.add(0, current);
      current = current.parent;
    }
    while (current != null);
    return list.toArray(new Topic[0]);
  }

  /**
   * Make copy of the topic in the target mind map
   *
   * @param targetMindMap target mind map, must not be null
   * @param parent        parent topic, can be null
   * @param withChildren  flag shows that subtree must be also copied if true
   * @return result topic, must not be null
   */
  public Topic makeCopy(final MindMap targetMindMap, final Topic parent,
                        final boolean withChildren) {
    final Topic newTopic = new Topic(
        targetMindMap,
        parent,
        this.text,
        this.extras.values().toArray(new Extra[0])
    );
    if (withChildren) {
      for (final Topic c : this.children) {
        c.makeCopy(targetMindMap, newTopic, withChildren);
      }
    }
    newTopic.attributes.putAll(this.attributes);
    newTopic.codeSnippets.putAll(this.codeSnippets);

    return newTopic;
  }

  /**
   * Make copy of the topic and its subtree in the target mind map
   *
   * @param targetMindMap target mind map, must not be null
   * @param parent        parent topic, can be null
   * @return result topic, must not be null
   */
  public Topic makeCopy(final MindMap targetMindMap, final Topic parent) {
    return this.makeCopy(targetMindMap, parent, true);
  }

  /**
   * Remove all extras from the topic
   *
   * @param includeSubtree if true then remove all extras from all subtree also
   * @param types          extra types to be removed
   * @return true if any extra was found and removed, false otherwise
   */
  public boolean removeAllExtras(
      final boolean includeSubtree,
      final Extra.ExtraType... types) {
    boolean result = false;

    for (final Extra.ExtraType t : types) {
      result |= this.extras.remove(t) != null;
    }
    if (includeSubtree) {
      for (final Topic c : this.children) {
        result |= c.removeAllExtras(includeSubtree, types);
      }
    }
    return result;
  }

  /**
   * Clear all attributes of the topic.
   */
  public void clearAttributes() {
    this.attributes.clear();
  }

  /**
   * Remove all attributes from topic
   *
   * @param includeSubtree if true then remove all attributes from subtree
   * @param attributeNames names of attributes to be removed
   * @return true if any attribute was found and removed
   */
  public boolean removeAttributes(
      final boolean includeSubtree,
      final String... attributeNames
  ) {
    boolean result = false;

    for (final String name : attributeNames) {
      result |= this.attributes.remove(name) != null;
    }
    if (includeSubtree) {
      for (final Topic c : this.children) {
        result |= c.removeAttributes(includeSubtree, attributeNames);
      }
    }
    return result;
  }

  /**
   * Remove file link from the topic if presented
   *
   * @param baseFolder base mind map folder, can be null
   * @param fileUri    file URI to be removed, can't be null
   * @return true if file was detected and removed, false otherwise
   */
  public boolean deleteFileLinkIfPresented(final File baseFolder,
                                           final MMapURI fileUri) {
    boolean result = false;
    if (this.extras.containsKey(Extra.ExtraType.FILE)) {
      final ExtraFile fileLink = (ExtraFile) this.extras.get(Extra.ExtraType.FILE);
      if (fileLink.isSameOrHasParent(baseFolder, fileUri)) {
        result = this.extras.remove(Extra.ExtraType.FILE) != null;
      }
    }
    for (final Topic c : this.children) {
      result |= c.deleteFileLinkIfPresented(baseFolder, fileUri);
    }
    return result;
  }

  /**
   * Replace file link in the topic if found
   *
   * @param baseFolder base mind map folder, can be null
   * @param oldFileUri file uri to be replaced, must not be null
   * @param newFileUri new file uri, must not be null
   * @return true if file link found and replaced, false otherwise
   */
  public boolean replaceFileLinkIfPresented(final File baseFolder,
                                            final MMapURI oldFileUri,
                                            final MMapURI newFileUri) {
    boolean result = false;
    if (this.extras.containsKey(Extra.ExtraType.FILE)) {
      final ExtraFile fileLink = (ExtraFile) this.extras.get(Extra.ExtraType.FILE);
      final ExtraFile replacement;

      if (fileLink.isSame(baseFolder, oldFileUri)) {
        replacement = new ExtraFile(newFileUri);
      } else {
        replacement = fileLink.replaceParentPath(baseFolder, oldFileUri, newFileUri);
      }

      if (replacement != null) {
        result = true;
        this.extras.remove(Extra.ExtraType.FILE);
        this.extras.put(Extra.ExtraType.FILE, replacement);
      }
    }

    for (final Topic c : this.children) {
      result |= c.replaceFileLinkIfPresented(baseFolder, oldFileUri, newFileUri);
    }
    return result;
  }

  /**
   * Check that the topic contains file uri
   *
   * @param baseFolder     mind map base folder, can be null
   * @param fileUri        file uri to check, must not be null
   * @param includeSubtree check subtree also if true
   * @return true if file uri detected, false otherwise
   */
  public boolean doesContainFileLink(final File baseFolder, final MMapURI fileUri,
                                     final boolean includeSubtree) {
    if (this.extras.containsKey(Extra.ExtraType.FILE)) {
      final ExtraFile fileLink = (ExtraFile) this.extras.get(Extra.ExtraType.FILE);
      if (fileLink.isSame(baseFolder, fileUri)) {
        return true;
      }
    }
    if (includeSubtree) {
      for (final Topic c : this.children) {
        if (c.doesContainFileLink(baseFolder, fileUri, includeSubtree)) {
          return true;
        }
      }
    }
    return false;
  }

  @Override
  public Iterator iterator() {
    final Iterator childredIterator = this.children.iterator();

    return new Iterator() {
      Topic childTopic;
      Iterator childIterator;

      @Override
      public void remove() {
        childredIterator.remove();
      }

      Iterator init() {
        if (childredIterator.hasNext()) {
          this.childTopic = childredIterator.next();
        }
        return this;
      }

      @Override
      public boolean hasNext() {
        return childredIterator.hasNext() || this.childTopic != null ||
            (this.childIterator != null && this.childIterator.hasNext());
      }

      @Override
      public Topic next() {
        final Topic result;
        if (this.childTopic != null) {
          result = this.childTopic;
          this.childTopic = null;
          this.childIterator = result.iterator();
        } else if (this.childIterator != null) {
          if (this.childIterator.hasNext()) {
            result = this.childIterator.next();
          } else {
            result = childredIterator.next();
            this.childIterator = result.iterator();
          }
        } else {
          throw new NoSuchElementException();
        }
        return result;
      }
    }.init();
  }

  /**
   * Check that the topic contains any code snipet for language from array (case sensitive).
   *
   * @param languageNames names of language
   * @return true if code snippet is detected for any language, false otherwise
   */
  public boolean doesContainCodeSnippetForAnyLanguage(
      final String... languageNames) {
    boolean result = false;
    if (!this.codeSnippets.isEmpty()) {
      for (final String s : languageNames) {
        if (this.codeSnippets.containsKey(s)) {
          result = true;
          break;
        }
      }
    }
    return result;
  }

  /**
   * Get all subtree of the topic as stream
   *
   * @return stream of subtree for the topic, must not be null
   */
  public Stream stream() {
    return StreamSupport.stream(this.spliterator(), false);
  }

  /**
   * Get number of children in the topic
   * @return number of children
   */
  public int size() {
    return this.children.size();
  }
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy