aboutsummaryrefslogtreecommitdiff
path: root/src/http_ffi.erl
blob: e9de7e7daf402619845eb0301411d4e7827e2a69 (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
-module(http_ffi).
-export([serve/4, response_default_headers/0]).

serve(Host, Port, OnStart, OnPortTaken) ->
    {ok, Pattern} = re:compile("name *= *\"(?<Name>.+)\""),
    {ok, Toml} = file:read_file("gleam.toml"),
    {match, [Name]} = re:run(Toml, Pattern, [{capture, all_names, binary}]),

    Html =
        <<
            "<!DOCTYPE html>\n"
            "<html lang=\"en\">\n"
            "<head>\n"
            "  <meta charset=\"UTF-8\">\n"
            "  <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\">\n"
            "  <title>Lustre preview server</title>\n"
            "\n"
            "  <script type=\"module\">\n"
            "    import { main } from './",
            Name/binary,
            "/",
            Name/binary,
            ".mjs'\n"
            "\n"
            "    document.addEventListener(\"DOMContentLoaded\", () => {\n"
            "      main();\n"
            "    });\n"
            "  </script>\n"
            "</head>\n"
            "<body>\n"
            "  <div data-lustre-app></div>\n"
            "</body>\n"
            "</html>"
        >>,

    file:write_file("build/dev/javascript/index.html", Html),

    AbsPath =
        string:trim(
            filename:absname("build/dev/javascript"), trailing, "/."
        ),

    inets:start(),
    Address = {127, 0, 0, 1},

    ActualPort =
        case port_available(Port) of
            true ->
                Port;
            false ->
                OnPortTaken(Port),
                first_available_port(Port + 1)
        end,

    {ok, Pid} =
        httpd:start_service([
            {bind_address, Address},
            {document_root, AbsPath},
            {server_root, AbsPath},
            {directory_index, ["index.html"]},
            {server_name, binary_to_list(Host)},
            {port, ActualPort},
            {default_type, "text/html"},
            {mime_types, mime_types()},
            {customize, ?MODULE},
            {modules, [mod_alias, mod_dir, mod_get]}
        ]),

    OnStart(ActualPort),

    receive
        {From, shutdown} ->
            ok = httpd:stop_service(Pid),
            From ! done
    end.

port_available(Port) ->
    case gen_tcp:listen(Port, []) of
        {ok, Sock} ->
            ok = gen_tcp:close(Sock),
            true;
        _ ->
            false
    end.

first_available_port(Port) ->
    case port_available(Port) of
        true -> Port;
        false -> first_available_port(Port + 1)
    end.

mime_types() ->
    [
        {"html", "text/html"},
        {"htm", "text/html"},
        {"js", "text/javascript"},
        {"mjs", "text/javascript"},
        {"css", "text/css"},
        {"gif", "image/gif"},
        {"jpg", "image/jpeg"},
        {"jpeg", "image/jpeg"},
        {"png", "image/png"}
    ].

response_default_headers() ->
    [
        {"cache-control", "no-store, no-cache, must-revalidate, private"},
        {"pragma", "no-cache"}
    ].