workflows and things

Author: TheTechromancer

.github/dependabot.yml

@@ -0,0 +1,21 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: "uv"
+    directory: "/"
+    schedule:
+      interval: "weekly"
+    target-branch: "master"
+    open-pull-requests-limit: 10
+  - package-ecosystem: github-actions
+    directory: /
+    groups:
+      github-actions:
+        patterns:
+          - "*"  # Group all Actions updates into a single larger pull request
+    schedule:
+      interval: weekly

.github/workflows/python-tests.yml

@@ -0,0 +1,36 @@
+name: Python Tests
+
+on:
+  push:
+    branches: [ master ]
+  pull_request:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    
+    steps:
+    - uses: actions/checkout@v6
+    
+    - name: Set up Python
+      uses: actions/setup-python@v6
+      with:
+        python-version: 3.x
+    
+    - name: Set up uv
+      uses: astral-sh/setup-uv@v7
+    
+    - name: Install Rust toolchain
+      uses: actions-rust-lang/setup-rust-toolchain@v1
+
+    - name: Install dependencies
+      run: |
+        uv sync
+    
+    - name: Build Python bindings
+      run: |
+        maturin develop --release --features pyo3
+    
+    - name: Run Python tests
+      run: |
+        uv run pytest test_cloudcheck.py -v

Cargo.lock

@@ -9,6 +9,12 @@ serde = { version = "1.0", features = ["derive"] }
 serde_json = "1.0"
 tokio = { version = "1", features = ["full"] }
 reqwest = { version = "0.12", features = ["json"] }
+pyo3 = { version = "0.27", features = ["auto-initialize"], optional = true }
+pyo3-async-runtimes = { version = "0.27", features = ["tokio-runtime"], optional = true }
+
+[features]
+default = ["pyo3"]
+pyo3 = ["dep:pyo3", "dep:pyo3-async-runtimes"]
 
 [[bin]]
 name = "cloudcheck"
@@ -17,4 +23,4 @@ path = "src/main.rs"
 [lib]
 name = "cloudcheck"
 path = "src/lib.rs"
-
+crate-type = ["cdylib", "rlib"]

Cargo.toml

@@ -0,0 +1,4 @@
+from .cloudcheck import CloudCheck
+
+__all__ = ["CloudCheck"]
+

cloudcheck/__init__.py

@@ -1,19 +1,24 @@
+[build-system]
+requires = ["maturin>=1.0,<2.0"]
+build-backend = "maturin"
+
 [project]
 name = "cloudcheck"
 version = "8.1.0rc1"
 description = "Detailed database of cloud providers. Instantly look up a domain or IP address"
 readme = "README.md"
 requires-python = ">=3.9"
-dependencies = [
-    "requests (>=2.32.5,<3.0.0)",
-]
-
-[tool.poetry.scripts]
-cloudcheck-update = 'cloudcheck_update.cli:main'
+dependencies = []
 
 [dependency-groups]
 dev = [
+    "maturin>=1.10.2",
     "pydantic>=2.12.5",
     "pytest>=8.4.2",
+    "pytest-asyncio>=0.24.0",
     "ruff>=0.14.8",
 ]
+
+[tool.maturin]
+features = ["pyo3"]
+module-name = "cloudcheck"

pyproject.toml

@@ -6,6 +6,9 @@ use std::sync::Arc;
 use std::time::{Duration, SystemTime};
 use tokio::sync::OnceCell;
 
+#[cfg(feature = "pyo3")]
+mod python;
+
 const CLOUDCHECK_SIGNATURE_URL: &str = "https://raw.githubusercontent.com/blacklanternsecurity/cloudcheck/refs/heads/cloudcheck-v8/cloud_providers_v2.json";
 
 #[derive(Debug, Clone, Serialize, Deserialize)]
@@ -22,6 +25,7 @@ struct ProviderData {
     domains: Vec<String>,
 }
 
+#[derive(Clone)]
 pub struct CloudCheck {
     radix: Arc<OnceCell<RadixTarget>>,
     providers: Arc<OnceCell<HashMap<String, Vec<CloudProvider>>>>,
@@ -188,3 +192,4 @@ mod tests {
         );
     }
 }
+

src/lib.rs

@@ -0,0 +1,49 @@
+use pyo3::prelude::*;
+use pyo3::types::PyDict;
+use crate::CloudCheck as RustCloudCheck;
+
+#[pymodule]
+fn cloudcheck(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_class::<CloudCheck>()?;
+    Ok(())
+}
+
+#[pyclass(name = "CloudCheck")]
+pub struct CloudCheck {
+    inner: RustCloudCheck,
+}
+
+#[pymethods]
+impl CloudCheck {
+    #[new]
+    fn new() -> Self {
+        CloudCheck {
+            inner: RustCloudCheck::new(),
+        }
+    }
+
+    fn lookup<'py>(&self, py: Python<'py>, target: &str) -> PyResult<Bound<'py, PyAny>> {
+        let inner = self.inner.clone();
+        let target = target.to_string();
+        pyo3_async_runtimes::tokio::future_into_py(py, async move {
+            match inner.lookup(&target).await {
+                Ok(providers) => {
+                    Python::attach(|py| -> PyResult<Vec<Py<PyAny>>> {
+                        let mut result = Vec::new();
+                        for provider in providers {
+                            let dict = PyDict::new(py);
+                            dict.set_item("name", provider.name)?;
+                            dict.set_item("tags", provider.tags)?;
+                            result.push(dict.unbind().into());
+                        }
+                        Ok(result)
+                    })
+                }
+                Err(e) => Err(PyErr::new::<pyo3::exceptions::PyRuntimeError, _>(
+                    format!("CloudCheck error: {}", e),
+                )),
+            }
+        })
+    }
+}
+

src/python.rs

@@ -0,0 +1,19 @@
+import pytest
+from cloudcheck import CloudCheck
+
+
+@pytest.mark.asyncio
+async def test_lookup_google_dns():
+    cloudcheck = CloudCheck()
+    results = await cloudcheck.lookup("8.8.8.8")
+    names = [provider["name"] for provider in results]
+    assert "Google" in names, f"Expected Google in results: {names}"
+
+
+@pytest.mark.asyncio
+async def test_lookup_amazon_domain():
+    cloudcheck = CloudCheck()
+    results = await cloudcheck.lookup("asdf.amazon.com")
+    names = [provider["name"] for provider in results]
+    assert "Amazon" in names, f"Expected Amazon in results: {names}"
+