Class: Generator

Inherits:
Object
  • Object
show all
Defined in:
lib/generator.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Generator

Returns a new instance of Generator.



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/generator.rb', line 11

def initialize
    @util = Util.instance
    @setup = Setup.instance

    #删除目标目录
    FileUtils.rmtree @setup.target_dir

    #扫描文件
    scan = Scan.new
    scan.execute

    #存储文件
    @store = Store.new scan.files

    @compiler = Compiler.new
    @page_size = @setup.site_config['page_size'] || 10
    @page_size = 10 if @page_size.class != Integer

    self.generate_articles
    self.copy_theme_resource

    #复制文件
    self.copy_workbench_resource
    #生成首页
    self.generate_home
    #生成索引页
    self.generate_all_index @store.tree, '', true

    #如果需要执行命令命令
    self.execute_after_build
    puts 'Done...'
end

Instance Method Details

#compiler(filename, template_name, data) ⇒ Object

编译模板



140
141
142
143
144
145
 
# File 'lib/generator.rb', line 140

def compiler(filename, template_name, data)
    data['product'] = @util.get_product 
    data['root/relative_path'] = @util.get_relative_dot(filename)

    @compiler.execute template_name, data, true, filename
end

#copy(source, filename) ⇒ Object

复制文件到目标



161
162
163
164
165
 
# File 'lib/generator.rb', line 161

def copy(source, filename)
    target = File::join @setup.target_dir, filename

    FileUtils.cp_r source, target
end

#copy_theme_resource ⇒ Object

复制theme中的资源到目录



148
149
150
151
152
153
154
155
156
157
158
 
# File 'lib/generator.rb', line 148

def copy_theme_resource
    this = self
    theme_dir = @compiler.theme_dir

    Dir::entries(theme_dir).each{ |filename|
        #忽略掉以.开头的
        next if (/^\.|(template)/i =~ filename) != nil

        this.copy File::join(theme_dir, filename), filename
    }
end

#copy_workbench_resource ⇒ Object

复制工作目录的所有资源, 除忽略/.md/配置文件以外的



168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
 
# File 'lib/generator.rb', line 168

def copy_workbench_resource
    this = self
    Dir::entries(@util.workbench).each{ |filename|
        #忽略掉以.开头的, 以及markdown文件, 还有用户忽略的文件
        next if @util.is_shadow_file?(filename) or
            @util.is_markdown_file?(filename) or
            @util.local_theme_dir == filename or
            @setup.is_user_ignore_file?(filename)

        #当前的路径
        current_path = @util.get_merge_path filename, @util.workbench

        #build和内容退出
        next if @setup.target_dir === current_path or
            @setup.content_dir === current_path or
            @util.is_config_file? current_path

        this.copy File::join(@util.workbench, filename), filename
    }
end

#execute_after_build ⇒ Object

build后执行的脚本



45
46
47
48
49
 
# File 'lib/generator.rb', line 45

def execute_after_build
    command = @setup.site_config['after_build_shell']
    return if not command
    exec command
end

#generate_all_index(node, dir, ignore_first_index) ⇒ Object

创建所有的索引页, 如果有文件夹, 则根据配置创建文件夹中的索引页不会创建首页的index



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/generator.rb', line 58

def generate_all_index(node, dir, ignore_first_index)
    this = self

    #创建children的所有索引
    children = @store.get_children node
    page_count = (children.length.to_f / @page_size).ceil

    #生成此文件夹下的所有索引页
    (1..page_count).each { |page_index|
        #忽略第一页的索引, 一般是首页的情况下
        next if page_index == 1 and ignore_first_index

        # puts dir, page_index
        this.generate_index children, page_index, 'index', dir
    }

    #遍历所有的文件夹节点, 递归调用
    node.each { |key, sub_node|
        if not @store.is_children_key(key)
            parent_dir = dir + key + '/'
            this.generate_all_index sub_node, parent_dir, false
        end
    }
end

#generate_articles ⇒ Object

创建所有文章页



84
85
86
87
88
89
90
91
92
93
 
# File 'lib/generator.rb', line 84

def generate_articles
    @store.articles.each { |key, article|
        relative_url = article['relative_url']

        data = {
            'article' => article
        }
        self.compiler(relative_url, 'article', data)
    }
end

#generate_home ⇒ Object

创建首页



51
52
53
54
 
# File 'lib/generator.rb', line 51

def generate_home
    children = @store.get_children(@store.tree)
    self.generate_index children, 1, 'home', ''
end

#generate_index(children, page_index, template_name, dir) ⇒ Object

创建索引页, 包括首页以及子目录的索引页



96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
 
# File 'lib/generator.rb', line 96

def generate_index(children, page_index, template_name, dir)
    articles = []
    start_index = page_index * @page_size - @page_size
    end_index = start_index + @page_size
    #总页数
    page_count = (children.length.to_f / @page_size).ceil

    path = dir + (page_index == 1 ? 'index.html' : "page-#{page_index}.html")

    children[start_index..end_index].each { |current|
        articles.push @store.articles[current]
    }



    data = {
        'articles' => articles,
        'nav' => self.get_nav(page_index, page_count)
    }

    self.compiler path, template_name, data
end

#get_nav(page_index, page_count) ⇒ Object

处理上一页下一页



120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/generator.rb', line 120

def get_nav(page_index, page_count)
    nav = {}
    #上一页
    if(page_index == 2)
        nav['previous'] = 'index.html'
    elsif(page_index > 2)
        nav['previous'] = "page-#{page_index - 1}.html"
    end

    #下一页
    if(page_index < page_count)
        nav['next'] = "page-#{page_index + 1}.html"
    end

    #以后要处理总的分页信息

    nav
end