Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.bookmarksandhistory.impl;

import org.gvsig.tools.bookmarksandhistory.BookmarksAndHistoryManager;

import java.io.IOException;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.bookmarksandhistory.Bookmark;

import org.gvsig.tools.bookmarksandhistory.Bookmarks;

import org.gvsig.tools.bookmarksandhistory.History;

import org.gvsig.tools.identitymanagement.SimpleIdentityManager;

import org.gvsig.tools.persistence.PersistenceManager;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.resourcesstorage.ResourcesStorage;

/**

 *

 * @author jjdelcerro

 */

public class DefaultBookmarksAndHistoryManager implements BookmarksAndHistoryManager {

  private final Map<String,Bookmarks<Object>> bookmarkGroups;

  private final Map<String,History<Object>> historyGroups;

  private ResourcesStorage primaryStorage;

  private ResourcesStorage secondaryStorage;

  public DefaultBookmarksAndHistoryManager() {

    this.bookmarkGroups = new HashMap<>();

    this.historyGroups = new HashMap<>();

  }

  @Override

  public ResourcesStorage getPrimaryStorage() {

    return primaryStorage;

  }

  @Override

  public ResourcesStorage getSecondaryStorage() {

    return secondaryStorage;

  }

  @Override

  public void setPrimaryStorage(ResourcesStorage primaryStorage) {

    this.primaryStorage = primaryStorage;

  }

  @Override

  public void setSecondaryStorage(ResourcesStorage secondaryStorage) {

    this.secondaryStorage = secondaryStorage;

  }

  @Override

  public void registerBookmarksGroup(String name, Bookmarks<Object> bookmarks) {

    this.bookmarkGroups.put(name, bookmarks);

  }

  @Override

  public void registerBookmarksGroup(String name) {

    this.bookmarkGroups.put(name, new BaseBookmarks<>());

  }

  @Override

  public Bookmarks<Object> getBookmarksGroup(String name) {

    return this.bookmarkGroups.get(name);

  }

  @Override

  public void registerHistoryGroup(String name, History<Object> history) {

    this.historyGroups.put(name, history);

  }

  @Override

  public void registerHistoryGroup(String name, int size) {

    this.historyGroups.put(name, new BaseHistory<>(size));

  }

  @Override

  public History<Object> getHistoryGroup(String name) {

    return this.historyGroups.get(name);

  }

  @Override

  public void save() {

    if( this.primaryStorage!=null ) {

      saveBookmarks(primaryStorage);

      saveHitory(primaryStorage);

    }

    if( this.secondaryStorage!=null ) {

      saveBookmarks(secondaryStorage);

      saveHitory(secondaryStorage);

    }

  }

  private void saveBookmarks(ResourcesStorage storage) {

    PersistenceManager persistenceManager = ToolsLocator.getPersistenceManager();

    SimpleIdentityManager userManager = ToolsLocator.getIdentityManager();

    String userName = userManager.getCurrentIdentity().getID();

    for (Map.Entry<String, Bookmarks<Object>> bookmarksEntry : this.bookmarkGroups.entrySet()) {

      String bookmarksGroupName = bookmarksEntry.getKey();

      Bookmarks<Object> bookmarks = bookmarksEntry.getValue();

      for (Bookmark<Object> bookmark : bookmarks ) {

        String resourceName;

        if( bookmark.isGlobal() ) {

          resourceName = userName + "@BOOKMARKS!" + bookmarksGroupName+"!"+bookmark.getName();

        } else {

          resourceName = "ALL@BOOKMARKS!" + bookmarksGroupName+"!"+bookmark.getName();

        }

        try (ResourcesStorage.Resource resource = storage.getResource(resourceName)) {

          Object data = bookmark.getValue();

          PersistentState state = persistenceManager.getState(data);

          persistenceManager.saveState(state, resource.asOutputStream());

        } catch (Exception ex) {

          // ???

        }

      }

    }

  }

  @Override

