Fixed 404 error on index pages. Also changed to REQUEST_URI instead of SCRIPT_URL.
[public/www-new.git] / src / web.d
1 //#!/usr/bin/rdmd
2 import std.stdio, std.path, std.process, std.file, std.array, std.string, std.algorithm, std.datetime, std.ascii;
3 import config;
4
5 string dweb_root;
6
7 string indent = "";
8 void html(string s) { writeln(indent ~ s); }
9 void html_pop(string s) { indent = indent[0..max(0, $-4)]; html(s); }
10 void html_push(string s) { html(s); indent ~= "    "; }
11
12 void write_link(string to, bool expand) {
13   bool isdir = dirExists(dweb_root ~ "/srv/" ~ to[url_root.length..$]);
14   string flair = nav_tree_chev? (expand? "» " : "› ") : "";
15   html_push("<li" ~ (expand? " class=\"thisPage\" " : "") ~ ">");
16   html("<a href=\"" ~ to ~ (isdir ? "/" : "") ~ "\">"
17         ~ flair ~ baseName(to) ~ (isdir ? "/" : "") ~ "</a>");
18   html_pop("</li>");
19 }
20
21 void nav_tree_r(string url, string cur_loc, string[] subdirs) {
22   string[] dirs = array(map!"a.name"(dirEntries(dweb_root ~ "/srv/" ~ cur_loc, SpanMode.shallow)));
23   sort(dirs);
24   if (dirs.length == 0) return;
25   bool inserted_ul = false;
26   bool next = false;
27   string next_loc;
28   foreach(string s; dirs) {
29     s= s[(dweb_root ~ "/srv/").length..$];
30     string name = stripExtension(baseName(s));
31     if (name.length == 0) continue; // e.g. ".md", should we do something else with these files?
32     if (name == "index") continue; // "index" will never appear in the nav_tree.
33     if (name[0] == '@') continue; // hidden file
34     if (!inserted_ul) { html_push("<ul>"); inserted_ul = true; }
35     bool expand = subdirs.length > 0 && name == subdirs[0];
36     write_link(url_root ~ stripExtension(s), expand);
37
38     if (expand && isDir(dweb_root ~ "/srv/" ~ s)) {
39       if (nav_tree_vert) {
40         html_push("<li>");
41         nav_tree_r(url, (cur_loc == "" ? "" : cur_loc ~ "/") ~ subdirs[0], subdirs[1..$]);
42         html_pop("</li>");
43       } else {
44         next = true;
45         next_loc = (cur_loc == "" ? "" : cur_loc ~ "/") ~ subdirs[0];
46       }
47     }
48   }
49
50   if (inserted_ul) html_pop("</ul>");
51   if (next) nav_tree_r(url, next_loc, subdirs[1..$]);
52 }
53
54 void do_nav_tree(string url) {
55   html_push("<div id=\"" ~ (nav_tree_vert ? "" : "horiz-") ~ "side-bar\">");
56   nav_tree_r(url, "", cast(string[])array(pathSplitter(url)));
57   html_pop("</div>\n");
58 }
59
60 void not_found(string path) {
61   html("The page <code>" ~ path ~ "</code> does not exist. (404)");
62 }
63
64 void do_header() {
65   html_push("<div id=\"header\">");
66   
67   html_push("<div class=\"superHeader\">");
68   
69   html_push("<div class=\"left\">");
70   html_pop("</div>"); 
71   
72   html_push("<div class=\"right\">");
73   html("<a href=\"" ~ url_root ~ "changelog\">changelog</a>");
74   html_pop("</div>");
75   
76   html_pop("</div>");
77
78   html_push("<div class=\"midHeader\">");
79   html_push("<h1 class=\"headerTitle\">");
80   html("<a href=\"" ~ url_root ~ "\">" ~ site_title ~ " <span id=\"headerSubTitle\">" ~ site_subtitle ~ "</span></a>");
81   html_pop("</h1>");
82   html_pop("</div>");
83   
84   html_push("<div class=\"subHeader\">");
85   html("<br>");
86   html_pop("</div>");
87   
88   html_pop("</div>\n");
89 }
90
91 bool dirExists(string path)  { try { if (isDir(path))  return true; else return false; } catch (Exception e) { return false; } }
92
93 void do_content(string url) {
94   html_push("<div id=\"main-copy\"" ~ (nav_tree_vert? " class=\"main-copy-side-bar\"" : "")  ~ ">");
95   // first, see if we have something that wants to handle url outright
96   foreach (string glob, string h; handlers) {
97     if (globMatch(url, glob)) {
98       html(shell(dweb_root ~ "/bin/" ~ h ~ " " ~ url));
99       html_pop("</div>");
100       return;
101     }
102   }
103   // if that failed, see if we can handle the file
104   if (url == "" ? true : url[$-1] == '/') url ~= "index";
105   foreach (f; array(map!"a.name"(dirEntries(dirName(dweb_root ~ "/srv/" ~ url), SpanMode.shallow)))) {
106     if (isDir(f)) continue;
107     string name = baseName(f); name = name[0] == '@' ? name[1..$] : name;
108     if (stripExtension(name) == baseName(url)) {
109       foreach (string glob, string h; handlers) { 
110         if (globMatch(name, glob)) {
111           html(shell(dweb_root ~ "/bin/" ~ h ~ " " ~ f));
112           html_pop("</div>");
113           return;
114         }
115       }
116     }
117   }
118   if (baseName(url) != "index") not_found(url);
119   html_pop("</div>");
120 }
121
122 void do_footer() {
123   html_push("<div id=\"footer\">");
124
125   html_push("<div class=\"left\">");
126   html("<a href=\"" ~ url_root ~ "dweb\">Powered by dweb</a>");
127   html_pop("</div>");
128
129   html_push("<div class=\"right\">");
130   html("&nbsp;");
131   html_pop("</div>");
132
133   html_pop("</div>\n");
134 }
135
136 bool evil(string s) {
137   foreach(char c; s) if (!isAlphaNum(c) && c != '/' && c != '-' && c != '_') return true;
138   return false;
139 }
140
141 void main(string[] args) {
142   init_handlers();
143   dweb_root = getcwd()[0..$-4]; // take out bin/
144  
145   html("Content-type: text/html\n");
146   html("<!DOCTYPE html>");
147   html_push("<html>\n");
148
149   string url = getenv("REQUEST_URI")[url_root.length..$];
150   if (evil(url)) { html ("bad url."); return; }
151   
152   string pagename = baseName(url);
153   if (pagename.length != 0) pagename = " - " ~ pagename;
154   pagename = site_title ~ pagename;
155   
156   html_push("<head>");
157   html("<title>" ~ pagename ~ "</title>");
158   html("<link rel=\"stylesheet\" href=\"" ~ url_root ~ "pub/style/style.css\" type=\"text/css\" media=\"screen, handheld\" title=\"default\">");
159   html("<meta charset=\"UTF-8\">");
160   html("<meta http-equiv=\"Content-Type\" content=\"text/html; charset=UTF-8\">");
161   html_pop("</head>\n");
162   
163   html_push("<body" ~ (page_container? " style=\"text-align: center\"" : "")~ ">");
164   if (page_container) html_push("<div id=\"container\">");
165   do_header();
166   do_nav_tree(url);
167   do_content(url);
168   do_footer();
169   if (page_container) html_pop("</div>");
170   html_pop("</body>\n");
171
172   html_pop("</html>");
173 }