flipper/android/core/SonarArray.java

/*
 *  Copyright (c) 2018-present, Facebook, Inc.
 *
 *  This source code is licensed under the MIT license found in the LICENSE
 *  file in the root directory of this source tree.
 *
 */
package com.facebook.sonar.core;

import java.util.ArrayList;
import java.util.List;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

public class SonarArray {
  final JSONArray mJson;

  SonarArray(JSONArray json) {
    mJson = (json != null ? json : new JSONArray());
  }

  SonarArray(String json) {
    try {
      mJson = new JSONArray(json);
    } catch (JSONException e) {
      throw new RuntimeException(e);
    }
  }

  public SonarDynamic getDynamic(int index) {
    return new SonarDynamic(mJson.opt(index));
  }

  public String getString(int index) {
    return mJson.optString(index);
  }

  public int getInt(int index) {
    return mJson.optInt(index);
  }

  public long getLong(int index) {
    return mJson.optLong(index);
  }

  public float getFloat(int index) {
    return (float) mJson.optDouble(index);
  }

  public double getDouble(int index) {
    return mJson.optDouble(index);
  }

  public boolean getBoolean(int index) {
    return mJson.optBoolean(index);
  }

  public SonarObject getObject(int index) {
    final Object o = mJson.opt(index);
    return new SonarObject((JSONObject) o);
  }

  public SonarArray getArray(int index) {
    final Object o = mJson.opt(index);
    return new SonarArray((JSONArray) o);
  }

  public int length() {
    return mJson.length();
  }

  public List<String> toStringList() {
    final int length = length();
    final List<String> list = new ArrayList<>(length);
    for (int i = 0; i < length; i++) {
      list.add(getString(i));
    }
    return list;
  }

  public String toJsonString() {
    return toString();
  }

  @Override
  public String toString() {
    return mJson.toString();
  }

  @Override
  public boolean equals(Object o) {
    return mJson.toString().equals(o.toString());
  }

  @Override
  public int hashCode() {
    return mJson.hashCode();
  }

  public static class Builder {
    private final JSONArray mJson;

    public Builder() {
      mJson = new JSONArray();
    }

    public Builder put(String s) {
      mJson.put(s);
      return this;
    }

    public Builder put(Integer i) {
      mJson.put(i);
      return this;
    }

    public Builder put(Long l) {
      mJson.put(l);
      return this;
    }

    public Builder put(Float f) {
      mJson.put(Float.isNaN(f) ? null : f);
      return this;
    }

    public Builder put(Double d) {
      mJson.put(Double.isNaN(d) ? null : d);
      return this;
    }

    public Builder put(Boolean b) {
      mJson.put(b);
      return this;
    }

    public Builder put(SonarValue v) {
      return put(v.toSonarObject());
    }

    public Builder put(SonarArray a) {
      mJson.put(a == null ? null : a.mJson);
      return this;
    }

    public Builder put(SonarArray.Builder b) {
      return put(b.build());
    }

    public Builder put(SonarObject o) {
      mJson.put(o == null ? null : o.mJson);
      return this;
    }

    public Builder put(SonarObject.Builder b) {
      return put(b.build());
    }

    public SonarArray build() {
      return new SonarArray(mJson);
    }
  }
}