Skip to content

Commit

Permalink
Merge pull request #8791 from JuliaLang/omm/doc-system
Browse files Browse the repository at this point in the history
Documentation System
  • Loading branch information
MikeInnes committed Nov 10, 2014
2 parents be21b7c + 6ecd0cd commit d43dfa8
Show file tree
Hide file tree
Showing 20 changed files with 1,657 additions and 2 deletions.
316 changes: 316 additions & 0 deletions base/docs.jl
Original file line number Diff line number Diff line change
@@ -0,0 +1,316 @@
module Docs

import Base.Markdown: @doc_str, @doc_mstr

export doc, @doc

# Basic API / Storage

const modules = Module[]

meta() = current_module().META

macro init ()
META = esc(:META)
quote
if !isdefined(:META)
const $META = ObjectIdDict()
doc($META, doc"Documentation metadata for $(string(current_module())).")
push!(modules, current_module())
nothing
end
end
end

function doc(obj, data)
meta()[obj] = data
end

function doc(obj)
for mod in modules
haskey(mod.META, obj) && return mod.META[obj]
end
end

function doc(obj::Union(Symbol, AbstractString))
doc(current_module().(symbol(obj)))
end

# Function / Method support

function newmethod(defs)
keylen = -1
key = nothing
for def in defs
length(def.sig) > keylen && (keylen = length(def.sig); key = def)
end
return key
end

function newmethod(funcs, f)
applicable = Method[]
for def in methods(f)
(!haskey(funcs, def) || funcs[def] != def.func) && push!(applicable, def)
end
return newmethod(applicable)
end

function trackmethod(def)
name = namify(unblock(def))
f = esc(name)
quote
if isdefined($(Expr(:quote, name))) && isgeneric($f)
funcs = [def => def.func for def in methods($f)]
$(esc(def))
$f, newmethod(funcs, $f)
else
$(esc(def))
$f, newmethod(methods($f))
end
end
end

type FuncDoc
order::Vector{Method}
meta::Dict{Method, Any}
source::Dict{Method, Any}
end

FuncDoc() = FuncDoc(Method[], Dict(), Dict())

getset(coll, key, default) = coll[key] = get(coll, key, default)

function doc(f::Function, m::Method, data, source)
fd = getset(meta(), f, FuncDoc())
isa(fd, FuncDoc) || error("Can't document a method when the function already has metadata")
!haskey(fd.meta, m) && push!(fd.order, m)
fd.meta[m] = data
fd.source[m] = source
end

function doc(f::Function)
docs = []
for mod in modules
if haskey(mod.META, f)
fd = mod.META[f]
if isa(fd, FuncDoc)
for m in fd.order
push!(docs, fd.meta[m])
end
elseif length(docs) == 0
return fd
end
end
end
return catdoc(docs...)
end

catdoc() = nothing
catdoc(xs...) = [xs...]

# Macros

isexpr(x::Expr, ts...) = x.head in ts
isexpr{T}(x::T, ts...) = T in ts

function unblock(ex)
isexpr(ex, :block) || return ex
exs = filter(ex->!isexpr(ex, :line), ex.args)
length(exs) == 1 || return ex
return exs[1]
end

namify(ex::Expr) = namify(ex.args[1])
namify(sy::Symbol) = sy

function mdify(ex)
if isa(ex, AbstractString)
:(@doc_str $(esc(ex)))
elseif isexpr(ex, :macrocall) && namify(ex) == symbol("@mstr")
:(@doc_mstr $(esc(ex.args[2])))
else
esc(ex)
end
end

function namedoc(meta, def, name)
quote
@init
$(esc(def))
doc($(esc(name)), $(mdify(meta)))
nothing
end
end

function funcdoc(meta, def)
quote
@init
f, m = $(trackmethod(def))
doc(f, m, $(mdify(meta)), $(esc(Expr(:quote, def))))
f
end
end

function objdoc(meta, def)
quote
@init
f = $(esc(def))
doc(f, $(mdify(meta)))
f
end
end

fexpr(ex) = isexpr(ex, :function) || (isexpr(ex, :(=)) && isexpr(ex.args[1], :call))

function docm(meta, def)
def′ = unblock(def)
isexpr(def′, :macro) && return namedoc(meta, def,
symbol(string("@", namify(def′))))
isexpr(def′, :type) && return namedoc(meta, def, namify(def′.args[2]))
fexpr(def′) && return funcdoc(meta, def)
isexpr(def, :macrocall) && (def = namify(def))
return objdoc(meta, def)
end

function docm(ex)
isexpr(ex, :->) && return docm(ex.args...)
isexpr(ex, :macrocall) && (ex = namify(ex))
:(doc($(esc(ex))))
end

macro doc (args...)
docm(args...)
end

