Ruby runtime = sc.getProvider().getRuntime();

if (runtime != null) {
    // CRITICAL: Unregister current thread BEFORE tearDown
    // tearDown() replaces ThreadService, must clean up old one first
    ThreadContext currentContext = runtime.getThreadService().getCurrentContext();
    runtime.getThreadService().unregisterCurrentThread(currentContext);
}

// Terminate the container
sc.terminate();

// Force runtime teardown
if (runtime != null) {
    runtime.tearDown(true);
}

// Clear global runtime reference
Ruby.clearGlobalRuntime();

import org.jruby.Ruby;
import org.jruby.embed.LocalContextScope;
import org.jruby.embed.ScriptingContainer;
import org.jruby.runtime.ThreadContext;

import java.lang.management.ManagementFactory;
import java.lang.management.MemoryMXBean;
import java.lang.management.MemoryUsage;

/**
 * Minimal test demonstrating ThreadLocal memory leak in JRuby 10.0.2.0
 *
 * Creates ScriptingContainer instances with proper cleanup sequence.
 * Despite following recommended cleanup, JRuby 10.0.2.0 leaks Ruby Runtime objects.
 */
public class JRubyMemoryLeakTest {

    static {
        // Disable bundler and RubyGems to isolate from system Ruby
        System.setProperty("jruby.cli.load.gemfile", "false");
        System.setProperty("jruby.cli.rubygems.enable", "false");
    }

    public static void main(String[] args) throws Exception {
        int iterations = Integer.parseInt(args.length > 0 ? args[0] : "10");

        System.out.println("JRuby ThreadLocal Memory Leak Test");
        System.out.println("===================================");
        System.out.println("JRuby Version: " + org.jruby.runtime.Constants.VERSION);
        System.out.println("Java Version: " + System.getProperty("java.version"));
        System.out.println("Iterations: " + iterations);
        System.out.println();

        printMemoryStats("BEFORE");

        for (int i = 1; i <= iterations; i++) {
            runScriptWithProperCleanup();

            if (i % 5 == 0) {
                System.gc();
                Thread.sleep(100);
                printMemoryStats("After iteration " + i);
            }
        }

        System.out.println("\n=== Final Heap Analysis ===");
        printObjectCounts();
    }

    /**
     * Executes Ruby code with proper cleanup sequence:
     * 1. unregisterCurrentThread() BEFORE tearDown
     * 2. terminate()
     * 3. tearDown(true)
     * 4. clearGlobalRuntime()
     */
    private static void runScriptWithProperCleanup() {
        ScriptingContainer sc = null;
        try {
            // Create container with SINGLETHREAD scope to avoid cross-thread issues
            sc = new ScriptingContainer(LocalContextScope.SINGLETHREAD);

            // Isolate from system Ruby environment
            sc.getEnvironment().put("GEM_HOME", "");
            sc.getEnvironment().put("GEM_PATH", "");
            sc.getEnvironment().put("BUNDLE_GEMFILE", "");

            // Execute some Ruby code (simulates real application usage)
            sc.runScriptlet("" +
                "def fibonacci(n)\n" +
                "  return n if n <= 1\n" +
                "  fibonacci(n-1) + fibonacci(n-2)\n" +
                "end\n" +
                "\n" +
                "result = fibonacci(10)\n" +
                "result\n"
            );

            // Proper cleanup sequence
            Ruby runtime = sc.getProvider().getRuntime();

            if (runtime != null) {
                // CRITICAL: Unregister current thread BEFORE tearDown
                // tearDown() replaces ThreadService, must clean up old one first
                ThreadContext currentContext = runtime.getThreadService().getCurrentContext();
                runtime.getThreadService().unregisterCurrentThread(currentContext);
            }

            // Terminate the container
            sc.terminate();

            // Force runtime teardown
            if (runtime != null) {
                runtime.tearDown(true);
            }

            // Clear global runtime reference
            Ruby.clearGlobalRuntime();

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            sc = null;
        }
    }

    private static void printMemoryStats(String label) {
        MemoryMXBean memoryBean = ManagementFactory.getMemoryMXBean();
        MemoryUsage heapUsage = memoryBean.getHeapMemoryUsage();

        System.out.printf("%s: Heap Used: %.2f MB, Committed: %.2f MB\n",
            label,
            heapUsage.getUsed() / 1024.0 / 1024.0,
            heapUsage.getCommitted() / 1024.0 / 1024.0
        );
    }

    private static void printObjectCounts() {
        try {
            String pid = ManagementFactory.getRuntimeMXBean().getName().split("@")[0];

            System.out.println("\n=== ScriptingContainer Count ===");
            ProcessBuilder pb1 = new ProcessBuilder(
                "bash", "-c",
                "jmap -histo:live " + pid + " | grep 'org.jruby.embed.ScriptingContainer' || echo 'No instances found'"
            );
            pb1.inheritIO();
            pb1.start().waitFor();

            System.out.println("\n=== Ruby Runtime Count ===");
            ProcessBuilder pb2 = new ProcessBuilder(
                "bash", "-c",
                "jmap -histo:live " + pid + " | grep -E 'org.jruby.Ruby[^a-zA-Z]|org.jruby.Ruby$' || echo 'No instances found'"
            );
            pb2.inheritIO();
            pb2.start().waitFor();

            System.out.println("\n=== Total JRuby Object Count ===");
            ProcessBuilder pb3 = new ProcessBuilder(
                "bash", "-c",
                "jmap -histo:live " + pid + " | grep 'org.jruby' | awk '{sum += $2; bytes += $3} END {print \"Total instances: \" sum \", Total bytes: \" bytes}'"
            );
            pb3.inheritIO();
            pb3.start().waitFor();

            System.out.println("\n=== Top 10 JRuby Classes ===");
            ProcessBuilder pb4 = new ProcessBuilder(
                "bash", "-c",
                "jmap -histo:live " + pid + " | grep 'org.jruby' | head -10"
            );
            pb4.inheritIO();
            pb4.start().waitFor();

        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}