/********************************************************************************
 * Copyright (c) 2022 Contributors to the Eclipse Foundation
 *
 * See the NOTICE file(s) distributed with this work for additional
 * information regarding copyright ownership.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Apache License, Version 2.0 which is available at
 * https://www.apache.org/licenses/LICENSE-2.0.
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 ********************************************************************************/
package org.eclipse.emt4j.agent.common;

import org.eclipse.emt4j.agent.common.file.Recorder;

import java.security.CodeSource;
import java.security.ProtectionDomain;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class LoadedJarRecorder {

    private Recorder recorder;
    private Map<CodeSource, Boolean> alreadyRecordCodeSource = new ConcurrentHashMap<>();

    public LoadedJarRecorder(Recorder recorder) {
        this.recorder = recorder;
    }

    public void recordJar(ProtectionDomain protectionDomain) throws InterruptedException {
        if (alreadyRecordCodeSource.containsKey(protectionDomain.getCodeSource())) {
            return;
        }
        alreadyRecordCodeSource.put(protectionDomain.getCodeSource(), true);
        recorder.record(DependencyBuilder.buildCodeSource(protectionDomain.getCodeSource().getLocation()));
    }
}
