/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you 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 org.apache.chemistry.opencmis.server.impl.atompub;
  
  import org.apache.chemistry.opencmis.commons.data.ObjectData;
  import org.apache.chemistry.opencmis.commons.exceptions.CmisRuntimeException;
  import org.apache.chemistry.opencmis.commons.impl.Constants;
  import static org.apache.chemistry.opencmis.commons.impl.Converter.convert;
  import org.apache.chemistry.opencmis.commons.impl.UrlBuilder;
  import org.apache.chemistry.opencmis.commons.impl.jaxb.CmisObjectType;
  import org.apache.chemistry.opencmis.commons.server.CallContext;
  import org.apache.chemistry.opencmis.commons.server.CmisService;
  import org.apache.chemistry.opencmis.commons.server.ObjectInfo;
  import static org.apache.chemistry.opencmis.server.impl.atompub.AtomPubUtils.RESOURCE_POLICIES;
  import static org.apache.chemistry.opencmis.server.impl.atompub.AtomPubUtils.compileBaseUrl;
  import static org.apache.chemistry.opencmis.server.impl.atompub.AtomPubUtils.compileUrl;
  import static org.apache.chemistry.opencmis.server.impl.atompub.AtomPubUtils.compileUrlBuilder;
  import static org.apache.chemistry.opencmis.server.shared.HttpUtils.getStringParameter;
  
  import java.util.List;
  
  import javax.servlet.http.HttpServletRequest;
  import javax.servlet.http.HttpServletResponse;
  
  
  /**
   * Policy Service operations.
   */
  public class PolicyService
  {
      private PolicyService(  )
      {
      }
  
      /**
       * Get applied policies.
       */
      public static void getAppliedPolicies( CallContext context, CmisService service, String repositoryId,
          HttpServletRequest request, HttpServletResponse response )
          throws Exception
      {
          // get parameters
          String objectId = getStringParameter( request, Constants.PARAM_ID );
          String filter = getStringParameter( request, Constants.PARAM_FILTER );
  
          // execute
          List<ObjectData> policies = service.getAppliedPolicies( repositoryId, objectId, filter, null );
  
          if ( policies == null )
          {
              throw new CmisRuntimeException( "Policies are null!" );
          }
  
          ObjectInfo objectInfo = service.getObjectInfo( repositoryId, objectId );
  
          if ( objectInfo == null )
          {
              throw new CmisRuntimeException( "Object Info is missing!" );
          }
  
          // set headers
          response.setStatus( HttpServletResponse.SC_OK );
          response.setContentType( Constants.MEDIATYPE_FEED );
  
          // write XML
          AtomFeed feed = new AtomFeed(  );
          feed.startDocument( response.getOutputStream(  ) );
          feed.startFeed( true );
  
          // write basic Atom feed elements
          feed.writeFeedElements( objectInfo.getId(  ), objectInfo.getCreatedBy(  ), objectInfo.getName(  ),
              objectInfo.getLastModificationDate(  ), null, null );
  
          // write links
          UrlBuilder baseUrl = compileBaseUrl( request, repositoryId );
  
          feed.writeServiceLink( baseUrl.toString(  ), repositoryId );
  
          feed.writeSelfLink( compileUrl( baseUrl, RESOURCE_POLICIES, objectInfo.getId(  ) ), null );
  
          // write entries
          AtomEntry entry = new AtomEntry( feed.getWriter(  ) );
  
         for ( ObjectData policy : policies )
         {
             if ( policy == null )
             {
                 continue;
             }
 
             writePolicyEntry( service, entry, objectInfo.getId(  ), policy, repositoryId, baseUrl );
         }
 
         // we are done
         feed.endFeed(  );
         feed.endDocument(  );
     }
 
     /**
      * Apply policy.
      */
     public static void applyPolicy( CallContext context, CmisService service, String repositoryId,
         HttpServletRequest request, HttpServletResponse response )
         throws Exception
     {
         // get parameters
         String objectId = getStringParameter( request, Constants.PARAM_ID );
 
         AtomEntryParser parser = new AtomEntryParser( request.getInputStream(  ), context.getTempDirectory(  ),
                 context.getMemoryThreshold(  ) );
 
         // execute
         service.applyPolicy( repositoryId, parser.getId(  ), objectId, null );
 
         ObjectInfo objectInfo = service.getObjectInfo( repositoryId, parser.getId(  ) );
 
         if ( objectInfo == null )
         {
             throw new CmisRuntimeException( "Object Info is missing!" );
         }
 
         ObjectData policy = objectInfo.getObject(  );
 
         if ( policy == null )
         {
             throw new CmisRuntimeException( "Policy is null!" );
         }
 
         // set headers
         UrlBuilder baseUrl = compileBaseUrl( request, repositoryId );
         UrlBuilder location = compileUrlBuilder( baseUrl, RESOURCE_POLICIES, objectId );
         location.addParameter( Constants.PARAM_POLICY_ID, policy.getId(  ) );
 
         response.setStatus( HttpServletResponse.SC_CREATED );
         response.setContentType( Constants.MEDIATYPE_ENTRY );
         response.setHeader( "Content-Location", location.toString(  ) );
         response.setHeader( "Location", location.toString(  ) );
 
         // write XML
         AtomEntry entry = new AtomEntry(  );
         entry.startDocument( response.getOutputStream(  ) );
         writePolicyEntry( service, entry, objectId, policy, repositoryId, baseUrl );
         entry.endDocument(  );
     }
 
     /**
      * Remove policy.
      */
     public static void removePolicy( CallContext context, CmisService service, String repositoryId,
         HttpServletRequest request, HttpServletResponse response )
     {
         // get parameters
         String objectId = getStringParameter( request, Constants.PARAM_ID );
         String policyId = getStringParameter( request, Constants.PARAM_POLICY_ID );
 
         // execute
         service.removePolicy( repositoryId, policyId, objectId, null );
 
         // set headers
         response.setStatus( HttpServletResponse.SC_NO_CONTENT );
     }
 
     /**
      * Writes an entry that is attached to an object.
      */
     private static void writePolicyEntry( CmisService service, AtomEntry entry, String objectId, ObjectData policy,
         String repositoryId, UrlBuilder baseUrl ) throws Exception
     {
         CmisObjectType resultJaxb = convert( policy );
 
         if ( resultJaxb == null )
         {
             return;
         }
 
         ObjectInfo info = service.getObjectInfo( repositoryId, policy.getId(  ) );
 
         if ( info == null )
         {
             throw new CmisRuntimeException( "Object Info not found!" );
         }
 
         // start
         entry.startEntry( false );
 
         // write the object
         entry.writeObject( policy, info, null, null, null, null );
 
         // write links
         UrlBuilder selfLink = compileUrlBuilder( baseUrl, RESOURCE_POLICIES, objectId );
         selfLink.addParameter( Constants.PARAM_POLICY_ID, info.getId(  ) );
         entry.writeSelfLink( selfLink.toString(  ), null );
 
         // we are done
         entry.endEntry(  );
     }
 }