aboutsummaryrefslogtreecommitdiff
path: root/spark-fabric/src/main/java/me/lucko/spark/fabric/smap/SourceMapProvider.java
blob: 1a4f246139b35b82549d904ab19f94bc19f13f00 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/*
 * This file is part of spark.
 *
 *  Copyright (c) lucko (Luck) <luck@lucko.me>
 *  Copyright (c) contributors
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package me.lucko.spark.fabric.smap;

import org.jetbrains.annotations.Nullable;

import java.util.HashMap;
import java.util.Map;

public class SourceMapProvider {
    private final Map<String, SourceMap> cache = new HashMap<>();

    public @Nullable SourceMap getSourceMap(String className) {
        if (this.cache.containsKey(className)) {
            return this.cache.get(className);
        }

        SourceMap smap = null;
        try {
            String value = SourceDebugCache.getSourceDebugInfo(className);
            if (value != null) {
                value = value.replaceAll("\r\n?", "\n");
                if (value.startsWith("SMAP\n")) {
                    smap = new SourceMap(value);
                }
            }
        } catch (Exception e) {
            // ignore
        }

        this.cache.put(className, smap);
        return smap;
    }

}