  public void load() {

    if( this.primaryStorage!=null ) {

      loadBookmarks(primaryStorage);

      loadHistory(primaryStorage);

    }

    if( this.secondaryStorage!=null ) {

      loadBookmarks(secondaryStorage);

      loadHistory(secondaryStorage);

    }

  }

  private void loadBookmarks(ResourcesStorage storage) {

    PersistenceManager persistenceManager = ToolsLocator.getPersistenceManager();

    SimpleIdentityManager userManager = ToolsLocator.getIdentityManager();

    String userName = userManager.getCurrentIdentity().getID();

    List<String> names = storage.getResourceNames();

    for (String name : names) {

      String[] userAndBookmark = StringUtils.split(name, "@");

      String currentUserName = userAndBookmark[0];

      String[] groupAndId = StringUtils.split(userAndBookmark[1], "!");

      if( groupAndId.length!=3 ) {

        continue;

      }

      String typeName = groupAndId[0];

      if( !StringUtils.equalsIgnoreCase(typeName, "BOOKMARKS") ) {

        continue;

      }

      String bookmarksGroupName = groupAndId[1];

      String bookmarkName = groupAndId[2];

      if( StringUtils.equalsIgnoreCase(currentUserName, userName) ) {

        try (ResourcesStorage.Resource resource = storage.getResource(name)) {

          Object data = persistenceManager.getObject(resource.asInputStream());

          Bookmarks<Object> bookmarks = this.bookmarkGroups.get(bookmarksGroupName);

          bookmarks.add(bookmarkName, data);

        } catch (IOException ex) {

        }

      } else if( StringUtils.equalsIgnoreCase(currentUserName, "ALL") ) {

        try (ResourcesStorage.Resource resource = storage.getResource(name)) {

          Object data = persistenceManager.getObject(resource.asInputStream());

          Bookmarks<Object> bookmarks = this.bookmarkGroups.get(bookmarksGroupName);

          Bookmark bookmark = bookmarks.add(bookmarkName, data);

          bookmark.setGlobal(true);

        } catch (IOException ex) {

        }

      }

    }

  }

  private void saveHitory(ResourcesStorage storage) {

    PersistenceManager persistenceManager = ToolsLocator.getPersistenceManager();

    SimpleIdentityManager userManager = ToolsLocator.getIdentityManager();

    String userName = userManager.getCurrentIdentity().getID();

    for (Map.Entry<String, History<Object>> historyEntry : this.historyGroups.entrySet()) {

      String historyGroupName = historyEntry.getKey();

      History<Object> history = historyEntry.getValue();

      for (int i = 0; i < history.size(); i++) {

        String resourceName = userName + "@HISTORY!" + historyGroupName+"!"+i;

        try (ResourcesStorage.Resource resource = storage.getResource(resourceName)) {

          Object data = history.get(i);

          PersistentState state = persistenceManager.getState(data);

          persistenceManager.saveState(state, resource.asOutputStream());

        } catch (Exception ex) {

          // ???

        }

      }

    }

  }

  private void loadHistory(ResourcesStorage storage) {

    PersistenceManager persistenceManager = ToolsLocator.getPersistenceManager();

    SimpleIdentityManager userManager = ToolsLocator.getIdentityManager();

    String userName = userManager.getCurrentIdentity().getID();

    List<String> names = storage.getResourceNames();

    for (String name : names) {

      String[] userAndHistory = StringUtils.split(name, "@");

      String currentUserName = userAndHistory[0];

      String[] groupAndId = StringUtils.split(userAndHistory[1], "!");

      if( groupAndId.length!=3 ) {

        continue;

      }

      String typeName = groupAndId[0];

      if( !StringUtils.equalsIgnoreCase(typeName, "HISTORY") ) {

        continue;

      }

      String groupName = groupAndId[1];

      if( StringUtils.equalsIgnoreCase(currentUserName, userName) ) {

        try (ResourcesStorage.Resource resource = storage.getResource(name)) {

          Object data = persistenceManager.getObject(resource.asInputStream());

          History<Object> history = this.historyGroups.get(groupName);

          history.add(data);

        } catch (IOException ex) {

        }

      }

    }

  }

}