-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #53 from at88mph/add-tar-zip
Add tar zip
- Loading branch information
Showing
10 changed files
with
345 additions
and
21 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
130 changes: 130 additions & 0 deletions
130
cadc-download-manager-server/src/main/java/ca/nrc/cadc/dlm/server/PackageServlet.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,130 @@ | ||
package ca.nrc.cadc.dlm.server; | ||
|
||
import ca.nrc.cadc.auth.AuthMethod; | ||
import ca.nrc.cadc.auth.AuthenticationUtil; | ||
import ca.nrc.cadc.config.ApplicationConfiguration; | ||
import ca.nrc.cadc.dlm.DownloadRequest; | ||
import ca.nrc.cadc.net.HttpPost; | ||
import ca.nrc.cadc.reg.Standards; | ||
import ca.nrc.cadc.reg.client.RegistryClient; | ||
import ca.nrc.cadc.util.StringUtil; | ||
import java.io.IOException; | ||
import java.net.URI; | ||
import java.net.URL; | ||
import java.security.PrivilegedAction; | ||
import java.util.HashMap; | ||
import java.util.List; | ||
import java.util.Map; | ||
import java.util.stream.Collectors; | ||
import javax.security.auth.Subject; | ||
import javax.servlet.http.HttpServlet; | ||
import javax.servlet.http.HttpServletRequest; | ||
import javax.servlet.http.HttpServletResponse; | ||
|
||
|
||
public abstract class PackageServlet extends HttpServlet { | ||
private static final String PACKAGE_SERVICE_RESOURCE_ID_KEY = "org.opencadc.dlm.package-download.service.id"; | ||
private static final String RESPONSE_FORMAT_PAYLOAD_KEY = "RESPONSEFORMAT"; | ||
private static final String ID_PAYLOAD_KEY = "ID"; | ||
private static final String RUN_ID_PAYLOAD_KEY = "runid"; | ||
|
||
final ApplicationConfiguration configuration = new ApplicationConfiguration(DispatcherServlet.DEFAULT_CONFIG_FILE_PATH); | ||
|
||
|
||
/** | ||
* Handle POSTed download request from an external page. | ||
* | ||
* @param request The HTTP Request | ||
* @param response The HTTP Response | ||
*/ | ||
@Override | ||
protected void doPost(HttpServletRequest request, HttpServletResponse response) { | ||
final URI packageServiceURI = this.configuration.lookupServiceURI(PackageServlet.PACKAGE_SERVICE_RESOURCE_ID_KEY, null); | ||
if (packageServiceURI == null) { | ||
throw new IllegalStateException("No service configured to perform Package Downloads"); | ||
} else { | ||
doRedirect(request, response, packageServiceURI); | ||
} | ||
} | ||
|
||
/** | ||
* Handle creating a payload and obtaining a redirect URL from the main service. | ||
* @param request The HTTP Request. | ||
* @param response The HTTP Response. | ||
* @param packageServiceURI The URI of the configured | ||
*/ | ||
void doRedirect(final HttpServletRequest request, final HttpServletResponse response, final URI packageServiceURI) { | ||
final Subject currentSubject = AuthenticationUtil.getCurrentSubject(); | ||
final URL packageServiceURL = lookupServiceURL(packageServiceURI, currentSubject); | ||
|
||
Subject.doAs(currentSubject, (PrivilegedAction<Void>) () -> { | ||
final HttpPost httpPost = new HttpPost(packageServiceURL, getPayload(request), false); | ||
|
||
try { | ||
httpPost.prepare(); | ||
} catch (Exception exception) { | ||
throw new RuntimeException(exception.getMessage(), exception); | ||
} | ||
|
||
final URL redirect = httpPost.getRedirectURL(); | ||
if (redirect == null) { | ||
throw new IllegalArgumentException("Unable to complete download (no redirect)."); | ||
} else { | ||
try { | ||
response.sendRedirect(redirect.toExternalForm()); | ||
} catch (IOException ioException) { | ||
throw new IllegalStateException("Redirect is invalid: " + redirect); | ||
} | ||
} | ||
|
||
return null; | ||
}); | ||
} | ||
|
||
private Map<String, Object> getPayload(final HttpServletRequest request) { | ||
final DownloadRequest downloadRequest = getDownloadRequest(request); | ||
|
||
final Map<String, Object> payload = new HashMap<>(); | ||
|
||
final List<String> publisherIDs = | ||
downloadRequest.getTuples().stream().map(downloadTuple -> downloadTuple.getID().toString()).collect(Collectors.toList()); | ||
|
||
if (publisherIDs.isEmpty()) { | ||
throw new IllegalArgumentException("Nothing specified to download. Use tuple=<URI>."); | ||
} | ||
|
||
payload.put(PackageServlet.ID_PAYLOAD_KEY, publisherIDs); | ||
|
||
final String contentType = getContentType(); | ||
if (!StringUtil.hasText(contentType)) { | ||
throw new IllegalStateException("Poorly configured package type (no content type specified)"); | ||
} | ||
|
||
payload.put(PackageServlet.RESPONSE_FORMAT_PAYLOAD_KEY, contentType); | ||
|
||
if (StringUtil.hasText(downloadRequest.runID)) { | ||
payload.put(PackageServlet.RUN_ID_PAYLOAD_KEY, downloadRequest.runID); | ||
} | ||
|
||
return payload; | ||
} | ||
|
||
private URL lookupServiceURL(URI packageServiceURI, Subject currentSubject) { | ||
final RegistryClient registryClient = new RegistryClient(); | ||
final AuthMethod authMethod = currentSubject == null ? AuthMethod.ANON : AuthenticationUtil.getAuthMethod(currentSubject); | ||
return registryClient.getServiceURL(packageServiceURI, Standards.PKG_10, authMethod); | ||
} | ||
|
||
private DownloadRequest getDownloadRequest(final HttpServletRequest request) { | ||
final DownloadRequest downloadReq = (DownloadRequest) request.getAttribute("downloadRequest"); | ||
downloadReq.runID = (String) request.getAttribute("runid"); | ||
|
||
return downloadReq; | ||
} | ||
|
||
/** | ||
* Pull the specific content type to set in the payload. | ||
* @return String content-type, never null. | ||
*/ | ||
abstract String getContentType(); | ||
} |
80 changes: 80 additions & 0 deletions
80
cadc-download-manager-server/src/main/java/ca/nrc/cadc/dlm/server/TARPackageServlet.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
/* | ||
************************************************************************ | ||
******************* CANADIAN ASTRONOMY DATA CENTRE ******************* | ||
************** CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES ************** | ||
* | ||
* (c) 2024. (c) 2024. | ||
* Government of Canada Gouvernement du Canada | ||
* National Research Council Conseil national de recherches | ||
* Ottawa, Canada, K1A 0R6 Ottawa, Canada, K1A 0R6 | ||
* All rights reserved Tous droits réservés | ||
* | ||
* NRC disclaims any warranties, Le CNRC dénie toute garantie | ||
* expressed, implied, or énoncée, implicite ou légale, | ||
* statutory, of any kind with de quelque nature que ce | ||
* respect to the software, soit, concernant le logiciel, | ||
* including without limitation y compris sans restriction | ||
* any warranty of merchantability toute garantie de valeur | ||
* or fitness for a particular marchande ou de pertinence | ||
* purpose. NRC shall not be pour un usage particulier. | ||
* liable in any event for any Le CNRC ne pourra en aucun cas | ||
* damages, whether direct or être tenu responsable de tout | ||
* indirect, special or general, dommage, direct ou indirect, | ||
* consequential or incidental, particulier ou général, | ||
* arising from the use of the accessoire ou fortuit, résultant | ||
* software. Neither the name de l'utilisation du logiciel. Ni | ||
* of the National Research le nom du Conseil National de | ||
* Council of Canada nor the Recherches du Canada ni les noms | ||
* names of its contributors may de ses participants ne peuvent | ||
* be used to endorse or promote être utilisés pour approuver ou | ||
* products derived from this promouvoir les produits dérivés | ||
* software without specific prior de ce logiciel sans autorisation | ||
* written permission. préalable et particulière | ||
* par écrit. | ||
* | ||
* This file is part of the Ce fichier fait partie du projet | ||
* OpenCADC project. OpenCADC. | ||
* | ||
* OpenCADC is free software: OpenCADC est un logiciel libre ; | ||
* you can redistribute it and/or vous pouvez le redistribuer ou le | ||
* modify it under the terms of modifier suivant les termes de | ||
* the GNU Affero General Public la “GNU Affero General Public | ||
* License as published by the License” telle que publiée | ||
* Free Software Foundation, par la Free Software Foundation | ||
* either version 3 of the : soit la version 3 de cette | ||
* License, or (at your option) licence, soit (à votre gré) | ||
* any later version. toute version ultérieure. | ||
* | ||
* OpenCADC is distributed in the OpenCADC est distribué | ||
* hope that it will be useful, dans l’espoir qu’il vous | ||
* but WITHOUT ANY WARRANTY; sera utile, mais SANS AUCUNE | ||
* without even the implied GARANTIE : sans même la garantie | ||
* warranty of MERCHANTABILITY implicite de COMMERCIALISABILITÉ | ||
* or FITNESS FOR A PARTICULAR ni d’ADÉQUATION À UN OBJECTIF | ||
* PURPOSE. See the GNU Affero PARTICULIER. Consultez la Licence | ||
* General Public License for Générale Publique GNU Affero | ||
* more details. pour plus de détails. | ||
* | ||
* You should have received Vous devriez avoir reçu une | ||
* a copy of the GNU Affero copie de la Licence Générale | ||
* General Public License along Publique GNU Affero avec | ||
* with OpenCADC. If not, see OpenCADC ; si ce n’est | ||
* <http://www.gnu.org/licenses/>. pas le cas, consultez : | ||
* <http://www.gnu.org/licenses/>. | ||
* | ||
* | ||
************************************************************************ | ||
*/ | ||
|
||
package ca.nrc.cadc.dlm.server; | ||
|
||
public class TARPackageServlet extends PackageServlet { | ||
public static final String TAR_PACKAGE_TARGET = "/tar-package"; | ||
private static final String TAR_CONTENT_TYPE = "application/x-tar"; | ||
|
||
|
||
@Override | ||
String getContentType() { | ||
return TARPackageServlet.TAR_CONTENT_TYPE; | ||
} | ||
} |
80 changes: 80 additions & 0 deletions
80
cadc-download-manager-server/src/main/java/ca/nrc/cadc/dlm/server/ZIPPackageServlet.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,80 @@ | ||
/* | ||
************************************************************************ | ||
******************* CANADIAN ASTRONOMY DATA CENTRE ******************* | ||
************** CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES ************** | ||
* | ||
* (c) 2024. (c) 2024. | ||
* Government of Canada Gouvernement du Canada | ||
* National Research Council Conseil national de recherches | ||
* Ottawa, Canada, K1A 0R6 Ottawa, Canada, K1A 0R6 | ||
* All rights reserved Tous droits réservés | ||
* | ||
* NRC disclaims any warranties, Le CNRC dénie toute garantie | ||
* expressed, implied, or énoncée, implicite ou légale, | ||
* statutory, of any kind with de quelque nature que ce | ||
* respect to the software, soit, concernant le logiciel, | ||
* including without limitation y compris sans restriction | ||
* any warranty of merchantability toute garantie de valeur | ||
* or fitness for a particular marchande ou de pertinence | ||
* purpose. NRC shall not be pour un usage particulier. | ||
* liable in any event for any Le CNRC ne pourra en aucun cas | ||
* damages, whether direct or être tenu responsable de tout | ||
* indirect, special or general, dommage, direct ou indirect, | ||
* consequential or incidental, particulier ou général, | ||
* arising from the use of the accessoire ou fortuit, résultant | ||
* software. Neither the name de l'utilisation du logiciel. Ni | ||
* of the National Research le nom du Conseil National de | ||
* Council of Canada nor the Recherches du Canada ni les noms | ||
* names of its contributors may de ses participants ne peuvent | ||
* be used to endorse or promote être utilisés pour approuver ou | ||
* products derived from this promouvoir les produits dérivés | ||
* software without specific prior de ce logiciel sans autorisation | ||
* written permission. préalable et particulière | ||
* par écrit. | ||
* | ||
* This file is part of the Ce fichier fait partie du projet | ||
* OpenCADC project. OpenCADC. | ||
* | ||
* OpenCADC is free software: OpenCADC est un logiciel libre ; | ||
* you can redistribute it and/or vous pouvez le redistribuer ou le | ||
* modify it under the terms of modifier suivant les termes de | ||
* the GNU Affero General Public la “GNU Affero General Public | ||
* License as published by the License” telle que publiée | ||
* Free Software Foundation, par la Free Software Foundation | ||
* either version 3 of the : soit la version 3 de cette | ||
* License, or (at your option) licence, soit (à votre gré) | ||
* any later version. toute version ultérieure. | ||
* | ||
* OpenCADC is distributed in the OpenCADC est distribué | ||
* hope that it will be useful, dans l’espoir qu’il vous | ||
* but WITHOUT ANY WARRANTY; sera utile, mais SANS AUCUNE | ||
* without even the implied GARANTIE : sans même la garantie | ||
* warranty of MERCHANTABILITY implicite de COMMERCIALISABILITÉ | ||
* or FITNESS FOR A PARTICULAR ni d’ADÉQUATION À UN OBJECTIF | ||
* PURPOSE. See the GNU Affero PARTICULIER. Consultez la Licence | ||
* General Public License for Générale Publique GNU Affero | ||
* more details. pour plus de détails. | ||
* | ||
* You should have received Vous devriez avoir reçu une | ||
* a copy of the GNU Affero copie de la Licence Générale | ||
* General Public License along Publique GNU Affero avec | ||
* with OpenCADC. If not, see OpenCADC ; si ce n’est | ||
* <http://www.gnu.org/licenses/>. pas le cas, consultez : | ||
* <http://www.gnu.org/licenses/>. | ||
* | ||
* | ||
************************************************************************ | ||
*/ | ||
|
||
package ca.nrc.cadc.dlm.server; | ||
|
||
public class ZIPPackageServlet extends PackageServlet { | ||
public static final String ZIP_PACKAGE_TARGET = "/zip-package"; | ||
private static final String ZIP_CONTENT_TYPE = "application/zip"; | ||
|
||
|
||
@Override | ||
String getContentType() { | ||
return ZIPPackageServlet.ZIP_CONTENT_TYPE; | ||
} | ||
} |
Oops, something went wrong.