Merge pull request #14 from mozilla-ai/pre-commit

Author: besaleli

.github/.gitkeep

@@ -0,0 +1,73 @@
+name: pre-commit
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Install dependencies
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y protobuf-compiler
+          protoc --version
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.13"
+
+      - name: Install uv
+        run: curl -LsSf https://astral.sh/uv/install.sh | sh
+
+      - name: Cache uv environments
+        uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cache/uv
+            .venv
+          key: uv-${{ runner.os }}-${{ hashFiles('uv.lock', 'pyproject.toml') }}
+          restore-keys: |
+            uv-${{ runner.os }}-
+
+      - name: Set up Rust
+        uses: actions-rust-lang/setup-rust-toolchain@v1
+
+      - name: Cache cargo
+        uses: actions/cache@v4
+        with:
+          path: |
+            ~/.cargo/bin
+            ~/.cargo/registry
+            ~/.cargo/git
+            target
+          key: cargo-${{ runner.os }}-${{ hashFiles('Cargo.lock') }}
+          restore-keys: |
+            cargo-${{ runner.os }}-
+
+      - name: Install pre-commit
+        run: pip install pre-commit detect-secrets
+
+      - name: Cache pre-commit environments
+        uses: actions/cache@v4
+        with:
+          path: ~/.cache/pre-commit
+          key: pre-commit-${{ runner.os }}-${{ hashFiles('.pre-commit-config.yaml') }}
+          restore-keys: |
+            pre-commit-${{ runner.os }}-
+
+      - name: Run project setup
+        run: make setup
+
+      - name: Generate dummy env file
+        run: uv run -m encoderbuild.utils.create_dummy_env_file > .env
+
+      - name: Run pre-commit
+        run: uv run --dev pre-commit run --all-files

.github/workflows/pre-commit.yaml

@@ -0,0 +1,21 @@
+repos:
+  - repo: local
+    hooks:
+      - id: cargo-fmt
+        name: cargo fmt
+        entry: bash -c 'cargo fmt --all -- --check'
+        language: system
+        types: [rust]
+      - id: cargo-clippy
+        name: cargo clippy
+        entry: bash -c 'cargo clippy --fix --allow-dirty -- -D warnings'
+        language: system
+        types: [rust]
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.6.9  # use the latest release tag
+    hooks:
+      - id: ruff
+        name: ruff check
+        args: [--fix]  # optional: auto-fix issues
+      - id: ruff-format
+        name: ruff format

.pre-commit-config.yaml

@@ -18,3 +18,7 @@ format:
 licenses:
 	@echo "Generating licenses..."
 	@cargo bundle-licenses --format yaml --output THIRDPARTY.yml
+
+.PHONY: pre-commit
+pre-commit:
+	@uv run pre-commit run --all-files

Makefile

@@ -81,9 +81,7 @@ pub fn get_banner() -> String {
 
     let spaces = " ".repeat(remaining_len);
 
-    format!(
-        "{BANNER}\nModel ID: {MODEL_ID}{spaces}{signature}\n"
-    )
+    format!("{BANNER}\nModel ID: {MODEL_ID}{spaces}{signature}\n")
 }
 
 pub fn get_model_id() -> String {

encoderfile/src/assets.rs

@@ -17,5 +17,6 @@ dependencies = [
 
 [dependency-groups]
 dev = [
+    "pre-commit>=4.3.0",
     "ruff>=0.14.2",
 ]