inline-fn 0.0.1

Inline the content of a function without duplicating code. Useful when duplicating code is desired but you are too embarrassed to do it manually.
Documentation 
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140

#[cfg(test)]
mod tests;

pub fn extract_function(code: &str, function_name: &str) -> Option<String> {
    let mut _in_function = false;
    let mut brace_count = 0;
    let mut _function_start = 0;
    let mut function_end = 0;
    let mut in_string = false;
    let mut in_char = false;
    let mut in_line_comment = false;
    let mut in_block_comment = false;
    let mut prev_char = '\0';
    let mut escaped = false;

    let function_pattern = format!(
        r"(?:\s|^)(?:pub\s+)?(?:\(.*\)\s+)?(?:async\s+)?(?:const\s+)?(?:unsafe\s+)?fn\s+{}\s*(?:<.*?>)?\s*\(",
        function_name
    );
    let re = regex::Regex::new(&function_pattern).ok()?;

    // Find the start of the function
    if let Some(m) = re.find(code) {
        _function_start = m.start();

        // Find where the function body starts (after opening brace)
        let function_def_part = &code[_function_start..];
        let mut paren_count = 0;
        let mut found_opening_brace = false;

        for (i, c) in function_def_part.chars().enumerate() {
            if in_line_comment {
                if c == '\n' {
                    in_line_comment = false;
                }
                continue;
            } else if in_block_comment {
                if prev_char == '*' && c == '/' {
                    in_block_comment = false;
                }
            } else if in_string {
                if c == '"' && !escaped {
                    in_string = false;
                }
                escaped = c == '\\' && !escaped;
            } else if in_char {
                if c == '\'' && !escaped {
                    in_char = false;
                }
                escaped = c == '\\' && !escaped;
            } else if c == '/' && prev_char == '/' {
                in_line_comment = true;
            } else if c == '*' && prev_char == '/' {
                in_block_comment = true;
            } else if c == '"' {
                in_string = true;
                escaped = false;
            } else if c == '\'' {
                in_char = true;
                escaped = false;
            } else if c == '(' {
                paren_count += 1;
            } else if c == ')' {
                paren_count -= 1;
            } else if c == '{' && paren_count == 0 {
                brace_count = 1;
                _function_start += i;
                found_opening_brace = true;
                break;
            }

            prev_char = c;
        }

        if !found_opening_brace {
            return None;
        }

        // Find the end of the function (matching closing brace)
        let remaining_code = &code[_function_start + 1..];
        prev_char = '\0';
        in_string = false;
        in_char = false;
        in_line_comment = false;
        in_block_comment = false;
        escaped = false;

        for (i, c) in remaining_code.chars().enumerate() {
            if in_line_comment {
                if c == '\n' {
                    in_line_comment = false;
                }
            } else if in_block_comment {
                if prev_char == '*' && c == '/' {
                    in_block_comment = false;
                }
            } else if in_string {
                if c == '"' && !escaped {
                    in_string = false;
                }
                escaped = c == '\\' && !escaped;
            } else if in_char {
                if c == '\'' && !escaped {
                    in_char = false;
                }
                escaped = c == '\\' && !escaped;
            } else if c == '/' && prev_char == '/' {
                in_line_comment = true;
            } else if c == '*' && prev_char == '/' {
                in_block_comment = true;
            } else if c == '"' {
                in_string = true;
                escaped = false;
            } else if c == '\'' {
                in_char = true;
                escaped = false;
            } else if c == '{' {
                brace_count += 1;
            } else if c == '}' {
                brace_count -= 1;
                if brace_count == 0 {
                    function_end = _function_start + i + 2; // +2 to include the closing brace
                    break;
                }
            }

            prev_char = c;
        }

        if function_end > _function_start {
            return Some(
                code[_function_start + 1..function_end - 1]
                    .trim()
                    .to_string(),
            );
        }
    }

    None
}