org.apache.wicket.util.resource.IResourceStream Java Examples

public IResourceStream createResourceStreamWriter()
{
  final IResourceStream iResourceStream = new AbstractResourceStreamWriter() {
    private static final long serialVersionUID = 7780552906708508709L;

    @Override
    public String getContentType()
    {
      return "application/json";
    }

    @Override
    public void write(final OutputStream output)
    {
      try {
        IOUtils.write(new Gson().toJson(result), output);
      } catch (IOException ex) {
        log.error("Exception encountered " + ex, ex);
      }
    }
  };
  return iResourceStream;
}
 

public IResourceStream createResourceStreamWriter()
{
  final IResourceStream iResourceStream = new AbstractResourceStreamWriter() {
    private static final long serialVersionUID = 7780552906708508709L;

    @Override
    public String getContentType()
    {
      return "application/zip";
    }

    @Override
    public void write(final OutputStream output)
    {
      ExportZipArchive.this.write(output);
    }
  };
  return iResourceStream;
}
 

/**
 * Constructor.
 *
 * @param sassStream
 *            The resource stream that loads the Sass content. Only UrlResourceStream is
 *            supported at the moment!
 * @param scopeClass
 *            The name of the class used as a scope to resolve "package!" dependencies/imports
 */
public SassResourceStream(IResourceStream sassStream, String scopeClass)
{
    Args.notNull(sassStream, "sassStream");

    while (sassStream instanceof ResourceStreamWrapper) {
        ResourceStreamWrapper wrapper = (ResourceStreamWrapper) sassStream;
        try {
            sassStream = wrapper.getDelegate();
        }
        catch (Exception x) {
            throw new WicketRuntimeException(x);
        }
    }

    if (!(sassStream instanceof UrlResourceStream)) {
        throw new IllegalArgumentException(String.format("%s can work only with %s",
                SassResourceStream.class.getSimpleName(), UrlResourceStream.class.getName()));
    }

    URL sassUrl = ((UrlResourceStream) sassStream).getURL();

    SassCacheManager cacheManager = SassCacheManager.get();

    this.sassSource = cacheManager.getSassContext(sassUrl, scopeClass);
}
 

private IResourceStream exportUimaTypeSystem()
{
    try (ByteArrayOutputStream bos = new ByteArrayOutputStream()) {
        TypeSystemDescription tsd = annotationService
                .getAllProjectTypes(getModelObject().getProject());
        tsd.toXML(bos);
        return new InputStreamResourceStream(new ByteArrayInputStream(bos.toByteArray()));
    }
    catch (Exception e) {
        error("Unable to generate the UIMA type system file: "
                + ExceptionUtils.getRootCauseMessage(e));
        ProjectLayersPanel.LOG.error("Unable to generate the UIMA type system file", e);
        RequestCycle.get().find(IPartialPageRequestHandler.class)
                .ifPresent(handler -> handler.addChildren(getPage(), IFeedback.class));
        return null;
    }
}
 

private IResourceStream exportLayerJson()
{
    try {
        AnnotationLayer layer = getModelObject();

        List<ExportedAnnotationLayer> exLayers = new ArrayList<>();

        ExportedAnnotationLayer exMainLayer = ImportUtil.exportLayerDetails(null, null,
                layer, annotationService);
        exLayers.add(exMainLayer);

        // If the layer is attached to another layer, then we also have to export
        // that, otherwise we would be missing it during re-import.
        if (layer.getAttachType() != null) {
            AnnotationLayer attachLayer = layer.getAttachType();
            ExportedAnnotationLayer exAttachLayer = ImportUtil.exportLayerDetails(null,
                    null, attachLayer, annotationService);
            exMainLayer.setAttachType(
                    new ExportedAnnotationLayerReference(exAttachLayer.getName()));
            exLayers.add(exAttachLayer);
        }

        return new InputStreamResourceStream(new ByteArrayInputStream(
                JSONUtil.toPrettyJsonString(exLayers).getBytes("UTF-8")));
    }
    catch (Exception e) {
        error("Unable to generate the JSON file: " + ExceptionUtils.getRootCauseMessage(e));
        ProjectLayersPanel.LOG.error("Unable to generate the JSON file", e);
        RequestCycle.get().find(IPartialPageRequestHandler.class)
                .ifPresent(handler -> handler.addChildren(getPage(), IFeedback.class));
        return null;
    }
}
 

