com.jdiai.visual.StreamToImageVideo Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jdi-nova Show documentation

JDI Nova

There is a newer version: 1.1.8

package com.jdiai.visual;

import com.jdiai.jsdriver.JDINovaException;

import java.io.File;
import java.io.FileOutputStream;
import java.util.Base64;

public class StreamToImageVideo {
    private final String base64;
    private final ImageTypes imageType;
    public StreamToImageVideo(String stream, ImageTypes imageType) {
        this.base64 = stream.contains("base64") ? stream.substring(stream.indexOf("base64")+7) : "";
        this.imageType = imageType;
    }
    public String asBase64() {
        return base64;
    }
    public byte[] asByteStream() {
        try {
            return Base64.getDecoder().decode(base64);
        } catch (Exception ex) {
            throw new JDINovaException(ex, "Failed to decode screenshot base64 stream: " + base64);
        }
    }
    public File asFile(String fileName) {
        String path = fileName + "." + imageType.suffix;
        FileOutputStream osf = null;
        File file;
        try {
            file = new File(path);
            file.getParentFile().mkdirs();
            osf = new FileOutputStream(file);
            osf.write(asByteStream());
            osf.flush();
            return file;
        } catch (Exception ex) {
            throw new JDINovaException(ex, "Failed to save screenshot to file: " + path);
        } finally {
            try {
                if (osf != null) {
                    osf.close();
                }
            } catch (Exception ignore) { }
        }
    }
}