// Copyright (C) 2015 The Android Open Source Project
//
// 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.ericsson.gerrit.plugins.highavailability.forwarder.rest;

import com.ericsson.gerrit.plugins.highavailability.cache.Constants;
import com.google.common.base.Strings;
import com.google.gerrit.reviewdb.client.Account;
import com.google.gerrit.reviewdb.client.AccountGroup;
import com.google.gerrit.server.events.EventGson;
import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.inject.Inject;
import com.google.inject.Singleton;

@Singleton
class GsonParser {
  private final Gson gson;

  @Inject
  public GsonParser(@EventGson Gson gson) {
    this.gson = gson;
  }

  public Object fromJson(String cacheName, String jsonString) {
    JsonElement json = gson.fromJson(Strings.nullToEmpty(jsonString), JsonElement.class);
    Object key;
    // Need to add a case for 'adv_bases'
    if (!json.isJsonObject()) {
      return json.getAsString();
    }
    JsonObject asJsonObject = json.getAsJsonObject();
    switch (cacheName) {
      case Constants.ACCOUNTS:
        key = asJsonObject.has("id") ? Account.id(asJsonObject.get("id").getAsInt()) : null;
        break;
      case Constants.GROUPS:
        key = asJsonObject.has("id") ? AccountGroup.id(asJsonObject.get("id").getAsInt()) : null;
        break;
      case Constants.GROUPS_BYINCLUDE:
      case Constants.GROUPS_MEMBERS:
        key =
            asJsonObject.has("uuid")
                ? AccountGroup.uuid(asJsonObject.get("uuid").getAsString())
                : null;
        break;
      case Constants.PROJECT_LIST:
        key = gson.fromJson(json, Object.class);
        break;
      default:
        try {
          key = gson.fromJson(json, String.class);
        } catch (Exception e) {
          key = gson.fromJson(json, Object.class);
        }
    }
    return key;
  }
}
