package org.kuali.ole.describe.bo;
   
   import org.apache.commons.lang.StringEscapeUtils;
   import org.kuali.ole.docstore.model.bo.*;
   import org.kuali.ole.docstore.model.enums.DocType;
   
   import java.util.Map;
   
   /**
   * Created with IntelliJ IDEA.
   * User: Sreekanth
   * Date: 12/20/12
   * Time: 10:57 AM
   * To change this template use File | Settings | File Templates.
   */
  public class DocumentTreeNode {
  
      private String title;
      private String uuid;
      private boolean select;
      private boolean returnCheck;
  
      private OleDocument oleDocument;
      private String selectedInstance;
      private Map<String, String> selectionMap;
      private String holdingLocation;
  
      public DocumentTreeNode() {
  
      }
  
      public String getTitle() {
          return title;
      }
  
      public void setTitle(String title) {
          this.title = title;
      }
  
      public String getUuid() {
          return uuid;
      }
  
      public void setUuid(String uuid) {
          this.uuid = uuid;
      }
  
  
      public void setWorkBibDocument(WorkBibDocument workBibDocument) {
          this.oleDocument = workBibDocument;
          StringBuilder titleBuilder = new StringBuilder();
          if (workBibDocument != null) {
              if (workBibDocument.getTitle() != null) {
                  titleBuilder.append(workBibDocument.getTitle());
              }
              if (titleBuilder.length() > 0) {
                  titleBuilder.append(" / ");
              }
              if (workBibDocument.getAuthor() != null) {
                  titleBuilder.append(workBibDocument.getAuthor());
              }
  
              setTitle(StringEscapeUtils.escapeHtml(titleBuilder.toString()));
          } else {
              setTitle("Bibliographic Title");
          }
  
          setUuid(workBibDocument.getId());
  
      }
  
      public void setWorkInstanceDocument(WorkInstanceDocument workInstanceDocument) {
          this.oleDocument = workInstanceDocument;
          if (workInstanceDocument.getHoldingsDocument() != null) {
              setTitle(buildTreeDataForHoldings(workInstanceDocument));
          } else {
              setTitle("Holdings");
          }
          setUuid(workInstanceDocument.getInstanceIdentifier());
  
  
      }
  
      public void setWorkItemDocument(WorkItemDocument workItemDocument) {
          this.oleDocument = workItemDocument;
          if (workItemDocument != null) {
              setTitle(buildTreeDataForItem(workItemDocument));
          } else {
              setTitle("Item");
          }
  
          setUuid(workItemDocument.getItemIdentifier());
      }
  
      public void setWorkHoldingsDocument(WorkItemDocument workItemDocument, WorkHoldingsDocument workHoldingsDocument) {
          this.oleDocument = workItemDocument;
          if (workItemDocument != null) {
              setTitle(buildTreeDataForItem(workItemDocument, workHoldingsDocument));
          } else {
             setTitle("Item");
         }
 
         setUuid(workItemDocument.getItemIdentifier());
     }
 
     public void setWorkEInstanceDocument(WorkEInstanceDocument workEInstanceDocument) {
         this.oleDocument = workEInstanceDocument;
         if (workEInstanceDocument != null && workEInstanceDocument.getWorkEHoldingsDocument() != null) {
             StringBuffer stringBuffer = new StringBuffer();
             if (workEInstanceDocument.getWorkEHoldingsDocument() != null) {
                 if (workEInstanceDocument.getWorkEHoldingsDocument().getLocation() != null && workEInstanceDocument.getWorkEHoldingsDocument().getLocation().length() > 0) {
                     stringBuffer.append(workEInstanceDocument.getWorkEHoldingsDocument().getLocation());
                 }
                 if (stringBuffer.length() > 0 && workEInstanceDocument.getWorkEHoldingsDocument().geteResourceName() != null && workEInstanceDocument.getWorkEHoldingsDocument().geteResourceName().length() > 0) {
                     stringBuffer.append("-");
                 }
                 if (workEInstanceDocument.getWorkEHoldingsDocument().geteResourceName() != null) {
                     stringBuffer.append(workEInstanceDocument.getWorkEHoldingsDocument().geteResourceName());
                 }
 
             }
             setTitle(stringBuffer.toString());
         } else {
             setTitle("eHoldings");
         }
 
         setUuid(workEInstanceDocument.getInstanceIdentifier() + " " + DocType.EINSTANCE.getCode());
     }
 
     public OleDocument getWorkBibDocument() {
         getTitle();
         getUuid();
         return oleDocument = new WorkBibDocument();
     }
 
     public OleDocument getWorkInstanceDocument() {
 /*        getTitle();
         getUuid();*/
         return oleDocument = new WorkInstanceDocument();
     }
 
     public OleDocument getWorkItemDocument() {
         getTitle();
         getUuid();
         return oleDocument = new WorkItemDocument();
     }
 
 
     public OleDocument getWorkEInstanceDocument() {
         getTitle();
         getUuid();
         return oleDocument = new WorkEInstanceDocument();
     }
 
 
     public String getSelectedInstance() {
         return selectedInstance;
     }
 
     public void setSelectedInstance(String selectedInstance) {
         this.selectedInstance = selectedInstance;
     }
 
 
     public boolean isSelect() {
         return select;
     }
 
     public void setSelect(boolean select) {
         this.select = select;
     }
 
     public Map<String, String> getSelectionMap() {
         return selectionMap;
     }
 
     public void setSelectionMap(Map<String, String> selectionMap) {
         this.selectionMap = selectionMap;
     }
 
     public boolean isReturnCheck() {
         return returnCheck;
     }
 
     public void setReturnCheck(boolean returnCheck) {
         this.returnCheck = returnCheck;
     }
 
 
     public String buildTreeDataForHoldings(WorkInstanceDocument workInstanceDocument) {
 
         StringBuffer holdingsLabelBufferForStaffOnly = new StringBuffer();
         WorkHoldingsDocument workHoldingsDocument = workInstanceDocument.getHoldingsDocument();
         return new EditorFormDataHandler().getHoldingsLabel(workHoldingsDocument);
     }
 
     public String buildTreeDataForItem(WorkItemDocument workItemDocument, WorkHoldingsDocument workHoldingsDocument) {
 
         return new EditorFormDataHandler().getItemLabel(workHoldingsDocument, workItemDocument);
 
     }
 
     public String buildTreeDataForItem(WorkItemDocument workItemDocument) {
         String itemLevelContent = null;
         StringBuilder treeBuilder = new StringBuilder();
         String locationName = this.getHoldingLocation();
 //        if (workItemDocument.getLocation() != null) {
 //            treeBuilder.append(workItemDocument.getLocation());
 //        }
         StringBuffer removeFromCallNumber = new StringBuffer();
         String callNumberPrefix = null;
         String itemLocation = null;
         if (workItemDocument.getLocation() != null) {
             itemLocation = workItemDocument.getLocation();
             if (!(itemLocation).equalsIgnoreCase(locationName)) {
 
                 treeBuilder.append(itemLocation);
             }
         }
         if (workItemDocument.getCallNumberPrefix() != null && workItemDocument.getCallNumberPrefix().length() > 0) {
             callNumberPrefix = workItemDocument.getCallNumberPrefix();
         }
         if (callNumberPrefix != null) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-").append(callNumberPrefix);
             } else {
                 treeBuilder.append(callNumberPrefix);
             }
         }
         if (workItemDocument.getCallNumber() != null && workItemDocument.getCallNumber().length() > 0) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-");
             }
             treeBuilder.append(workItemDocument.getCallNumber());
         }
         if (workItemDocument.getEnumeration() != null && workItemDocument.getEnumeration().length() > 0) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-");
             }
             treeBuilder.append(workItemDocument.getEnumeration());
             removeFromCallNumber.append(" " + workItemDocument.getEnumeration());
         }
         if (workItemDocument.getChronology() != null && workItemDocument.getChronology().length() > 0) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-");
             }
             treeBuilder.append(workItemDocument.getChronology());
             removeFromCallNumber.append(" " + workItemDocument.getChronology());
 
         }
         if (workItemDocument.getCopyNumber() != null && workItemDocument.getCopyNumber().length() > 0) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-");
             }
             treeBuilder.append(workItemDocument.getCopyNumber());
             removeFromCallNumber.append(" " + workItemDocument.getCopyNumber());
         }
         if (workItemDocument.getBarcode() != null && workItemDocument.getBarcode().length() > 0) {
             if (treeBuilder.length() > 0) {
                 treeBuilder.append("-");
             }
             treeBuilder.append(workItemDocument.getBarcode());
         }
         if (treeBuilder.length() == 0) {
             if (workItemDocument.getVolumeNumber() != null) {
                 treeBuilder.append(workItemDocument.getVolumeNumber());
             } else {
                 treeBuilder.append("Item");
             }
         }
         itemLevelContent = treeBuilder.toString().replaceAll(removeFromCallNumber.toString(), "");
 
         return itemLevelContent;
 
     }
 
     public String getHoldingLocation() {
         return holdingLocation;
     }
 
     public void setHoldingLocation(String holdingLocation) {
         this.holdingLocation = holdingLocation;
     }
 }