public IResourceStream locate(final Class clazz, final String path) {
	IResourceStream located = super.locate(clazz, trimFolders(path));
	if(located != null){
		return located;
	}
	return super.locate(clazz, path);
}
 

public IResourceStream locate(final Class clazz, final String path) {
	IResourceStream located = super.locate(clazz, trimFolders(path));
	if(located != null){
		return located;
	}
	return super.locate(clazz, path);
}
 

public IResourceStream getMarkupResourceStream(MarkupContainer container,
		Class<?> containerClass, String wrapTag) {
	if(container.getClass().equals(containerClass)) {
		return new OPageResourceStream(pageDocumentModel.getObject(), wrapTag);
	} else {
		return DEFAULT_MARKUP_PROVIDER.getMarkupResourceStream(container, containerClass);
	}
}
 

void commonInit()
{
    downloadBehavior = new AbstractAjaxBehavior()
    {
        private static final long serialVersionUID = 3472918725573624819L;

        @Override
        public void onRequest()
        {
            IResourceStream is = AjaxDownloadLink.this.getModelObject();
            
            if (is != null) {
                // If no filename has been set explicitly, try to get it from the resource
                String name = filename != null ? filename.getObject() : null;
                if (name == null) {
                    if (is instanceof FileResourceStream) {
                        name = ((FileResourceStream) is).getFile().getName();
                    }
                    else if (is instanceof FileSystemResourceStream) {
                        name = ((FileSystemResourceStream) is).getPath().getFileName()
                                .toString();
                    }
                }
                
                ResourceStreamRequestHandler handler = new ResourceStreamRequestHandler(
                        AjaxDownloadLink.this.getModelObject(), name);
                handler.setContentDisposition(ContentDisposition.ATTACHMENT);
                getComponent().getRequestCycle().scheduleRequestHandlerAfterCurrent(handler);
            }
        }
    };
    add(downloadBehavior);
}
 

@Override
protected AbstractLink newLink(String id) {
	IResource resource = new ResourceStreamResource()
	{
		@Override
		protected IResourceStream getResourceStream(Attributes attrs)
		{
			return new DataExportResourceStreamWriter(dataExporter, table);
		}
	}.setFileName(fileNameModel.getObject() + "." + dataExporter.getFileNameExtension());

	return new ResourceLink<Void>(id, resource);
}
 

private IResourceStream exportLayer()
{
    switch (exportMode) {
    case JSON:
        return exportLayerJson();
    case UIMA:
        return exportUimaTypeSystem();
    default:
        throw new IllegalStateException("Unknown mode: [" + exportMode + "]");
    }
}
 

public GuidelinesDialogContent(String aId, final ModalWindow modalWindow,
        final IModel<AnnotatorState> aModel)
{
    super(aId);

    // Overall progress by Projects
    RepeatingView guidelineRepeater = new RepeatingView("guidelineRepeater");
    add(guidelineRepeater);
    
    for (String guidelineFileName : projectService
            .listGuidelines(aModel.getObject().getProject())) {
        AbstractItem item = new AbstractItem(guidelineRepeater.newChildId());

        guidelineRepeater.add(item);

        // Add a popup window link to display annotation guidelines
        PopupSettings popupSettings = new PopupSettings(RESIZABLE | SCROLLBARS)
                        .setHeight(500)
                        .setWidth(700);

        IResourceStream stream = new FileResourceStream(projectService
                .getGuideline(aModel.getObject().getProject(), guidelineFileName));
        ResourceStreamResource resource = new ResourceStreamResource(stream);
        ResourceLink<Void> rlink = new ResourceLink<>("guideine", resource);
        rlink.setPopupSettings(popupSettings);
        item.add(new Label("guidelineName", guidelineFileName));
        item.add(rlink);
    }
    
    add(new LambdaAjaxLink("cancel", (target) -> modalWindow.close(target)));
}
 

