Add support for PyPy

Author: messense

src/lib.rs

@@ -118,6 +118,15 @@ const DLLTOOL_MSVC: &str = "llvm-dlltool";
 #[cfg(windows)]
 const LIB_MSVC: &str = "lib.exe";
 
+/// Python interpreter implementations
+#[derive(Debug, Clone, Copy)]
+pub enum PythonImplementation {
+    /// CPython
+    CPython,
+    /// PyPy
+    PyPy,
+}
+
 /// Windows import library generator for Python
 ///
 /// Generates `python3.dll` or `pythonXY.dll` import library directly from the
@@ -159,6 +168,8 @@ pub struct ImportLibraryGenerator {
     env: String,
     /// Major and minor Python version (for `pythonXY.dll` only)
     version: Option<(u8, u8)>,
+    /// Python interpreter implementation
+    implementation: PythonImplementation,
 }
 
 impl ImportLibraryGenerator {
@@ -174,6 +185,7 @@ impl ImportLibraryGenerator {
             arch: arch.to_string(),
             env: env.to_string(),
             version: None,
+            implementation: PythonImplementation::CPython,
         }
     }
 
@@ -185,6 +197,12 @@ impl ImportLibraryGenerator {
         self
     }
 
+    /// Sets Python interpreter implementation
+    pub fn implementation(&mut self, implementation: PythonImplementation) -> &mut Self {
+        self.implementation = implementation;
+        self
+    }
+
     /// Generates the Python DLL import library in `out_dir`.
     ///
     /// The version-agnostic `python3.dll` import library is generated
@@ -224,14 +242,22 @@ impl ImportLibraryGenerator {
     ///
     /// Returns the newly created `python3.def` or `pythonXY.def` file path.
     fn write_def_file(&self, out_dir: &Path) -> Result<PathBuf> {
-        let (def_file, def_file_content) = match self.version {
-            None => ("python3.def", include_str!("python3.def")),
-            Some((3, 7)) => ("python37.def", include_str!("python37.def")),
-            Some((3, 8)) => ("python38.def", include_str!("python38.def")),
-            Some((3, 9)) => ("python39.def", include_str!("python39.def")),
-            Some((3, 10)) => ("python310.def", include_str!("python310.def")),
-            Some((3, 11)) => ("python311.def", include_str!("python311.def")),
-            _ => return Err(Error::new(ErrorKind::Other, "Unsupported Python version")),
+        let (def_file, def_file_content) = match self.implementation {
+            PythonImplementation::CPython => match self.version {
+                None => ("python3.def", include_str!("python3.def")),
+                Some((3, 7)) => ("python37.def", include_str!("python37.def")),
+                Some((3, 8)) => ("python38.def", include_str!("python38.def")),
+                Some((3, 9)) => ("python39.def", include_str!("python39.def")),
+                Some((3, 10)) => ("python310.def", include_str!("python310.def")),
+                Some((3, 11)) => ("python311.def", include_str!("python311.def")),
+                _ => return Err(Error::new(ErrorKind::Other, "Unsupported Python version")),
+            },
+            PythonImplementation::PyPy => match self.version {
+                Some((3, 7)) => ("libpypy3-c.def", include_str!("libpypy3-c.def")),
+                Some((3, 8)) => ("libpypy3-c.def", include_str!("libpypy3-c.def")),
+                Some((3, 9)) => ("libpypy3.9-c.def", include_str!("libpypy3.9-c.def")),
+                _ => return Err(Error::new(ErrorKind::Other, "Unsupported PyPy version")),
+            },
         };
 
         let mut defpath = out_dir.to_owned();
@@ -248,11 +274,19 @@ impl ImportLibraryGenerator {
     ///
     /// Returns the full import library file path under `out_dir`.
     fn implib_file_path(&self, out_dir: &Path, libext: &str) -> PathBuf {
-        let libname = match self.version {
-            Some((major, minor)) => {
-                format!("python{}{}{}", major, minor, libext)
-            }
-            None => format!("python3{}", libext),
+        let libname = match self.implementation {
+            PythonImplementation::CPython => match self.version {
+                Some((major, minor)) => {
+                    format!("python{}{}{}", major, minor, libext)
+                }
+                None => format!("python3{}", libext),
+            },
+            PythonImplementation::PyPy => match self.version {
+                Some((major, minor)) if (major, minor) >= (3, 9) => {
+                    format!("libpypy{}.{}-c{}", major, minor, libext)
+                }
+                _ => format!("libpypy3-c{}", libext),
+            },
         };
 
         let mut libpath = out_dir.to_owned();
@@ -477,6 +511,15 @@ mod tests {
                 .generate(&dir)
                 .unwrap();
         }
+
+        // PyPy
+        for minor in 7..=9 {
+            ImportLibraryGenerator::new("x86_64", "gnu")
+                .version(Some((3, minor)))
+                .implementation(PythonImplementation::PyPy)
+                .generate(&dir)
+                .unwrap();
+        }
     }
 
     #[cfg(unix)]
@@ -507,6 +550,15 @@ mod tests {
                 .generate(&dir)
                 .unwrap();
         }
+
+        // PyPy
+        for minor in 7..=9 {
+            ImportLibraryGenerator::new("x86_64", "msvc")
+                .version(Some((3, minor)))
+                .implementation(PythonImplementation::PyPy)
+                .generate(&dir)
+                .unwrap();
+        }
     }
 
     #[test]