dstep.translator.IncludeHandler source code

1 /**
2  * Copyright: Copyright (c) 2012 Jacob Carlborg. All rights reserved.
3  * Authors: Jacob Carlborg
4  * Version: Initial created: Jun 15, 2012
5  * License: $(LINK2 http://www.boost.org/LICENSE_1_0.txt, Boost Software License 1.0)
6  */
7 module dstep.translator.IncludeHandler;
8 
9 import Path = std.path;
10 
11 import mambo.core._;
12 
13 private IncludeHandler includeHandler_;
14 
15 @property IncludeHandler includeHandler ()
16 {
17     return includeHandler_;
18 }
19 
20 static this ()
21 {
22     includeHandler_ = new IncludeHandler;
23 }
24 
25 class IncludeHandler
26 {
27     private string[] rawIncludes;
28     private string[] imports;
29     static string[string] knownIncludes;
30 
31     static this ()
32     {
33         knownIncludes = [
34             "complex" : "core.stdc.complex",
35             "config" : "core.stdc.config",
36             "ctype" : "core.stdc.ctype",
37             "errno" : "core.stdc.errno",
38             "fenv" : "core.stdc.fenv",
39             "float" : "core.stdc.float",
40             "inttypes" : "core.stdc.inttypes",
41             "limits" : "core.stdc.limits",
42             "locale" : "core.stdc.locale",
43             "math" : "core.stdc.math",
44             "signal" : "core.stdc.signal",
45             "stdarg" : "core.stdc.stdarg",
46             "stddef" : "core.stdc.stddef",
47             "stdint" : "core.stdc.stdint",
48             "stdio" : "core.stdc.stdio",
49             "stdlib" : "core.stdc.stdlib",
50             "string" : "core.stdc.string",
51             "tgmath" : "core.stdc.tgmath",
52             "time" : "core.stdc.time",
53             "wchar" : "core.stdc.wchar_",
54             "wctype" : "core.stdc.wctype",
55 
56             "dirent" : "core.sys.posix.dirent",
57             "dlfcn" : "core.sys.posix.dlfcn",
58             "fcntl" : "core.sys.posix.fcntl",
59             "netdb" : "core.sys.posix.netdb",
60             "poll" : "core.sys.posix.poll",
61             "pthread" : "core.sys.posix.pthread",
62             "pwd" : "core.sys.posix.pwd",
63             "sched" : "core.sys.posix.sched",
64             "semaphore" : "core.sys.posix.semaphore",
65             "setjmp" : "core.sys.posix.setjmp",
66             "signal" : "core.sys.posix.signal",
67             "termios" : "core.sys.posix.termios",
68             "ucontext" : "core.sys.posix.ucontext",
69             "unistd" : "core.sys.posix.unistd",
70             "utime" : "core.sys.posix.utime",
71 
72             "arpa/inet" : "core.sys.posix.arpa.inet",
73 
74             "net/if" : "core.sys.posix.net.if_",
75 
76             "netinet/in" : "core.sys.posix.netinet.in_",
77             "netinet/tcp" : "core.sys.posix.netinet.tcp",
78 
79             "sys/ipc" : "core.sys.posix.sys.ipc",
80             "sys/mman" : "core.sys.posix.sys.mman",
81             "sys/select" : "core.sys.posix.sys.select",
82             "sys/shm" : "core.sys.posix.sys.shm",
83             "sys/socket" : "core.sys.posix.sys.socket",
84             "sys/stat" : "core.sys.posix.sys.stat",
85             "sys/time" : "core.sys.posix.sys.time",
86             "sys/types" : "core.sys.posix.sys.types",
87             "sys/_types" : "core.sys.posix.sys.types",
88             "sys/uio" : "core.sys.posix.sys.uio",
89             "sys/un" : "core.sys.posix.sys.un",
90             "sys/utsname" : "core.sys.posix.sys.utsname",
91             "sys/wait" : "core.sys.posix.sys.wait",
92 
93             "windows" : "core.sys.windows.windows"
94         ];
95     }
96 
97     void addInclude (string include)
98     {
99         rawIncludes ~= include;
100     }
101 
102     void addImport (string imp)
103     {
104         imports ~= imp;
105     }
106 
107     void addCompatible ()
108     {
109         imports ~= "core.stdc.config";
110     }
111 
112     string[] toImports ()
113     {
114         auto r =  rawIncludes.map!((e) {
115             if (auto i = isKnownInclude(e))
116                 return toImport(i);
117 
118             else
119                 return "";
120         });
121 
122         auto imps = imports.map!(e => toImport(e));
123 
124         return r.append(imps).filter!(e => e.any).unique.toArray;
125     }
126 
127 private:
128 
129     string toImport (string str)
130     {
131         return "import " ~ str ~ ";";
132     }
133 
134     string isKnownInclude (string include)
135     {
136         include = Path.stripExtension(include);
137 
138         if (auto r = knownIncludes.find!((k, _) => include.endsWith(k)))
139             return r.value;
140 
141         return null;
142     }
143 }