com.trivago.cluecumber.json.pojo.Embedding Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cluecumber-report-plugin
Plugin for generating clear Cucumber BDD test result reports.
There is a newer version: 2.9.4
/*
 * Copyright 2018 trivago N.V.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.trivago.cluecumber.json.pojo;

import com.google.gson.annotations.SerializedName;
import com.trivago.cluecumber.constants.MimeType;
import org.codehaus.plexus.util.Base64;

import java.nio.charset.StandardCharsets;

public class Embedding {
	
    private String data;
    private String decodedData;
    @SerializedName("mime_type")
    private MimeType mimeType = MimeType.UNKNOWN;
        
    private transient String filename;
    
    public String getData() {
        return data;
    }

    public void setData(final String data) {
        this.data = data;
    }

    public String getDecodedData() {
        return decodedData;
    }

    public void encodeData(final String data) {
        if(mimeType.getContentType().equalsIgnoreCase("text/xml") || mimeType.getContentType().equalsIgnoreCase("application/xml")){
            String xmlString = new String(Base64.decodeBase64(data.getBytes(StandardCharsets.UTF_8)), StandardCharsets.UTF_8);
            decodedData = xmlString.replaceAll("<", "<").replaceAll(">", ">");
        }else{
            decodedData = new String(Base64.decodeBase64(data.getBytes(StandardCharsets.UTF_8)), StandardCharsets.UTF_8);
        }
    }

    public MimeType getMimeType() {
        return mimeType;
    }

    public void setMimeType(final MimeType mimeType) {
        this.mimeType = mimeType;
    }

    public String getFilename() {
        return filename;
    }

    public void setFilename(final String filename) {
        this.filename = filename;
    }

    public boolean isImage() {
        return
                mimeType.getContentType().equalsIgnoreCase("image/png") ||
                        mimeType.getContentType().equalsIgnoreCase("image/gif") ||
                        mimeType.getContentType().equalsIgnoreCase("image/bmp") ||
                        mimeType.getContentType().equalsIgnoreCase("image/jpg") ||                        
                        mimeType.getContentType().equalsIgnoreCase("image/jpeg") ||                        
                        mimeType.getContentType().equalsIgnoreCase("image/svg") ||
                        mimeType.getContentType().equalsIgnoreCase("image/svg+xml");
    }

    public boolean isPlainText() {
        return mimeType.getContentType().equalsIgnoreCase("text/plain");
    }    
    
    public String getFileEnding() {
        switch (mimeType) {
        case PNG:
        case GIF:
        case BMP:
        case JPG:
        case JPEG:
        case HTML:
        case XML:
        case JSON:
        case APPLICATION_XML:
            return mimeType.getContentType().substring(mimeType.getContentType().indexOf('/') + 1);
        case SVG:
        case SVG_XML:
            return "svg";
        case TXT:
            return "txt";
        case PDF:
            return "pdf";
        default:
            return "unknown";
        }
    }        
}