DuckDB Sqruff Extension

This is a simple wrapper around the sqruff Rust library for formatting SQL, using the DuckDB Rust extension template.

The project is currently very barebones since it's a proof of concept, and requires building locally.

Usage currently

$ duckdb -unsigned
D INSTALL '/path/to/duckdb_sqruff.duckdb_extension';
D LOAD duckdb_sqruff;

-- Set ascii mode, as duckbox does not render newlines
D .mode ascii
D FROM duckdb_sqruff('WITH cte AS (SELECT a, b, c from some_table) select * from cte where a > 5;');
column0
WITH cte AS (SELECT a, b, c FROM some_table)

SELECT * FROM cte WHERE a > 5;

Ideal usage (not yet implemented)

-- The ~ at the front triggers a parser error, which this extension checks for and formats the query instead
D ~WITH cte AS (SELECT a, b, c from some_table) select * from cte where a > 5;
column0
WITH cte AS (SELECT a, b, c FROM some_table)

SELECT * FROM cte WHERE a > 5;

Cloning

Clone the repo with submodules

git clone --recurse-submodules <repo>

Dependencies

In principle, these extensions can be compiled with the Rust toolchain alone. However, this template relies on some additional tooling to make life a little easier and to be able to share CI/CD infrastructure with extension templates for other languages:

• Python3
• Python3-venv
• Make
• Git

Installing these dependencies will vary per platform:

• For Linux, these come generally pre-installed or are available through the distro-specific package manager.
• For MacOS, homebrew.
• For Windows, chocolatey.

Building

After installing the dependencies, building is a two-step process. Firstly run:

make configure

This will ensure a Python venv is set up with DuckDB and DuckDB's test runner installed. Additionally, depending on configuration, DuckDB will be used to determine the correct platform for which you are compiling.

Then, to build the extension run:

make debug

This delegates the build process to cargo, which will produce a shared library in target/debug/<shared_lib_name>. After this step, a script is run to transform the shared library into a loadable extension by appending a binary footer. The resulting extension is written to the build/debug directory.

To create optimized release binaries, simply run make release instead.

Testing

This extension uses the DuckDB Python client for testing. This should be automatically installed in the make configure step. The tests themselves are written in the SQLLogicTest format, just like most of DuckDB's tests. A sample test can be found in test/sql/<extension_name>.test. To run the tests using the debug build:

make test_debug

or for the release build:

make test_release

Version switching

Testing with different DuckDB versions is really simple:

First, run

make clean_all

to ensure the previous make configure step is deleted.

Then, run

DUCKDB_TEST_VERSION=v1.2.1 make configure

to select a different duckdb version to test with

Finally, build and test with

make debug
make test_debug

Known issues

This is a bit of a footgun, but the extensions produced by this template may (or may not) be broken on windows on python3.11 with the following error on extension load:

IO Error: Extension '<name>.duckdb_extension' could not be loaded: The specified module could not be found

This was resolved by using python 3.12