17 |
|
#include "Sleep.hpp" |
18 |
|
#include "Zlib/GzipWriter.hpp" |
19 |
|
|
20 |
+ |
int Usage() |
21 |
+ |
{ |
22 |
+ |
api::Cout << _B("Usage: ") << api::GetExecutablePath().GetName() << _B(" [-sitemap=.+] -base=.+ [-fs=.*] [-blog=.*]") << ios::NewLine; |
23 |
+ |
|
24 |
+ |
return 1; |
25 |
+ |
} |
26 |
+ |
|
27 |
|
int Main(const app::Options &options) |
28 |
|
{ |
29 |
< |
cse::String sitemap(_B("sitemap.gz")); |
29 |
> |
cse::String sitemap(_B("sitemap.gz")), base; |
30 |
|
_L<_R<Command> > commands; |
31 |
|
|
32 |
< |
api::Pcre::RegEx sitemap_(_B("^-sitemap=(.+)$")), fs(_B("^-fs=(.*)$")); |
32 |
> |
api::Pcre::RegEx sitemap_(_B("^-sitemap=(.+)$")), base_(_B("^-base=(.+)$")), fs_(_B("^-fs=(.*)$")), blog_(_B("^-blog=(.*)$")), exclude(_B("^-exclude=(.+)$")); |
33 |
|
|
34 |
|
_foreach (const app::ArgumentList, arg, app::GetArguments()) |
35 |
|
{ |
36 |
|
api::Pcre::RegEx::Match match; |
37 |
+ |
_R<FileSystemCommand> fs; |
38 |
+ |
_R<BlogCommand> blog; |
39 |
|
|
40 |
|
if (match = sitemap_(*arg)) |
41 |
|
sitemap = match[1]; |
42 |
< |
else if (match = fs(*arg)) |
42 |
> |
else if (match = base_(*arg)) |
43 |
> |
base = match[1]; |
44 |
> |
else if (match = fs_(*arg)) |
45 |
|
commands.InsertLast(new _H<FileSystemCommand>(match[1])); |
46 |
+ |
else if (match = blog_(*arg)) |
47 |
+ |
commands.InsertLast(new _H<BlogCommand>(match[1])); |
48 |
|
else |
49 |
< |
{ |
37 |
< |
api::Cout << _B("Usage: ") << api::GetExecutablePath().GetName() << _B(" [-sitemap=.+] [-fs=.*]") << ios::NewLine; |
38 |
< |
|
39 |
< |
return 1; |
40 |
< |
} |
49 |
> |
return Usage(); |
50 |
|
} |
51 |
|
|
52 |
< |
GoogleTron tron(sitemap, commands); |
52 |
> |
if (base.IsEmpty() || commands.IsEmpty()) |
53 |
> |
return Usage(); |
54 |
> |
|
55 |
> |
GoogleTron tron(sitemap, base, commands); |
56 |
|
|
57 |
|
return 0; |
58 |
|
} |
59 |
|
|
60 |
< |
GoogleTron::GoogleTron(const cse::String &sitemap, const _L<_R<Command> > &commands) : working(true), sitemap(sitemap), output(hop::BindAll(&GoogleTron::Output, this)) |
60 |
> |
GoogleTron::GoogleTron(const cse::String &sitemap, const cse::String &base, const _L<_R<Command> > &commands) : working(true), sitemap(sitemap), base(base), output(hop::BindAll(&GoogleTron::Output, this)) |
61 |
|
{ |
62 |
|
_foreach (const _L<_R<Command> >, command_, commands) |
63 |
|
if (_R<FileSystemCommand> command = dynamic_cast<FileSystemCommand *>(command_->GetValue())) |
64 |
|
threads.Add(hop::BindAll(&GoogleTron::FileSystem_, this, command)); |
65 |
+ |
else if (_R<BlogCommand> command = dynamic_cast<BlogCommand *>(command_->GetValue())) |
66 |
+ |
threads.Add(hop::BindAll(&GoogleTron::Blog_, this, command)); |
67 |
|
} |
68 |
|
|
69 |
|
GoogleTron::~GoogleTron() |
129 |
|
|
130 |
|
int GoogleTron::FileSystem_(const _R<FileSystemCommand> &command) |
131 |
|
{ |
132 |
< |
FileSystem fs(sitemap, queue, queueLock, command); |
132 |
> |
FileSystem fs(sitemap, base, queue, queueLock, command); |
133 |
> |
|
134 |
> |
return 0; |
135 |
> |
} |
136 |
> |
|
137 |
> |
int GoogleTron::Blog_(const _R<BlogCommand> &command) |
138 |
> |
{ |
139 |
> |
Blog blog(sitemap, base, queue, queueLock, command); |
140 |
|
|
141 |
|
return 0; |
142 |
|
} |