@Override
protected IResourceStream getResourceStream(Attributes attributes) {
	final boolean admin = hasAdminLevel(getRights());
	final StringBuilder sb = new StringBuilder();
	chatForm.process(
			() -> {
				if (admin) {
					setFileName(String.format(CHAT_FNAME_TMPL, "global"));
					export(chatDao.getGlobal(0, Integer.MAX_VALUE), sb);
				}
				return true;
			}
			, r -> {
				if (admin || isModerator(cm, getUserId(), r.getId())) {
					setFileName(String.format(CHAT_FNAME_TMPL, "room_" + r.getId()));
					export(chatDao.getRoom(r.getId(), 0, Integer.MAX_VALUE, true), sb);
				}
				return true;
			}, u -> {
				setFileName(String.format(CHAT_FNAME_TMPL, "user_" + u.getId()));
				export(chatDao.getUser(u.getId(), 0, Integer.MAX_VALUE), sb);
				return true;
			});
	StringResourceStream srs = new StringResourceStream(sb, "text/csv");
	srs.setCharset(UTF_8);
	return srs;
}
 

@Override
public IResource getResource() {
	ConcatBundleResource bundleResource = new ConcatBundleResource(getProvidedResources()) {

		@Override
		protected byte[] readAllResources(List<IResourceStream> resources)
				throws IOException, ResourceStreamNotFoundException {
			ByteArrayOutputStream output = new ByteArrayOutputStream();
			for (IResourceStream curStream : resources) {
				IOUtils.copy(curStream.getInputStream(), output);
				output.write(";\n".getBytes());
			}

			byte[] bytes = output.toByteArray();

			if (getCompressor() != null) {
				String nonCompressed = new String(bytes, "UTF-8");
				bytes = getCompressor().compress(nonCompressed).getBytes("UTF-8");
			}

			return bytes;
		}
		
	};
	ITextResourceCompressor compressor = getCompressor();
	if (compressor != null) {
		bundleResource.setCompressor(compressor);
	}
	return bundleResource;
}
 

@Override
public IResource getResource() {
	ConcatBundleResource bundleResource = new ConcatBundleResource(getProvidedResources()) {

		@Override
		protected byte[] readAllResources(List<IResourceStream> resources)
				throws IOException, ResourceStreamNotFoundException {
			ByteArrayOutputStream output = new ByteArrayOutputStream();
			for (IResourceStream curStream : resources) {
				IOUtils.copy(curStream.getInputStream(), output);
				output.write("\n".getBytes());
			}

			byte[] bytes = output.toByteArray();

			if (getCompressor() != null) {
				String nonCompressed = new String(bytes, "UTF-8");
				bytes = getCompressor().compress(nonCompressed).getBytes("UTF-8");
			}

			return bytes;
		}
		
	};
	ITextResourceCompressor compressor = getCompressor();
	if (compressor != null) {
		bundleResource.setCompressor(compressor);
	}
	return bundleResource;
}
 

public static void setDownloadTarget(final String filename, final IResourceStream resourceStream)
{
  final ResourceStreamRequestHandler handler = new ResourceStreamRequestHandler(resourceStream);
  handler.setFileName(filename).setContentDisposition(ContentDisposition.ATTACHMENT);
  RequestCycle.get().scheduleRequestHandlerAfterCurrent(handler);
  log.info("Starting download for file. filename:" + filename + ", content-type:" + resourceStream.getContentType());
}
 

private Behavior makeDownloadBehavior(final String aKey1, final String aKey2)
{
    return new AjaxEventBehavior("click")
    {
        private static final long serialVersionUID = 1L;

        @Override
        protected void onEvent(AjaxRequestTarget aTarget)
        {
            AJAXDownload download = new AJAXDownload() {
                private static final long serialVersionUID = 1L;
                
                @Override
                protected IResourceStream getResourceStream()
                {
                    return new AbstractResourceStream() {
                        private static final long serialVersionUID = 1L;

                        @Override
                        public InputStream getInputStream()
                            throws ResourceStreamNotFoundException
                        {
                            try {
                                CodingAgreementResult result = PairwiseCodingAgreementTable.this
                                        .getModelObject().getStudy(aKey1, aKey2);
                                
                                switch (formatField.getModelObject()) {
                                case CSV:
                                    return AgreementUtils.generateCsvReport(result);
                                case DEBUG:
                                    return generateDebugReport(result);
                                default:
                                    throw new IllegalStateException("Unknown export format ["
                                            + formatField.getModelObject() + "]");
                                }
                            }
                            catch (Exception e) {
                                // FIXME Is there some better error handling here?
                                LOG.error("Unable to generate agreement report", e);
                                throw new ResourceStreamNotFoundException(e);
                            }
                        }

                        @Override
                        public void close()
                            throws IOException
                        {
                            // Nothing to do
                        }
                    };
                }
            };
            getComponent().add(download);
            download.initiate(aTarget,
                    "agreement" + formatField.getModelObject().getExtension());
        }
    };      
}
 

