Java Code Examples for org.codehaus.plexus.util.IOUtil#copy()

void extractJar(File jar, ManifestTransformer manifestTransformer) throws MojoExecutionException {
    try (JarFile jarFile = new JarFile(jar)) {
        for (Enumeration<JarEntry> jarEntries = jarFile.entries(); jarEntries.hasMoreElements(); ) {
            JarEntry jarEntry = jarEntries.nextElement();
            if (manifestTransformer.canTransform(jarEntry)) {
                jarEntry = manifestTransformer.transform(jarEntry);
            }
            if (!jarEntry.isDirectory() && !content.contains(jarEntry.getName())) {
                content.add(jarEntry.getName());
                makeDirsRecursively(jarEntry.getName());
                try (InputStream in = getInputStream(jarEntry, jarFile, manifestTransformer)) {
                    jarOutputStream.putNextEntry(jarEntry);
                    IOUtil.copy(in, jarOutputStream);
                }
            }
        }
    } catch (IOException e) {
        throw new MojoExecutionException("Error adding " + jar.getName() + " to target JAR: " + e.getMessage(), e);
    }
}
 

protected void writeFile( String filePath, String content )
    throws IOException
{
    FileOutputStream fs = new FileOutputStream( filePath );
    try
    {
        IOUtil.copy( content, fs );
    }
    finally
    {
        if ( fs != null )
        {
            IOUtil.close( fs );
        }
    }
}
 

private String interpolateBaseDirAndRepo( String content )
{
    StringReader sr = new StringReader( content );
    StringWriter result = new StringWriter();

    Map<Object, Object> context = new HashMap<Object, Object>();

    context.put( "BASEDIR", StringUtils.quoteAndEscape( getBasedir(), '"' ) );
    context.put( "REPO", StringUtils.quoteAndEscape( getRepo(), '"' ) );
    InterpolationFilterReader interpolationFilterReader = new InterpolationFilterReader( sr, context, "@", "@" );
    try
    {
        IOUtil.copy( interpolationFilterReader, result );
    }
    catch ( IOException e )
    {
        // shouldn't happen...
    }
    return result.toString();
}
 

private static void writeFile( File outputFile, Reader reader )
    throws DaemonGeneratorException
{
    FileWriter out = null;

    try
    {
        outputFile.getParentFile().mkdirs();
        out = new FileWriter( outputFile );

        IOUtil.copy( reader, out );
    }
    catch ( IOException e )
    {
        throw new DaemonGeneratorException( "Error writing output file: " + outputFile.getAbsolutePath(), e );
    }
    finally
    {
        IOUtil.close( reader );
        IOUtil.close( out );
    }
}
 

private void saveAndCompare( String expectedResource )
    throws IOException
{
    StringOutputStream string = new StringOutputStream();
    formattedProperties.save( string );

    StringOutputStream expected = new StringOutputStream();
    InputStream asStream = getClass().getResourceAsStream( expectedResource );
    try
    {
        IOUtil.copy( asStream, expected );
    }
    finally
    {
        IOUtil.close( asStream );
    }

    String unified = StringUtils.unifyLineSeparators( expected.toString() );
    assertEquals( unified, string.toString() );
}
 

void addFile(File srcFile, String targetFileName, Directory targetDir) throws MojoExecutionException {
    String destPath = targetDir.getName() + targetFileName;
    if (content.contains(destPath)) {
        return;
    }
    makeDirsRecursively(destPath);
    content.add(destPath);
    try (InputStream in = new FileInputStream(srcFile)) {
        jarOutputStream.putNextEntry(new JarEntry(destPath));
        IOUtil.copy(in, jarOutputStream);
    } catch (IOException e) {
        throw new MojoExecutionException("Error adding " + srcFile.getName() + " to target JAR: " + e.getMessage(), e);
    }
}
 

/**
 * Writes a StringBuilder into a file.
 *
 * @param outFile The file to read.
 * @param input The contents of the file.
 * @throws IOException when things go wrong.
 */
protected final void writeFile( File outFile, StringBuilder input )
    throws IOException
{
    Writer writer = WriterFactory.newXmlWriter( outFile );
    try
    {
        IOUtil.copy( input.toString(), writer );
    }
    finally
    {
        IOUtil.close( writer );
    }
}
 

