From 43d2745cbec3c938b35d01e1d6df2c5dd2c304eb Mon Sep 17 00:00:00 2001
From: Jakob Odersky <jakob@odersky.com>
Date: Wed, 4 Dec 2024 11:56:21 +0100
Subject: [PATCH] wip

---
 .../src/mill/pythonlib/PythonModule.scala     | 58 ++++++++++++++++++-
 1 file changed, 57 insertions(+), 1 deletion(-)

diff --git a/pythonlib/src/mill/pythonlib/PythonModule.scala b/pythonlib/src/mill/pythonlib/PythonModule.scala
index 219673a3254..133ce7efc34 100644
--- a/pythonlib/src/mill/pythonlib/PythonModule.scala
+++ b/pythonlib/src/mill/pythonlib/PythonModule.scala
@@ -66,7 +66,7 @@ trait PythonModule extends PipModule with TaskModule { outer =>
   def mainScript: T[PathRef] = Task.Source { millSourcePath / "src" / "main.py" }
 
   override def pythonToolDeps: T[Seq[String]] = Task {
-    super.pythonToolDeps() ++ Seq("mypy==1.13.0", "pex==2.24.1")
+    super.pythonToolDeps() ++ Seq("mypy==1.13.0", "pex==2.24.1", "ruff>=0.8.1")
   }
 
   /**
@@ -180,6 +180,62 @@ trait PythonModule extends PipModule with TaskModule { outer =>
     override def moduleDeps: Seq[PythonModule] = Seq(outer)
   }
 
+  // def getRuff(): PathRef = Task {
+  //   import coursier.cache.FileCache
+  //   import coursier.util.Artifact
+
+  //   val cache = FileCache()
+  //   val file = cache
+  //     .file(Artifact(""))
+  //     .run
+  //     .unsafeRun()(cache.ec)
+  //     .fold(ex => throw new Exception(ex), identity)
+  //   os.unzip(os.Path(file), T.dest)
+  //   PathRef(T.dest)
+  // }
+
+  /** Command line options to pass the the black code formatter.
+   *
+   * This is the way to configure black in mill, since mill doesn't use a
+   * pyproject.toml file.
+   */
+  def blackOptions: T[Seq[String]] = Task { Seq.empty[String] }
+
+  /** Reformat all source files of this module. */
+  def reformat(): Command[Unit] = Task.Command {
+    runner().run(
+      // format: off
+      (
+        "-m", "ruff",
+        "format",
+        blackOptions(),
+        sources().map(_.path)
+      ),
+      // format: on
+      workingDir = T.dest
+    )
+  }
+
+  /** Check the format of all source files of this module. */
+  def checkFormat(): Command[Unit] = Task.Command {
+    runner().run(
+      // format: off
+      (
+        "-m", "ruff",
+        "format",
+        "--check",
+        blackOptions(),
+        sources().map(_.path)
+      ),
+      // format: on
+      workingDir = T.dest
+    )
+  }
+
+  def fix(): Command[Unit] = ???
+
+  def check(): Command[Unit] = ???
+
 }
 
 object PythonModule {