librashader-preprocess 0.10.1

use crate::{PreprocessError, SourceOutput};
use encoding_rs::{DecoderResult, WINDOWS_1252};
use librashader_common::shader_features::ShaderFeatures;
use std::fs::File;
use std::io::Read;
use std::path::{Path, PathBuf};
use std::str::Lines;

#[cfg(feature = "line_directives")]
const GL_GOOGLE_CPP_STYLE_LINE_DIRECTIVE: &str =
    "#extension GL_GOOGLE_cpp_style_line_directive : require";

const DEFINE_HAS_ORIGINALASPECT_UNIFORM: &str = "#define _HAS_ORIGINALASPECT_UNIFORMS";

const DEFINE_HAS_FRAMETIME_UNIFORMS: &str = "#define _HAS_FRAMETIME_UNIFORMS";

fn read_file(path: impl AsRef<Path>) -> Result<String, PreprocessError> {
    let path = path.as_ref();
    let mut buf = Vec::new();
    File::open(path)
        .and_then(|mut f| {
            f.read_to_end(&mut buf)?;
            Ok(())
        })
        .map_err(|e| PreprocessError::IOError(path.to_path_buf(), e))?;

    match String::from_utf8(buf) {
        Ok(s) => Ok(s),
        Err(e) => {
            let buf = e.into_bytes();
            let decoder = WINDOWS_1252.new_decoder();
            let Some(len) = decoder.max_utf8_buffer_length_without_replacement(buf.len()) else {
                return Err(PreprocessError::EncodingError(path.to_path_buf()));
            };

            let mut latin1_string = String::with_capacity(len);

            let (result, _) = WINDOWS_1252
                .new_decoder()
                .decode_to_string_without_replacement(&buf, &mut latin1_string, true);
            if result == DecoderResult::InputEmpty {
                Ok(latin1_string)
            } else {
                Err(PreprocessError::EncodingError(path.to_path_buf()))
            }
        }
    }
}

pub fn read_source(
    path: impl AsRef<Path>,
    features: ShaderFeatures,
) -> Result<String, PreprocessError> {
    let path = path.as_ref();
    let source = read_file(path)?;
    let mut output = String::new();

    let source = source.trim();
    let mut lines = source.lines();

    if let Some(header) = lines.next() {
        if !header.starts_with("#version ") {
            return Err(PreprocessError::MissingVersionHeader);
        }
        output.push_line(header);
    } else {
        return Err(PreprocessError::UnexpectedEof);
    }

    #[cfg(feature = "line_directives")]
    output.push_line(GL_GOOGLE_CPP_STYLE_LINE_DIRECTIVE);

    if features.contains(ShaderFeatures::ORIGINAL_ASPECT_UNIFORMS) {
        output.push_line(DEFINE_HAS_ORIGINALASPECT_UNIFORM);
    }

    if features.contains(ShaderFeatures::FRAMETIME_UNIFORMS) {
        output.push_line(DEFINE_HAS_FRAMETIME_UNIFORMS);
    }

    output.mark_line(2, path.file_name().and_then(|f| f.to_str()).unwrap_or(""));
    preprocess(lines, path, &mut output)?;

    Ok(output)
}

fn preprocess(
    lines: Lines,
    file_name: impl AsRef<Path>,
    output: &mut String,
) -> Result<(), PreprocessError> {
    let file_name = file_name.as_ref();
    let include_path = file_name.parent().unwrap();
    let file_name = file_name.file_name().and_then(|f| f.to_str()).unwrap_or("");

    fn include_callback(
        output: &mut String,
        source: String,
        include_path: PathBuf,
        file_name: &str,
        line_no: usize,
    ) -> Result<(), PreprocessError> {
        let source = source.trim();
        let lines = source.lines();

        let include_file = include_path
            .file_name()
            .and_then(|f| f.to_str())
            .unwrap_or("");
        output.mark_line(1, include_file);
        preprocess(lines, include_path, output)?;
        output.mark_line(line_no + 1, file_name);
        Ok(())
    }

    for (line_no, line) in lines.enumerate() {
        if let Some(mut include_file) = line.strip_prefix("#include ") {
            // Handle comments in include pragma
            // (ref: #185, https://github.com/kokoko3k/koko-aio-slang/blob/619ad630422991606bc70751f084045c62863034/shaders-ng/config.inc#L765)
            if let Some((file_path, _comment)) = include_file.split_once("//") {
                include_file = file_path;
            }

            let include_file = include_file.trim().trim_matches('"');
            if include_file.is_empty() {
                return Err(PreprocessError::UnexpectedEol(line_no));
            }

            let mut include_path = include_path.to_path_buf();
            include_path.push(include_file);

            let source = read_file(&include_path)?;
            include_callback(output, source, include_path, file_name, line_no)?;

            continue;
        }
        // RetroArch does not consider #pragma include_optional with extra spaces.
        // https://github.com/libretro/RetroArch/blob/e1b2e29d51c1ea9a4f5ba6a726ebdc7be45e662b/gfx/drivers_shader/glslang_util.c#L192
        if let Some(include_file) = line.strip_prefix("#pragma include_optional") {
            let include_file = include_file.trim().trim_matches('"');
            if include_file.is_empty() {
                return Err(PreprocessError::UnexpectedEol(line_no));
            }

            let mut include_path = include_path.to_path_buf();
            include_path.push(include_file);

            match read_file(&include_path) {
                Ok(source) => include_callback(output, source, include_path, file_name, line_no)?,
                // ioerror indicates that the file is not found.
                Err(PreprocessError::IOError(..)) => {
                    output.push_line(&format!("// include_optional not found: {include_file}"));
                    output.mark_line(line_no, file_name);
                }
                // other errors should not be ignored.
                Err(e) => return Err(e),
            }

            continue;
        }

        if line.starts_with("#endif") || line.starts_with("#pragma") {
            output.push_line(line);
            output.mark_line(line_no + 2, file_name);
            continue;
        }

        output.push_line(line)
    }
    Ok(())
}