path: root/plugins/org.eclipse.osee.define.report.api/src/org/eclipse/osee/define/report/api/DataRightResult.java

/*******************************************************************************
 * Copyright (c) 2014 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.define.report.api;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlTransient;
import org.eclipse.osee.framework.jdk.core.util.Strings;

/**
 * @author Angel Avila
 */
@XmlRootElement
public class DataRightResult {

   @XmlTransient
   private List<DataRightAnchor> dataRightAnchors;

   @XmlTransient
   private List<DataRight> dataRights;

   private Map<DataRightId, DataRight> dataRightIdToDataRight;

   private Map<String, DataRightAnchor> guidToAnchor;

   @XmlElement
   public Collection<DataRightAnchor> getDataRightAnchors() {
      if (dataRightAnchors == null) {
         dataRightAnchors = new ArrayList<DataRightAnchor>();
      }
      return dataRightAnchors;
   }

   public void setDataRightAnchors(List<DataRightAnchor> dataRightAnchors) {
      this.dataRightAnchors = dataRightAnchors;
   }

   @XmlElement
   public Collection<DataRight> getDataRights() {
      if (dataRights == null) {
         dataRights = new ArrayList<DataRight>();
      }
      return dataRights;
   }

   public void setDataRights(List<DataRight> dataRights) {
      this.dataRights = dataRights;
   }

   public String getContent(String guid, PageOrientation orientation) {

      checkInitialized();
      String footer = null;

      // account for orientation
      String portrait = String.format(ReportConstants.PAGE_ADDS, ReportConstants.PORTRAIT_ORIENT);
      String landscape = String.format(ReportConstants.PAGE_ADDS, ReportConstants.LANDSCAPE_ORIENT);
      String page_adds = orientation.isLandscape() ? landscape : portrait;

      DataRightAnchor anchor = guidToAnchor.get(guid);
      if (anchor != null) {
         boolean isSetDataRightFooter = anchor.isSetDataRightFooter();
         boolean isContinuous = anchor.isContinuous();
         if (isSetDataRightFooter) {
            // set footer section since next footer differs
            DataRightId key = anchor.getDataRightId();
            if (key != null) {
               DataRight dataRight = dataRightIdToDataRight.get(key);
               if (dataRight != null) {
                  footer = normalize(dataRight.getContent());
                  footer = String.format(ReportConstants.NEW_PAGE_TEMPLATE, footer + page_adds);
               }
            }
         } else if (!isContinuous) {
            // set page break since next footer differs;
            footer = String.format(ReportConstants.NEW_PAGE_TEMPLATE, page_adds);
         }
      }
      return Strings.isValid(footer) ? footer : "";
   }

   private String normalize(String partialFooter) {
      String toReturn = partialFooter;
      if ("NO DATA RIGHTS ARTIFACT FOUND".equals(toReturn)) {
         toReturn = String.format("<w:p><w:r><w:t>%s</w:t></w:r></w:p>", toReturn);
      }
      return toReturn;
   }

   public void reset() {
      dataRightIdToDataRight = null;
      guidToAnchor = null;
   }

   private void checkInitialized() {
      if (dataRightIdToDataRight == null || guidToAnchor == null) {
         dataRightIdToDataRight = new HashMap<DataRightId, DataRight>();
         guidToAnchor = new HashMap<String, DataRightAnchor>();
         for (DataRightAnchor anchor : dataRightAnchors) {
            guidToAnchor.put(anchor.getId(), anchor);
         }

         for (DataRight dataRight : dataRights) {
            dataRightIdToDataRight.put(dataRight.getId(), dataRight);
         }
      }
   }
}