1 files changed, 90 insertions, 0 deletions
diff --git a/src/librustdoc/plugins.rs b/src/librustdoc/plugins.rs
new file mode 100644
index 00000000000..4bad87088a1
--- /dev/null
+++ b/src/librustdoc/plugins.rs
@@ -0,0 +1,90 @@
+// Copyright 2012-2013 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+use clean;
+
+use extra;
+use dl = std::unstable::dynamic_lib;
+
+pub type PluginJson = Option<(~str, extra::json::Json)>;
+pub type PluginResult = (clean::Crate, PluginJson);
+pub type plugin_callback = extern fn (clean::Crate) -> PluginResult;
+
+/// Manages loading and running of plugins
+pub struct PluginManager {
+    priv dylibs: ~[dl::DynamicLibrary],
+    priv callbacks: ~[plugin_callback],
+    /// The directory plugins will be loaded from
+    prefix: Path,
+}
+
+impl PluginManager {
+    /// Create a new plugin manager
+    pub fn new(prefix: Path) -> PluginManager {
+        PluginManager {
+            dylibs: ~[],
+            callbacks: ~[],
+            prefix: prefix,
+        }
+    }
+
+    /// Load a plugin with the given name.
+    ///
+    /// Turns `name` into the proper dynamic library filename for the given
+    /// platform. On windows, it turns into name.dll, on OS X, name.dylib, and
+    /// elsewhere, libname.so.
+    pub fn load_plugin(&mut self, name: ~str) {
+        let x = self.prefix.push(libname(name));
+        let lib_result = dl::DynamicLibrary::open(Some(&x));
+        let lib = lib_result.unwrap();
+        let plugin = unsafe { lib.symbol("rustdoc_plugin_entrypoint") }.unwrap();
+        self.dylibs.push(lib);
+        self.callbacks.push(plugin);
+    }
+
+    /// Load a normal Rust function as a plugin.
+    ///
+    /// This is to run passes over the cleaned crate. Plugins run this way
+    /// correspond to the A-aux tag on Github.
+    pub fn add_plugin(&mut self, plugin: plugin_callback) {
+        self.callbacks.push(plugin);
+    }
+    /// Run all the loaded plugins over the crate, returning their results
+    pub fn run_plugins(&self, crate: clean::Crate) -> (clean::Crate, ~[PluginJson]) {
+        let mut out_json = ~[];
+        let mut crate = crate;
+        for &callback in self.callbacks.iter() {
+            let (c, res) = callback(crate);
+            crate = c;
+            out_json.push(res);
+        }
+        (crate, out_json)
+    }
+}
+
+#[cfg(target_os="win32")]
+fn libname(mut n: ~str) -> ~str {
+    n.push_str(".dll");
+    n
+}
+
+#[cfg(target_os="macos")]
+fn libname(mut n: ~str) -> ~str {
+    n.push_str(".dylib");
+    n
+}
+
+#[cfg(not(target_os="win32"), not(target_os="macos"))]
+fn libname(n: ~str) -> ~str {
+    let mut i = ~"lib";
+    i.push_str(n);
+    i.push_str(".so");
+    i
+}