A plugin for esbuild that adds support for inlining functions with transpiler hints. C++ has function specifiers and now JS has esbuild-plugin-inline-functions
.
npm i esbuild-plugin-inline-functions
This plugin can be used anywhere esbuild is, for example with tsup
as I am in Koota.
// tsup.config.ts
import { defineConfig } from 'tsup';
import { inlineFunctionsPlugin } from 'esbuild-plugin-inline-functions';
export default defineConfig({
entry: ['src/index.ts'],
format: ['esm', 'cjs'],
esbuildPlugins: [inlineFunctionsPlugin()],
});
To get started, simply add the /* @inline */
hint in front of any function declaration.
export /* @inline */ function getUser(id) {
return userCache.get(id) || database.users.find(u => u.id === id);
}
export /* @inline */ function getUserName(user) {
return user?.name || 'Unknown';
}
And then any instance of that function being called will get inlined during the build step.
// Calls the inline-hinted functions
function displayUserInfo(userId) {
const greeting = `Hello, ${getUserName(getUser(userId))}!`;
const status = getUser(userId)?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
// The transpiled output
function displayUserInfo(userId) {
// getUser() and getUserName() are inlined
const getUser_result_0_$f = userCache.get(userId) || database.users.find(u => u.id === userId);
const greeting = `Hello, ${getUser_result_0_$f?.name || 'Unknown'}!`;
// Second getUser() is inlined
const getUser_result_1_$f = userCache.get(userId) || database.users.find(u => u.id === userId);
const status = getUser_result_1_$f?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
- Automatically handles imports and dependencies
- Preserves control flow like early returns and conditionals
- Eliminates function call overhead
- Works with arrow functions and regular function declarations
- Maintains correct variable scoping and execution order
You'll notice the output can have redundant variable reads. For performance-critical code, these redundant reads can get expensive. We want to access data once, but the transpiler needs confidence that values won't change between function calls. Use the /* @pure */
hint to tell the transpiler the function has no side effects.
export /* @inline @pure */ function getUser(id) {
return userCache.get(id) || database.users.find(u => u.id === id);
}
export /* @inline @pure */ function getUserName(user) {
return user?.name || 'Unknown';
}
With this knowledge the transpiler can safely hoist data reads into single calls.
// Same function calls as before
function displayUserInfo(userId) {
const greeting = `Hello, ${getUserName(getUser(userId))}!`;
const status = getUser(userId)?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
// Transpiled with @pure
function displayUserInfo(userId) {
// The duplicate map reads are hoisted
const user_0_$f = userCache.get(userId) || database.users.find(u => u.id === userId);
const greeting = `Hello, ${user_0_$f?.name || 'Unknown'}!`;
const status = user_0_$f?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
Any function call that does not have a @pure
hint will cause the transpiler to conservatively opt out of hoisting. If these functions are pure, but you don't want to inline them you can still give them the @pure
hint and the transpiler will know it is safe to hoist values from other inlined functions in the code block.
// Non-inlined but marked as pure for optimization
export /* @pure */ function formatUser(user) {
return {
displayName: user?.name || 'Unknown',
isActive: user?.active || false,
initials: (user?.name || 'U').charAt(0).toUpperCase()
};
}
function displayUserInfo(userId) {
const greeting = `Hello, ${getUserName(getUser(userId))}!`;
// Not inlined but marked pure
const formatted = formatUser(getUser(userId));
const status = formatted.isActive ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
// Transpiled with pure functions, some inlined
function displayUserInfo(userId) {
// All getUser() calls are hoisted because they're pure
const user_0_$f = userCache.get(userId) || database.users.find(u => u.id === userId);
const greeting = `Hello, ${user_0_$f?.name || 'Unknown'}!`;
// formatUser() is not inlined but marked pure, so hoisting is safe
const formatted = formatUser(user_0_$f); // Reuses hoisted value
const status = formatted.isActive ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
By marking a function with an @inline
hint, it is inlined for all calls, however you might want to inline selectively. This can be done by instead adding the hint to the function call.
function displayUserInfo(userId) {
const greeting = `Hello, ${getUserName(getUser(userId))}!`;
const status = /* @inline */ getUser(userId)?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
// Transpiled with only one getUser() inlined
function displayUserInfo(userId) {
const greeting = `Hello, ${getUserName(getUser(userId))}!`;
// Only the marked getUser() call is inlined
const getUser_result_0_$f = userCache.get(userId) || database.users.find(u => u.id === userId);
const status = getUser_result_0_$f?.active ? 'Active' : 'Inactive';
return `${greeting} Status: ${status}`;
}
Function calls have a cost. They create a new scope, allocate memory and then need to be garbage collected. If you are doing performance critical code like a video game then every millisecond matters and these papercuts can add up. Manually inlining can come with performance benefits but DX is sacrificed. The same function gets copied multiple places and it generally makes the codebase harder for people to read and contribute to. This plugin allows you to get the best of both worlds.
Sometimes! VMs do an amazing job of optimizing code but ultimately the VM only knows your app one code block at a time and attempts to discover optimizations with built in heuristics. We have the advantage of knowing our own codebase and its hot paths ahead of time so we can help the VM out. We do this by transpiling code to be pre-optimized for the VM. This means instead of hoping the VM's heuristics decide a function is safe to inline, we do it ourselves and the VM has fewer decisions to make.
This is an experiment I created while I was at the Recurse Center that turned out more successful than I expected. I have since integrated it into Koota where its features are being put to the test. I plan to extract this out into a Babel plugin and make it available for dev as well as production with a Vite plugin. I may look into add other hints oriented at optimizing performance.