Split derive-impl and derive

Author: coolreader18

Cargo.lock

@@ -15,7 +15,7 @@ include = ["LICENSE", "Cargo.toml", "src/**/*.rs"]
 resolver = "2"
 members = [
     "compiler", "compiler/ast", "compiler/core", "compiler/codegen", "compiler/parser",
-    ".", "common", "derive", "jit", "vm", "pylib", "stdlib", "wasm/lib",
+    ".", "common", "derive", "jit", "vm", "pylib", "stdlib", "wasm/lib", "derive-impl",
 ]
 
 [features]

Cargo.toml

@@ -1,12 +1,12 @@
 use rustpython_codegen::{compile, symboltable};
-use rustpython_compiler_core::CodeObject;
 use rustpython_parser::{
     ast::{fold::Fold, ConstantOptimizer},
     error::ParseErrorType,
     parser,
 };
 
 pub use rustpython_codegen::compile::CompileOpts;
+pub use rustpython_compiler_core::CodeObject;
 pub use rustpython_compiler_core::{BaseError as CompileErrorBody, Mode};
 
 #[derive(Debug, thiserror::Error)]

compiler/src/lib.rs

@@ -0,0 +1,18 @@
+[package]
+name = "rustpython-derive-impl"
+version = "0.0.0"
+edition = "2021"
+
+[dependencies]
+rustpython-compiler-core = { path = "../compiler/core", version = "0.1.1" }
+rustpython-doc = { git = "https://github.com/RustPython/__doc__", branch = "main" }
+
+indexmap = "1.8.1"
+itertools = "0.10.3"
+maplit = "1.0.2"
+once_cell = "1.10.0"
+proc-macro2 = "1.0.37"
+quote = "1.0.18"
+syn = { version = "1.0.91", features = ["full", "extra-traits"] }
+syn-ext = { version = "0.4.0", features = ["full"] }
+textwrap = { version = "0.15.0", default-features = false }

derive-impl/Cargo.toml

@@ -0,0 +1,79 @@
+#![recursion_limit = "128"]
+#![doc(html_logo_url = "https://raw.githubusercontent.com/RustPython/RustPython/main/logo.png")]
+#![doc(html_root_url = "https://docs.rs/rustpython-derive/")]
+
+extern crate proc_macro;
+
+#[macro_use]
+extern crate maplit;
+
+#[macro_use]
+mod error;
+#[macro_use]
+mod util;
+
+mod compile_bytecode;
+mod from_args;
+mod pyclass;
+mod pymodule;
+mod pypayload;
+mod pystructseq;
+
+use error::{extract_spans, Diagnostic};
+use proc_macro2::TokenStream;
+use quote::ToTokens;
+use rustpython_doc as doc;
+use syn::{AttributeArgs, DeriveInput, Item};
+
+pub use compile_bytecode::Compiler;
+
+fn result_to_tokens(result: Result<TokenStream, impl Into<Diagnostic>>) -> TokenStream {
+    result
+        .map_err(|e| e.into())
+        .unwrap_or_else(ToTokens::into_token_stream)
+}
+
+pub fn derive_from_args(input: DeriveInput) -> TokenStream {
+    result_to_tokens(from_args::impl_from_args(input))
+}
+
+pub fn pyclass(attr: AttributeArgs, item: Item) -> TokenStream {
+    if matches!(item, syn::Item::Impl(_) | syn::Item::Trait(_)) {
+        result_to_tokens(pyclass::impl_pyimpl(attr, item))
+    } else {
+        result_to_tokens(pyclass::impl_pyclass(attr, item))
+    }
+}
+
+pub use pyclass::PyExceptionDef;
+pub fn define_exception(exc_def: PyExceptionDef) -> TokenStream {
+    result_to_tokens(pyclass::impl_define_exception(exc_def))
+}
+
+pub fn pyexception(attr: AttributeArgs, item: Item) -> TokenStream {
+    result_to_tokens(pyclass::impl_pyexception(attr, item))
+}
+
+pub fn pymodule(attr: AttributeArgs, item: Item) -> TokenStream {
+    result_to_tokens(pymodule::impl_pymodule(attr, item))
+}
+
+pub fn pystruct_sequence(input: DeriveInput) -> TokenStream {
+    result_to_tokens(pystructseq::impl_pystruct_sequence(input))
+}
+
+pub fn pystruct_sequence_try_from_object(input: DeriveInput) -> TokenStream {
+    result_to_tokens(pystructseq::impl_pystruct_sequence_try_from_object(input))
+}
+
+pub fn py_compile(input: TokenStream, compiler: &dyn Compiler) -> TokenStream {
+    result_to_tokens(compile_bytecode::impl_py_compile(input, compiler))
+}
+
+pub fn py_freeze(input: TokenStream, compiler: &dyn Compiler) -> TokenStream {
+    result_to_tokens(compile_bytecode::impl_py_freeze(input, compiler))
+}
+
+pub fn pypayload(input: DeriveInput) -> TokenStream {
+    result_to_tokens(pypayload::impl_pypayload(input))
+}

derive/src/compile_bytecode.rs derive-impl/src/compile_bytecode.rsderive/src/compile_bytecode.rs renamed to derive-impl/src/compile_bytecode.rs

@@ -1284,7 +1284,7 @@ where
 }
 
 #[derive(Debug)]
-pub(crate) struct PyExceptionDef {
+pub struct PyExceptionDef {
     pub class_name: Ident,
     pub base_class: Ident,
     pub ctx_name: Ident,