diff options
| author | Justin M. Keyes <justinkz@gmail.com> | 2025-08-28 23:43:10 -0400 |
|---|---|---|
| committer | Justin M. Keyes <justinkz@gmail.com> | 2025-12-30 01:44:24 -0500 |
| commit | 20e77c5d886af54d1f7b6844cffc11129f579ad9 (patch) | |
| tree | e7098a90f9333fa52ae98b0bdd9d0cc82830131f /runtime/lua/vim/_core/stringbuffer.lua | |
| parent | 03377b95523324a2a1657435f12c13a493ee5360 (diff) | |
build: ship "_core/*" as bytecode (built-into Nvim binary)
Problem:
We want to encourage implementing core features in Lua instead of C, but
it's clumsy because:
- Core Lua code (built into `nvim` so it is available even if VIMRUNTIME
is missing/invalid) requires manually updating CMakeLists.txt, or
stuffing it into `_editor.lua`.
- Core Lua modules are not organized similar to C modules, `_editor.lua`
is getting too big.
Solution:
- Introduce `_core/` where core Lua code can live. All Lua modules added
there will automatically be included as bytecode in the `nvim` binary.
- Move these core modules into `_core/*`:
```
_defaults.lua
_editor.lua
_options.lua
_system.lua
shared.lua
```
TODO:
- Move `_extui/ => _core/ui2/`
Diffstat (limited to 'runtime/lua/vim/_core/stringbuffer.lua')
| -rw-r--r-- | runtime/lua/vim/_core/stringbuffer.lua | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/runtime/lua/vim/_core/stringbuffer.lua b/runtime/lua/vim/_core/stringbuffer.lua new file mode 100644 index 0000000000..0241784a9b --- /dev/null +++ b/runtime/lua/vim/_core/stringbuffer.lua @@ -0,0 +1,111 @@ +-- Basic shim for LuaJIT's stringbuffer. +-- Note this does not implement the full API. +-- This is intentionally internal-only. If we want to expose it, we should +-- reimplement this a userdata and ship it as `string.buffer` +-- (minus the FFI stuff) for Lua 5.1. +local M = {} + +local has_strbuffer, strbuffer = pcall(require, 'string.buffer') + +if has_strbuffer then + M.new = strbuffer.new + + -- Lua 5.1 does not have __len metamethod so we need to provide a len() + -- function to use instead. + + --- @param buf vim._core.stringbuffer + --- @return integer + function M.len(buf) + return #buf + end + + return M +end + +--- @class vim._core.stringbuffer +--- @field private buf string[] +--- @field package len integer absolute length of the `buf` +--- @field package skip_ptr integer +local StrBuffer = {} +StrBuffer.__index = StrBuffer + +--- @return string +function StrBuffer:tostring() + if #self.buf > 1 then + self.buf = { table.concat(self.buf) } + end + + -- assert(self.len == #(self.buf[1] or ''), 'len mismatch') + + if self.skip_ptr > 0 then + if self.buf[1] then + self.buf[1] = self.buf[1]:sub(self.skip_ptr + 1) + self.len = self.len - self.skip_ptr + end + self.skip_ptr = 0 + end + + -- assert(self.len == #(self.buf[1] or ''), 'len mismatch') + + return self.buf[1] or '' +end + +StrBuffer.__tostring = StrBuffer.tostring + +--- @private +--- Efficiently peak at the first `n` characters of the buffer. +--- @param n integer +--- @return string +function StrBuffer:_peak(n) + local skip, buf1 = self.skip_ptr, self.buf[1] + if buf1 and (n + skip) < #buf1 then + return buf1:sub(skip + 1, skip + n) + end + return self:tostring():sub(1, n) +end + +--- @param chunk string +function StrBuffer:put(chunk) + local s = tostring(chunk) + self.buf[#self.buf + 1] = s + self.len = self.len + #s + return self +end + +--- @param str string +function StrBuffer:set(str) + return self:reset():put(str) +end + +--- @param n? integer +--- @return string +function StrBuffer:get(n) + n = n or self.len + local r = self:_peak(n) + self:skip(n) + return r +end + +--- @param n integer +function StrBuffer:skip(n) + self.skip_ptr = math.min(self.len, self.skip_ptr + n) + return self +end + +function StrBuffer:reset() + self.buf = {} + self.skip_ptr = 0 + self.len = 0 + return self +end + +function M.new() + return setmetatable({}, StrBuffer):reset() +end + +--- @param buf vim._core.stringbuffer +function M.len(buf) + return buf.len - buf.skip_ptr +end + +return M |