@Override
public IResourceStream getMarkupResourceStream(MarkupContainer container,
		Class<?> containerClass) {
	return delegate.getMarkupResourceStream(container, containerClass, "wicket:extend");
}
 

@Override
public IResourceStream getMarkupResourceStream(MarkupContainer container,
		Class<?> containerClass) {
	return delegate.getMarkupResourceStream(container, containerClass);
}
 

@Override
public IResourceStream getResourceStream()
{
  return new FileResourceStream(file);
}
 

@Override
public IResourceStream getMarkupResourceStream(MarkupContainer container,
		Class<?> containerClass) {
	return getMarkupResourceStream(container, containerClass, null);
}
 

@Override
public IResourceStream getMarkupResourceStream(final MarkupContainer container,
        final Class<?> containerClass) {
    return new StringResourceStream("<html><body>"
            + "<form wicket:id=\"form\"><span wicket:id=\"field\"></span></form></body></html>");
}
 

public static final IResourceStream getResourceStream(OccurrenceIF occ)  {
  if (occ == null) return null;
  Reader reader = occ.getReader();
  return (reader == null ? null : new Base64EncodedResourceStream(reader));
}
 

@Override
public IResourceStream getResourceStream() {
  Reader reader = getReader();
  return (reader == null ? null : new Base64EncodedResourceStream(reader));
}
 

public OccurrenceResourceStreamRequestTarget(IResourceStream resourceStream, String topicMapId, String occurrenceId) {
  super(resourceStream);
  this.topicMapId = topicMapId;
  this.occurrenceId = occurrenceId;
}
 

private IResourceStream exportAllAgreements()
    {
        return new AbstractResourceStream()
        {
            private static final long serialVersionUID = 1L;

            @Override
            public InputStream getInputStream()
                throws ResourceStreamNotFoundException
            {
                AnnotationFeature feature = getModelObject().getFeature();
                DefaultAgreementTraits traits = getModelObject().getTraits();
                
                Map<String, List<CAS>> casMap = casMapSupplier.get();

                List<DiffAdapter> adapters = CasDiff.getDiffAdapters(annotationService,
                        asList(feature.getLayer()));

                CasDiff diff = doDiff(adapters, traits.getLinkCompareBehavior(), casMap);

//                AgreementResult agreementResult = AgreementUtils.makeStudy(diff,
//                        feature.getLayer().getName(), feature.getName(),
//                        pref.excludeIncomplete, casMap);
                // TODO: for the moment, we always include incomplete annotations during this
                // export.
                CodingAgreementResult agreementResult = makeCodingStudy(diff,
                        feature.getLayer().getName(), feature.getName(), false, casMap);
                
                try {
                    return AgreementUtils.generateCsvReport(agreementResult);
                }
                catch (Exception e) {
                    // FIXME Is there some better error handling here?
                    LOG.error("Unable to generate report", e);
                    throw new ResourceStreamNotFoundException(e);
                }
            }

            @Override
            public void close()
                throws IOException
            {
                // Nothing to do
            }
        };
    }
 

public AjaxDownloadLink(String aId, IModel<String> aFilename, IModel<IResourceStream> aData)
{
    super(aId, aData);
    filename = aFilename;
    commonInit();
}
 

public AjaxDownloadLink(String aId, IModel<IResourceStream> aData)
{
    super(aId, aData);
    filename = null;
    commonInit();
}
 

@Override
public IResourceStream getResourceStream()
{
    IResourceStream resourceStream = super.getResourceStream();
    return new SassResourceStream(resourceStream, getScope().getName());
}
 

/**
 * {@inheritDoc}
 */
@Override
public IResourceStream find(final Class<?> aClass, final String s) {
    return configureMultiWebApplicationPath().find(aClass, s);
}