package tinystruct.examples;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import org.tinystruct.AbstractApplication;

import org.tinystruct.ApplicationContext;

import org.tinystruct.ApplicationException;

import org.tinystruct.system.ApplicationManager;

/**

* Image to Base64 conversion

* Usage: dispatcher --image-path=/path/to/image image2base64

* @author m0ver

*/

public class image2 extends AbstractApplication {

private String imagePath;

@Override

public void init() {

this.setAction("image2base64", "image2base64");

}

@Override

public String version() {

return "1.0";

}

public String image2base64() throws ApplicationException {

if(this.context.getAttribute("--image-path")==null || this.context.getAttribute("--image-path").toString().trim().length()==0) {

throw new ApplicationException("Invalid file");

}

this.imagePath = this.context.getAttribute("--image-path").toString();

return this.image2base64(this.imagePath);

}

public String image2base64(String f) throws ApplicationException {

File file = new File(f);

String name = file.getName();

if(!name.endsWith(".jpeg") && !name.endsWith(".jpg")&& !name.endsWith(".png")&& !name.endsWith(".gif")&& !name.endsWith(".bmp"))

{

throw new ApplicationException("Invalid file");

}

ByteArrayOutputStream ous = null;

FileInputStream fs = null;

byte[] data = new byte[]{};

try {

byte[] buffer = new byte[4096];

ous = new ByteArrayOutputStream();

int read = 0;

fs = new FileInputStream(file);

while ((read = fs.read(buffer)) != -1) {

ous.write(buffer, 0, read);

}

data = ous.toByteArray();

if(data.length > 0)

return "<img src=\"data:image/png;base64," + org.tinystruct.system.util.Base64.encode(data) + "\" />";

} catch (FileNotFoundException e) {

throw new ApplicationException(e.getMessage(), e);

} catch (IOException e) {

throw new ApplicationException(e.getMessage(), e);

} finally {

try {

if (ous != null)

ous.close();

if (fs != null)

fs.close();

} catch (IOException e) {

e.printStackTrace();

}

}

return null;

}

public static void main(String[]args) throws ApplicationException {

ApplicationContext ctx = new ApplicationContext();

ApplicationManager.install(new image2());

ApplicationManager.call("image2base64", ctx);

}

}