summaryrefslogtreecommitdiffstats
path: root/src/civetweb/test/resource_script_demo.lua
blob: e21e4654c7894a4c11be180e9bd6264578d1a512 (plain)
1
2
3
4
5
6
7
8
9
10
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
-- This is a Lua script that handles sub-resources, e.g. resource_script_demo.lua/path/file.ext

scriptUri = "resource_script_demo.lua"
envVar = "resource_script_demo_storage"

resourcedir = os.getenv(envVar) or "R:\\RESOURCEDIR"
method = mg.request_info.request_method:upper()

if resourcedir then
  attr = lfs.attributes(resourcedir)
end

if (not mg.request_info.uri:find(scriptUri)) or (not resourcedir) or (not attr) or (attr.mode~="directory") then
    mg.write("HTTP/1.0 500 OK\r\n")
    mg.write("Connection: close\r\n")
    mg.write("Content-Type: text/html; charset=utf-8\r\n")
    mg.write("\r\n")
    mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
    mg.write("<body>\r\nServer error.<br>\r\n")
    mg.write("The server admin must make sure this script is available as URI " .. scriptUri .. "<br>\r\n")
    mg.write("The server admin must set the environment variable " .. envVar .. " to a directory.<br>\r\n")
    mg.write("</body>\r\n</html>\r\n")
    return
end
subresource = mg.request_info.uri:match(scriptUri .. "/(.*)")

if not subresource then
    if method=="GET" then
        mg.write("HTTP/1.0 200 OK\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>No resource specified.<br>resourcedir is " .. resourcedir .. "</body></html>\r\n")
    else
        mg.write("HTTP/1.0 405 Method Not Allowed\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Method not allowed.</body></html>\r\n")
    end
    return
end


if method=="GET" then
    file = resourcedir .. "/" .. subresource
    if lfs.attributes(file) then
        mg.send_file(file)
    else
        mime = mg.get_mime_type(file)
        mg.write("HTTP/1.0 404 Not Found\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Resource of type \"" .. mime .. "\" not found.</body></html>\r\n")
    end
    return
end

if method=="PUT" then
    file = resourcedir .. "/" .. subresource
    mime = mg.get_mime_type(file)
    if lfs.attributes(file) then
        mg.write("HTTP/1.0 405 Method Not Allowed\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Resource of type \"" .. mime .. "\" already exists.</body></html>\r\n")
    else
        local f = io.open(file, "w")

        local data = {}
        repeat
            local l = mg.read();
            data[#data+1] = l;
        until ((l == "") or (l == nil));

        f:write(table.concat(data, ""))
        f:close()
        mg.write("HTTP/1.0 200 OK\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Resource of type \"" .. mime .. "\" created.</body></html>\r\n")
    end
    return
end

if method=="DELETE" then
    file = resourcedir .. "/" .. subresource
    mime = mg.get_mime_type(file)
    if lfs.attributes(file) then
        os.remove(file)
        mg.write("HTTP/1.0 200 OK\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Resource of type \"" .. mime .. "\" deleted.</body></html>\r\n")
    else
        mime = mg.get_mime_type(file)
        mg.write("HTTP/1.0 404 Not Found\r\n")
        mg.write("Connection: close\r\n")
        mg.write("Content-Type: text/html; charset=utf-8\r\n")
        mg.write("\r\n")
        mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
        mg.write("<body>Resource of type \"" .. mime .. "\" not found.</body></html>\r\n")
    end
    return
end

-- Any other method
mg.write("HTTP/1.0 405 Method Not Allowed\r\n")
mg.write("Connection: close\r\n")
mg.write("Content-Type: text/html; charset=utf-8\r\n")
mg.write("\r\n")
mg.write("<html><head><title>Civetweb Lua script resource handling test</title></head>\r\n")
mg.write("<body>Method not allowed.</body></html>\r\n")