# Metametadata

@doc """
# Documentation
The `@doc` macro can be used to both set and retrieve documentation /
metadata. By default, documentation is written as Markdown, but any
object can be placed before the arrow. For example:
@doc \"""
# The Foo Function
`foo(x)`: Foo the living hell out of `x`.
\""" ->
function foo() ...
The `->` is not required if the object is on the same line, e.g.
@doc "foo" foo
# Retrieving Documentation
You can retrieve docs for functions, macros and other objects as
follows:
@doc foo
@doc @time
@doc md""
# Functions & Methods
Placing documentation before a method definition (e.g. `function foo()
...` or `foo() = ...`) will cause that specific method to be
documented, as opposed to the whole function. Method docs are
concatenated together in the order they were defined to provide docs
for the function.
""" @doc

@doc "`doc(obj)`: Get the doc metadata for `obj`." doc

@doc """
`catdoc(xs...)`: Combine the documentation metadata `xs` into a single meta object.
""" catdoc

# Text / HTML objects

import Base: print, writemime

export HTML, @html_str, @html_mstr

export HTML, Text

@doc """
`HTML(s)`: Create an object that renders `s` as html.
HTML("<div>foo</div>")
You can also use a stream for large amounts of data:
HTML() do io
println(io, "<div>foo</div>")
end
""" ->
type HTML{T}
content::T
end

function HTML(xs...)
HTML() do io
for x in xs
writemime(io, MIME"text/html"(), x)
end
end
end

writemime(io::IO, ::MIME"text/html", h::HTML) = print(io, h.content)
writemime(io::IO, ::MIME"text/html", h::HTML{Function}) = h.content(io)

@doc "Create an `HTML` object from a literal string." ->
macro html_str (s)
:(HTML($s))
end

@doc (@doc html"") ->
macro html_mstr (s)
:(HTML($(Base.triplequoted(s))))
end

function catdoc(xs::HTML...)
HTML() do io
for x in xs
writemime(io, MIME"text/html"(), x)
end
end
end

export Text, @text_str, @text_mstr

# @doc """
# `Text(s)`: Create an object that renders `s` as plain text.

# HTML("foo")

# You can also use a stream for large amounts of data:

# Text() do io
# println(io, "foo")
# end
# """ ->
type Text{T}
content::T
end

print(io::IO, t::Text) = print(io, t.content)
print(io::IO, t::Text{Function}) = t.content(io)
writemime(io::IO, ::MIME"text/plain", t::Text) = print(io, t)

@doc "Create a `Text` object from a literal string." ->
macro text_str (s)
:(Text($s))
end

@doc (@doc text"") ->
macro text_mstr (s)
:(Text($(Base.triplequoted(s))))
end

function catdoc(xs::Text...)
Text() do io
for x in xs
writemime(io, MIME"text/plain"(), x)
end
end
end

end
13 changes: 12 additions & 1 deletion base/exports.jl
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@ export
Test,
BLAS,
LAPACK,
Docs,
Markdown,

# Types
AbstractMatrix,
Expand Down Expand Up @@ -1221,6 +1223,8 @@ export
popdisplay,
pushdisplay,
redisplay,
HTML,
Text,

# distributed arrays
dfill,
Expand Down Expand Up @@ -1339,6 +1343,10 @@ export
@r_str,
@r_mstr,
@v_str,
@text_str,
@text_mstr,
@html_str,
@html_mstr,
@int128_str,
@uint128_str,
@bigint_str,
Expand Down Expand Up @@ -1392,4 +1400,7 @@ export
@simd,
@label,
@goto,
@inline
@inline,
@doc,
@doc_str,
@doc_mstr
14 changes: 13 additions & 1 deletion base/help.jl
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,7 @@ isname(n::Symbol) = true
isname(ex::Expr) = ((ex.head == :. && isname(ex.args[1]) && isname(ex.args[2]))
|| (ex.head == :quote && isname(ex.args[1])))

macro help(ex)
macro help_(ex)
if ex === :? || ex === :help
return Expr(:call, :help)
elseif !isa(ex, Expr) || isname(ex)
Expand All @@ -199,4 +199,16 @@ macro help(ex)
end
end

macro help (ex)
if ex === :? || ex === :help
return :(@help_ $(esc(ex)))
else
quote
let doc = @doc $(esc(ex))
doc nothing ? doc : @help_ $(esc(ex))
end
end
end
end

end # module
5 changes: 5 additions & 0 deletions base/markdown/Common/Common.jl
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
include("block.jl")
include("inline.jl")

@flavor common [list, indentcode, blockquote, hashheader, paragraph,
escapes, en_dash, inline_code, asterisk_bold, asterisk_italic, image, link]
Loading

0 comments on commit d43dfa8

Please sign in to comment.