From ceb6dbf76688d1d25e65f1d03d1c5589a5665228 Mon Sep 17 00:00:00 2001
From: messense <messense@icloud.com>
Date: Wed, 9 Jun 2021 08:58:18 +0800
Subject: [PATCH] Detect Python implementation from `SOABI` on Unix

---
 pyo3-build-config/src/impl_.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/pyo3-build-config/src/impl_.rs b/pyo3-build-config/src/impl_.rs
index 9146a55f..c94aa466 100644
--- a/pyo3-build-config/src/impl_.rs
+++ b/pyo3-build-config/src/impl_.rs
@@ -358,6 +358,7 @@ fn parse_sysconfigdata(config_path: impl AsRef<Path>) -> Result<HashMap<String,
     script += r#"
 print("version_major", build_time_vars["VERSION"][0])  # 3
 print("version_minor", build_time_vars["VERSION"][2])  # E.g., 8
+print("SOABI", build_time_vars.get("SOABI", ""))
 KEYS = [
     "WITH_THREAD",
     "Py_DEBUG",
@@ -524,9 +525,19 @@ fn load_cross_compile_from_sysconfigdata(
         None => format!("{}.{}", major, minor),
     };
     let calcsize_pointer = sysconfig_data.get_numeric("SIZEOF_VOID_P").ok();
+    let soabi = match sysconfig_data.get("SOABI") {
+        Some(s) => s,
+        None => bail!("sysconfigdata did not define SOABI"),
+    };
 
     let version = PythonVersion { major, minor };
-    let implementation = PythonImplementation::CPython;
+    let implementation = if soabi.starts_with("pypy") {
+        PythonImplementation::PyPy
+    } else if soabi.starts_with("cpython") {
+        PythonImplementation::CPython
+    } else {
+        bail!("unsupported Python interpreter");
+    };
 
     Ok(InterpreterConfig {
         version,