Module: PuppetStrings::Json

Defined in:
lib/puppet-strings/json.rb

Overview

The module for JSON related functionality.

Class Method Summary collapse

Class Method Details

.docstring_to_hash(docstring, select_tags = nil) ⇒ Hash

Converts a YARD::Docstring (or String) to a docstring hash for JSON output.

Parameters:

  • docstring (YARD::Docstring, String) —

    The docstring to convert to a hash.

  • select_tags (Array) (defaults to: nil) —

    List of tags to select. Other tags will be filtered out.

Returns:

  • (Hash) —

    Returns a hash representation of the given docstring.



55
56
57
58
59
60
61
62
63
64
 
# File 'lib/puppet-strings/json.rb', line 55

def self.docstring_to_hash(docstring, select_tags=nil)
  hash = {}
  hash[:text] = docstring
  if docstring.is_a? YARD::Docstring
    tags = tags_to_hashes(docstring.tags.select { |t| select_tags.nil? || select_tags.include?(t.tag_name.to_sym) })

    hash[:tags] = tags unless tags.empty?
  end
  hash
end

.render(file = nil) ⇒ void

This method returns an undefined value.

Renders the current YARD registry as JSON to the given file (or STDOUT if nil).

Parameters:

  • file (String) (defaults to: nil) —

    The path to the output file to render the registry to. If nil, output will be to STDOUT.



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/puppet-strings/json.rb', line 8

def self.render(file = nil)
  document = {
    puppet_classes: YARD::Registry.all(:puppet_class).sort_by!(&:name).map!(&:to_hash),
    defined_types: YARD::Registry.all(:puppet_defined_type).sort_by!(&:name).map!(&:to_hash),
    resource_types: YARD::Registry.all(:puppet_type).sort_by!(&:name).map!(&:to_hash),
    providers: YARD::Registry.all(:puppet_provider).sort_by!(&:name).map!(&:to_hash),
    puppet_functions: YARD::Registry.all(:puppet_function).sort_by!(&:name).map!(&:to_hash),
    # TODO: Need Ruby documentation?
  }

  if file
    File.open(file, 'w') do |f|
      f.write(JSON.pretty_generate(document))
      f.write("\n")
    end
  else
    puts JSON.pretty_generate(document)
  end
end

.tags_to_hashes(tags) ⇒ Array

Converts a list of tags into an array of hashes.

Parameters:

  • tags (Array) —

    List of tags to be converted into an array of hashes.

Returns:

  • (Array) —

    Returns an array of tag hashes.



31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/puppet-strings/json.rb', line 31

def self.tags_to_hashes(tags)
  # Skip over the API tags that are public
  tags.select { |t| (t.tag_name != 'api' || t.text != 'public') }.map do |t|
    next t.to_hash if t.respond_to?(:to_hash)

    tag = { tag_name: t.tag_name }
    # grab nested information for @option tags
    if tag[:tag_name] == 'option'
      tag[:opt_name] = t.pair.name
      tag[:opt_text] = t.pair.text
      tag[:opt_types] = t.pair.types
      tag[:parent] = t.name
    end
    tag[:text] = t.text if t.text
    tag[:types] = t.types if t.types
    tag[:name] = t.name if t.name
    tag
  end
end