plugins/org.eclipse.osee.framework.manager.servlet/src/org/eclipse/osee/framework/manager/servlet/DataServlet.java

/*******************************************************************************
 * Copyright (c) 2010 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.framework.manager.servlet;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.util.ArrayList;
import java.util.List;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.eclipse.osee.framework.core.exception.OseeNotFoundException;
import org.eclipse.osee.framework.core.server.UnsecuredOseeHttpServlet;
import org.eclipse.osee.framework.database.IOseeDatabaseService;
import org.eclipse.osee.framework.jdk.core.type.OseeCoreException;
import org.eclipse.osee.framework.jdk.core.util.Conditions;
import org.eclipse.osee.framework.jdk.core.util.Lib;
import org.eclipse.osee.framework.jdk.core.util.Strings;
import org.eclipse.osee.framework.manager.servlet.data.ArtifactUtil;
import org.eclipse.osee.framework.resource.management.IResourceManager;
import org.eclipse.osee.logger.Log;
import org.eclipse.osee.orcs.OrcsApi;
import org.eclipse.osee.orcs.data.BranchReadable;
import org.eclipse.osee.orcs.search.BranchQuery;

public class DataServlet extends UnsecuredOseeHttpServlet {

   private static final long serialVersionUID = -1399699606153734250L;

   private final IResourceManager resourceManager;
   private final OrcsApi orcsApi;
   private final IOseeDatabaseService dbService;

   public DataServlet(Log logger, IResourceManager resourceManager, OrcsApi orcsApi, IOseeDatabaseService dbService) {
      super(logger);
      this.resourceManager = resourceManager;
      this.orcsApi = orcsApi;
      this.dbService = dbService;
   }

   @Override
   protected void doGet(HttpServletRequest request, HttpServletResponse response) throws IOException {
      String urlRequest = request.getRequestURI();
      try {
         handleUriRequest(dbService, resourceManager, urlRequest, response, orcsApi);
      } catch (OseeCoreException ex) {
         handleError(response, HttpURLConnection.HTTP_INTERNAL_ERROR, "", ex);
      }
   }

   private void handleError(HttpServletResponse response, int status, String message, Throwable ex) throws IOException {
      response.setContentType("text/plain");
      getLogger().error(ex, message);
      response.sendError(status, Lib.exceptionToString(ex));
   }

   public static void handleUriRequest(IOseeDatabaseService dbService, IResourceManager resourceManager, String urlRequest, HttpServletResponse response, OrcsApi orcsApi) throws OseeCoreException {
      UrlParser parser = new UrlParser();
      parser.parse(urlRequest);
      Long branchUuid = Long.valueOf(parser.getAttribute("branch"));
      String artifactGuid = parser.getAttribute("artifact");
      BranchQuery query = orcsApi.getQueryFactory(null).branchQuery();
      BranchReadable branch = query.andUuids(branchUuid).getResults().getExactlyOne();

      String uri = ArtifactUtil.getUri(dbService, artifactGuid, branch);
      ArtifactFileServlet.handleArtifactUri(resourceManager, urlRequest, uri, response);
   }

   private static final class UrlParser {
      private final List<String> contexts;

      public UrlParser() {
         this.contexts = new ArrayList<String>();
      }

      public void parse(String urlPath) {
         contexts.clear();
         if (Strings.isValid(urlPath)) {
            String[] items = urlPath.split("/");
            for (String item : items) {
               contexts.add(item);
            }
         }
      }

      public String getAttribute(String key) throws OseeCoreException {
         Conditions.checkNotNull(key, "attribute");
         int contextCount = contexts.size();
         for (int index = 0; index < contextCount; index++) {
            String context = contexts.get(index);
            if (context.equals(key)) {
               if (index + 1 < contextCount) {
                  return contexts.get(index + 1);
               }
            }
         }
         throw new OseeNotFoundException("Unable to find [%s]", key);
      }
   }
}