| Riley Bruins | fc533c9 | 2024-07-11 22:50:36 +0200 | [diff] [blame] | 1 | " Vim filetype plugin |
| 2 | " Language: Mojo |
| 3 | " Maintainer: Riley Bruins <ribru17@gmail.com> |
| 4 | " Last Change: 2024 Jul 07 |
| 5 | |
| 6 | if exists('b:did_ftplugin') |
| 7 | finish |
| 8 | endif |
| 9 | let b:did_ftplugin = 1 |
| 10 | |
| 11 | setlocal include=^\\s*\\(from\\\|import\\) |
| 12 | setlocal define=^\\s*\\(\\(async\\s\\+\\)\\?def\\\|class\\) |
| 13 | |
| 14 | " For imports with leading .., append / and replace additional .s with ../ |
| 15 | let b:grandparent_match = '^\(.\.\)\(\.*\)' |
| 16 | let b:grandparent_sub = '\=submatch(1)."/".repeat("../",strlen(submatch(2)))' |
| 17 | |
| 18 | " For imports with a single leading ., replace it with ./ |
| 19 | let b:parent_match = '^\.\(\.\)\@!' |
| 20 | let b:parent_sub = './' |
| 21 | |
| 22 | " Replace any . sandwiched between word characters with / |
| 23 | let b:child_match = '\(\w\)\.\(\w\)' |
| 24 | let b:child_sub = '\1/\2' |
| 25 | |
| 26 | setlocal includeexpr=substitute(substitute(substitute( |
| 27 | \v:fname, |
| 28 | \b:grandparent_match,b:grandparent_sub,''), |
| 29 | \b:parent_match,b:parent_sub,''), |
| 30 | \b:child_match,b:child_sub,'g') |
| 31 | |
| 32 | setlocal suffixesadd=.mojo |
| 33 | setlocal comments=b:#,fb:- |
| 34 | setlocal commentstring=#\ %s |
| 35 | |
| 36 | let b:undo_ftplugin = 'setlocal include<' |
| 37 | \ . '|setlocal define<' |
| 38 | \ . '|setlocal includeexpr<' |
| 39 | \ . '|setlocal suffixesadd<' |
| 40 | \ . '|setlocal comments<' |
| 41 | \ . '|setlocal commentstring<' |