package org.schabi.newpipe.player;

import android.content.Context;
import android.content.ContextWrapper;

/**
 * Fixes a leak caused by AudioManager using an Activity context.
 * Tracked at https://android-review.googlesource.com/#/c/140481/1 and
 * https://github.com/square/leakcanary/issues/205
 * Source:
 * https://gist.github.com/jankovd/891d96f476f7a9ce24e2
 */
public class AudioServiceLeakFix extends ContextWrapper {
    AudioServiceLeakFix(final Context base) {
        super(base);
    }

    public static ContextWrapper preventLeakOf(final Context base) {
        return new AudioServiceLeakFix(base);
    }

    @Override
    public Object getSystemService(final String name) {
        if (Context.AUDIO_SERVICE.equals(name)) {
            return getApplicationContext().getSystemService(name);
        }
        return super.getSystemService(name);
    }
}