components/cronet/android/api/src/org/chromium/net/apihelpers/JsonCronetCallback.java

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.net.apihelpers;

import org.json.JSONException;
import org.json.JSONObject;

import org.chromium.net.UrlResponseInfo;

/**
 * A specialization of {@link InMemoryTransformCronetCallback} that interprets the response body as
 * JSON.
 */
public abstract class JsonCronetCallback extends InMemoryTransformCronetCallback<JSONObject> {
    private static final StringCronetCallback STRING_CALLBACK =
            new StringCronetCallback() {
                @Override
                protected boolean shouldFollowRedirect(
                        UrlResponseInfo info, String newLocationUrl) {
                    throw new UnsupportedOperationException();
                }
            };

    @Override // Override to return the subtype
    public JsonCronetCallback addCompletionListener(
            CronetRequestCompletionListener<? super JSONObject> listener) {
        super.addCompletionListener(listener);
        return this;
    }

    @Override
    protected JSONObject transformBodyBytes(UrlResponseInfo info, byte[] bodyBytes) {
        String bodyString = STRING_CALLBACK.transformBodyBytes(info, bodyBytes);
        try {
            return new JSONObject(bodyString);
        } catch (JSONException e) {
            // As suggested by JSONException javadoc
            throw new IllegalArgumentException("Cannot parse the string as JSON!", e);
        }
    }
}