private static String interpolateBaseDirAndRepo( String content )
{
    StringReader sr = new StringReader( content );
    StringWriter result = new StringWriter();

    Map<Object, Object> context = new HashMap<Object, Object>();

    final String baseDir = System.getProperty( "basedir", System.getProperty( "app.home" ) );
    if ( baseDir != null && baseDir.length() > 0) {
        context.put( "BASEDIR", StringUtils.quoteAndEscape(baseDir, '"') );
    }

    final String repo = System.getProperty( "app.repo" );
    if ( repo != null && repo.length() > 0 ) {
        context.put("REPO", StringUtils.quoteAndEscape(repo, '"'));
    }

    InterpolationFilterReader interpolationFilterReader = new InterpolationFilterReader( sr, context, "@", "@" );
    try
    {
        IOUtil.copy( interpolationFilterReader, result );
    }
    catch ( IOException e )
    {
        // shouldn't happen...
    }
    return result.toString();
}
 

protected File createFilteredFile( String file )
    throws IOException, FileNotFoundException, DaemonGeneratorException, XmlPullParserException
{
    String version = getAppAssemblerBooterVersion();
    Properties context = new Properties();
    context.setProperty( "appassembler.version", version );

    File tempPom = File.createTempFile( "appassembler", "tmp" );
    tempPom.deleteOnExit();

    InterpolationFilterReader reader =
        new InterpolationFilterReader( new FileReader( getTestFile( file ) ), context, "@", "@" );
    FileWriter out = null;

    try
    {
        out = new FileWriter( tempPom );

        IOUtil.copy( reader, out );
    }
    catch ( IOException e )
    {
        throw new DaemonGeneratorException( "Error writing output file: " + tempPom.getAbsolutePath(), e );
    }
    finally
    {
        IOUtil.close( reader );
        IOUtil.close( out );
    }
    return tempPom;
}
 

private File copyToTempDir(String resource) throws IOException {
    File dir = Files.createTempDirectory("d-m-p").toFile();
    File ret = new File(dir, "Dockerfile");
    try (FileOutputStream os = new FileOutputStream(ret)) {
        IOUtil.copy(getClass().getResourceAsStream(resource), os);
    }
    return ret;
}
 

JarOutputStream addToJar(final String name, final InputStream input, final JarOutputStream jar) throws IOException {
	try {
		debug("\t[Added to Jar]: " + name);
		jar.putNextEntry(new ZipEntry(name));
		IOUtil.copy(input, jar);
		jar.closeEntry();
	} catch (final ZipException ignore) {} // ignore duplicate entries and other errors
	IOUtil.close(input);
	return jar;
}
 

private static void readPomInfo(String location, List<PomInfo> pomInfos) {
    Properties properties = new Properties();
    String metaPath = "META-INF";
    try {
        InputStream is;
        if (location.contains("!")) {
            // read inside jar's
            is = new ClassPathResource(location.substring(location.indexOf("!") + 1)).getInputStream();
        } else {
            URL realUrl = new URL(location);
            is = realUrl.openStream();
        }

        PomInfo pomInfo = null;
        ZipInputStream zip = new ZipInputStream(is);
        ZipEntry zipEntry;
        while ((zipEntry = zip.getNextEntry()) != null) {
            String zipEntryPath = zipEntry.getName();
            if (zipEntryPath.startsWith(metaPath + "/maven") && zipEntryPath.endsWith("pom.xml")) {
                logger.debug("zipEntryPath: " + zipEntryPath);
                pomInfo = readPom(zip);
                break;
            }

            if (zipEntryPath.equals(metaPath + "/MANIFEST.MF")) {
                properties.load(zip);
            }
        }

        // calculate jar size
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        IOUtil.copy(is, output);
        long jarSize = output.toByteArray().length;

        if (null != pomInfo) {
            pomInfo.setLocation(location);
            pomInfo.setSize(jarSize);
            pomInfos.add(pomInfo);
        }
    } catch (Exception e) {
        logger.error("get jar maven pom failed! location:" + location, e);
    }
}
 

private void createOpenShiftConfig(File homeDir,String testConfig) throws IOException {
    File kubeDir = new File(homeDir,".kube");
    kubeDir.mkdirs();
    File config = new File(kubeDir,"config");
    IOUtil.copy(getClass().getResourceAsStream(testConfig),new FileWriter(config));
}