console.log("Script loaded successfully");
// hook_okhttp_client()
if (Java.available) {
    hook_json()
    // hook_okhttp3_client()
    Java.perform(function () {
        try {
            var OkHttpClient = Java.use("okhttp3.OkHttpClient");
            console.log("OkHttp detected in the app");
        } catch (e) {
            console.log("OkHttp not detected in the app");
        }
    });
    Java.perform(function () {
        console.log("start hook java.net.URL");
        var URL = Java.use('java.net.URL');
        URL.$init.overload('java.lang.String').implementation = function (spec) {
            if (spec.includes("appsflyer")) {
                console.log("URL request: " + spec);
                if (spec.includes("conversions")) {
                    var stackTrace = Java.use('java.lang.Exception').$new().getStackTrace().toString();
                    console.log(stackTrace);
                }
            }
            return this.$init(spec);
        };
    });
}

function printMethods(className) {
    var jclass = Java.use(className);
    var methods = jclass.class.getDeclaredMethods();
    console.log("Printing methods of " + className + ":\n");
    methods.forEach(function (method) {
        console.log(method);
    });
}

function hook_okhttp_client() {
    if (Java.available) {
        Java.perform(function () {
            try {
                var OkHttpClient = Java.use("okhttp3.OkHttpClient");

                OkHttpClient.newCall.overload('okhttp3.Request').implementation = function (request) {
                    console.log("OkHttp Request URL: " + request.url().toString());
                    console.log("OkHttp Request Headers: " + request.headers().toString());

                    if (request.method() == "POST") {
                        console.log("OkHttp Request Body: " + request.body().contentType().toString());
                        // Here you can further extract the request body if needed.
                    }

                    var response = this.newCall(request).execute();
                    console.log("OkHttp Response: " + response.body().string());

                    // Note: Calling response.body().string() consumes the response body.
                    // You might need to recreate the response if the app expects to read it again.

                    return this.newCall(request);
                };

            } catch (e) {
                console.log("Error hooking OkHttp: " + e);
            }
        });
    }
}

function hook_HttpURLConnection_stream() {
    console.log("start hook_HttpURLConnection_stream")
    if (Java.available) {
        Java.perform(function () {
            var HttpURLConnection = Java.use("java.net.HttpURLConnection");

            HttpURLConnection.getOutputStream.implementation = function () {
                var outputStream = this.getOutputStream();
                var OutputStreamWrapper = Java.use("java.io.OutputStream");

                var newOutputStream = Java.registerClass({
                    name: "CustomOutputStream",
                    superClass: OutputStreamWrapper,
                    methods: {
                        write: function (buffer, byteOffset, byteCount) {
                            var data = Array.prototype.slice.call(buffer.slice(byteOffset, byteOffset + byteCount));
                            console.log("Request data: " + String.fromCharCode.apply(null, data));
                            outputStream.write(buffer, byteOffset, byteCount);
                        }
                    }
                });

                return newOutputStream.$new(outputStream);
            };

            HttpURLConnection.getInputStream.implementation = function () {
                var inputStream = this.getInputStream();
                var InputStreamWrapper = Java.use("java.io.InputStream");

                var newInputStream = Java.registerClass({
                    name: "CustomInputStream",
                    superClass: InputStreamWrapper,
                    methods: {
                        read: function (buffer, byteOffset, byteCount) {
                            var bytesRead = inputStream.read(buffer, byteOffset, byteCount);
                            if (bytesRead != -1) {
                                var data = Array.prototype.slice.call(buffer.slice(byteOffset, byteOffset + bytesRead));
                                console.log("Response data: " + String.fromCharCode.apply(null, data));
                            }
                            return bytesRead;
                        }
                    }
                });

                return newInputStream.$new(inputStream);
            };
        });
    }

}

function hook_okhttp3_client() {
    Java.perform(function () {
        var OkHttpClient = Java.use("okhttp3.OkHttpClient");
        OkHttpClient.newCall.overload("okhttp3.Request").implementation = function (request) {
            console.log("okhttp3 Request -> " + request.url().toString());

            var response = this.newCall(request).execute();

            console.log("okhttp3 Response -> " + response.body().string());

            return response;
        }
    });
}

function hook_retrofit() {
    Java.perform(function () {
        var retrofitBuilder = Java.use("retrofit2.Retrofit$Builder");
        retrofitBuilder.build.implementation = function () {
            var retrofit = this.build();
            var httpClient = retrofit.callFactory().clone();
            httpClient.interceptors().add(new Java.use('okhttp3.Interceptor')({
                intercept: function (chain) {
                    console.log("HTTP Request -> " + chain.request().toString())
                    var response = chain.proceed(chain.request());
                    console.log("HTTP Response -> " + response.toString());
                    return response;
                }
            }))
            return retrofit.newBuilder()
                .callFactory(httpClient)
                .build();
        }
    });
}


function hook_json() {
    Java.perform(function () {
        var JSONObject = Java.use('org.json.JSONObject');
        JSONObject.toString.overload().implementation = function () {
            var result = this.toString.call(this);
            console.log("Serialized JSONObject: " + result);
            return result;
        };
    });
}