feat: support reading mise.toml

veimox · veimox · commit d3afba65cebb · 2025-03-13T19:52:53.000+01:00
diff --git a/__tests__/data/mise.toml b/__tests__/data/mise.toml
@@ -0,0 +1,2 @@
+[tools]
+python = "3.8.0"
diff --git a/__tests__/utils.test.ts b/__tests__/utils.test.ts
@@ -130,6 +130,18 @@ describe('Version from file test', () => {
       expect(_fn(pythonVersionFilePath)).toEqual([pythonVersion]);
     }
   );
+  it.each([getVersionInputFromTomlFile, getVersionInputFromFile])(
+    'Version from mise mise.tml test',
+    async _fn => {
+      await io.mkdirP(tempDir);
+      const pythonVersionFileName = 'mise.toml';
+      const pythonVersionFilePath = path.join(tempDir, pythonVersionFileName);
+      const pythonVersion = '3.8.0';
+      const pythonVersionFileContent = `[tools]\npython = "${pythonVersion}"`;
+      fs.writeFileSync(pythonVersionFilePath, pythonVersionFileContent);
+      expect(_fn(pythonVersionFilePath)).toEqual([pythonVersion]);
+    }
+  );
   it.each([getVersionInputFromTomlFile, getVersionInputFromFile])(
     'Version undefined',
     async _fn => {
diff --git a/src/utils.ts b/src/utils.ts
@@ -235,15 +235,21 @@ export function getVersionInputFromTomlFile(versionFile: string): string[] {
   pyprojectFile = pyprojectFile.replace(/\r\n/g, '\n');
 
   const pyprojectConfig = toml.parse(pyprojectFile);
-  let keys = [];
+  let keys: string[] = [];
 
   if ('project' in pyprojectConfig) {
     // standard project metadata (PEP 621)
     keys = ['project', 'requires-python'];
-  } else {
+  } 
+  else if ('tool' in pyprojectConfig) {
     // python poetry
     keys = ['tool', 'poetry', 'dependencies', 'python'];
   }
+  else if ('tools' in pyprojectConfig){
+    // mise
+    keys =  ['tools', 'python']
+  }
+  
   const versions = [];
   const version = extractValue(pyprojectConfig, keys);
   if (version !== undefined) {
