com.bosch.osmi.sw360.cvesearch.datasource.json.CveSearchJsonParser.java Source code

Introduction

Here is the source code for com.bosch.osmi.sw360.cvesearch.datasource.json.CveSearchJsonParser.java

Source

/*
 * Copyright (c) Bosch Software Innovations GmbH 2016.
 * Part of the SW360 Portal Project.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package com.bosch.osmi.sw360.cvesearch.datasource.json;

import com.bosch.osmi.sw360.cvesearch.datasource.CveSearchData;
import com.google.gson.*;

import java.io.BufferedReader;
import java.lang.reflect.Type;
import java.util.function.Function;

public class CveSearchJsonParser implements Function<BufferedReader, Object> {
    private Type type;
    private Gson gson;

    public CveSearchJsonParser(Type type) {
        this.type = type;
        GsonBuilder gsonBuilder = new GsonBuilder();
        gsonBuilder.registerTypeAdapter(CveSearchData.DateTimeObject.class, new DateTimeObjectDeserializer());
        gsonBuilder.registerTypeAdapter(CveSearchData.VulnerableConfigurationEntry.class,
                new VulnerableConfigurationEntryDeserializer());
        gson = gsonBuilder.create();
    }

    private class VulnerableConfigurationEntryDeserializer
            implements JsonDeserializer<CveSearchData.VulnerableConfigurationEntry> {
        @Override
        public CveSearchData.VulnerableConfigurationEntry deserialize(JsonElement jsonElement, Type type,
                JsonDeserializationContext jsonDeserializationContext) throws JsonParseException {
            if (jsonElement.isJsonPrimitive()) {
                final String id = jsonElement.getAsString();
                return new CveSearchData.VulnerableConfigurationEntry(id);
            } else {
                final JsonObject jsonObject = jsonElement.getAsJsonObject();
                final String title = jsonObject.get("title").getAsString();
                final String id = jsonObject.get("id").getAsString();
                return new CveSearchData.VulnerableConfigurationEntry(title, id);
            }
        }
    }

    private class DateTimeObjectDeserializer implements JsonDeserializer<CveSearchData.DateTimeObject> {
        @Override
        public CveSearchData.DateTimeObject deserialize(JsonElement jsonElement, Type type,
                JsonDeserializationContext jsonDeserializationContext) throws JsonParseException {
            if (jsonElement.isJsonPrimitive()) {
                final String formattedDate = jsonElement.getAsString();
                return new CveSearchData.DateTimeObject(formattedDate);
            } else {
                final JsonObject jsonObject = jsonElement.getAsJsonObject();
                if (jsonObject.has("$date")) {
                    return new CveSearchData.DateTimeObject(jsonObject.get("$date").getAsLong());
                }
                return null;
            }
        }
    }

    @Override
    public Object apply(BufferedReader json) {
        return gson.fromJson(json, type);
    }
}