<?php

function vsCodeGetTranslationsFromFile($file, $path, $namespace)
{
    $key = pathinfo($file, PATHINFO_FILENAME);

    if ($namespace) {
        $key = "{$namespace}::{$key}";
    }

    $lang = collect(explode(DIRECTORY_SEPARATOR, str_replace($path, '', $file)))
        ->filter()
        ->first();

    $fileLines = Illuminate\Support\Facades\File::lines($file);
    $lines = [];
    $inComment = false;

    foreach ($fileLines as $index => $line) {
        $trimmed = trim($line);

        if (substr($trimmed, 0, 2) === '/*') {
            $inComment = true;
            continue;
        }

        if ($inComment) {
            if (substr($trimmed, -2) !== '*/') {
                continue;
            }

            $inComment = false;
        }

        if (substr($trimmed, 0, 2) === '//') {
            continue;
        }

        $lines[] = [$index + 1, $trimmed];
    }

    return [
        'k' => $key,
        'la' => $lang,
        'vs' => collect(Illuminate\Support\Arr::dot((Illuminate\Support\Arr::wrap(__($key, [], $lang)))))
            ->map(
                fn ($value, $key) => vsCodeTranslationValue(
                    $key,
                    $value,
                    str_replace(base_path(DIRECTORY_SEPARATOR), '', $file),
                    $lines
                )
            )
            ->filter(),
    ];
}

function vsCodeTranslationValue($key, $value, $file, $lines): ?array
{
    if (is_array($value)) {
        return null;
    }

    $lineNumber = 1;
    $keys = explode('.', $key);
    $index = 0;
    $currentKey = array_shift($keys);

    foreach ($lines as $index => $line) {
        if (
            strpos($line[1], '"' . $currentKey . '"', 0) !== false ||
            strpos($line[1], "'" . $currentKey . "'", 0) !== false
        ) {
            $lineNumber = $line[0];
            $currentKey = array_shift($keys);
        }

        if ($currentKey === null) {
            break;
        }
    }

    return [
        'v' => $value,
        'p' => $file,
        'li' => $lineNumber,
        'pa' => preg_match_all("/\:([A-Za-z0-9_]+)/", $value, $matches)
            ? $matches[1]
            : [],
    ];
}

function vscodeCollectTranslations(string $path, ?string $namespace = null)
{
    $realPath = realpath($path);

    if (!is_dir($realPath)) {
        return collect();
    }

    return collect(Illuminate\Support\Facades\File::allFiles($realPath))->map(
        fn ($file) => vsCodeGetTranslationsFromFile($file, $path, $namespace)
    );
}

$loader = app('translator')->getLoader();
$namespaces = $loader->namespaces();

$reflection = new ReflectionClass($loader);
$property = $reflection->hasProperty('paths')
    ? $reflection->getProperty('paths')
    : $reflection->getProperty('path');
$property->setAccessible(true);

$paths = Illuminate\Support\Arr::wrap($property->getValue($loader));

$default = collect($paths)->flatMap(
    fn ($path) => vscodeCollectTranslations($path)
);

$namespaced = collect($namespaces)->flatMap(
    fn ($path, $namespace) => vscodeCollectTranslations($path, $namespace)
);

$final = [];

foreach ($default->merge($namespaced) as $value) {
    foreach ($value['vs'] as $key => $v) {
        $dotKey = "{$value['k']}.{$key}";

        if (!isset($final[$dotKey])) {
            $final[$dotKey] = [];
        }

        $final[$dotKey][$value['la']] = $v;

        if ($value['la'] === Illuminate\Support\Facades\App::currentLocale()) {
            $final[$dotKey]['default'] = $v;
        }
    }
}

return collect($final);