1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79
80. 80
81. 81
82. 82
83. 83
84. 84
85. 85
86. 86
87. 87
88. 88
89. 89
90. 90
91. 91
92. 92
93. 93
94. 94
95. 95
96. 96
97. 97
98. 98
99. 99
100. 100
101. 101
102. 102
103. 103
104. 104
105. 105
106. 106
107. 107
108. 108
109. 109
110. 110
111. 111
112. 112
113. 113
114. 114
115. 115
116. 116
117. 117
118. 118
119. 119
120. 120
121. 121
122. 122
123. 123
124. 124
125. 125

// Copyright (C) 2025 Shota FUJI
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
//
// SPDX-License-Identifier: GPL-3.0-only

const std = @import("std");

const config = @import("config");

const PollOptions = struct {
    bin: []const u8,
    tz: []const u8 = "UTC",
    twilight_angle: ?[]const u8 = null,
    longitude: []const u8 = "31.132484E",
    latitude: []const u8 = "29.977435N",
};

fn poll(allocator: std.mem.Allocator, opts: PollOptions) !std.process.Child.RunResult {
    var env = std.process.EnvMap.init(allocator);
    defer env.deinit();

    try env.put("TZ", opts.tz);

    var args = std.ArrayList([]const u8).init(allocator);
    defer args.deinit();

    try args.append(opts.bin);
    try args.append("poll");

    if (opts.twilight_angle) |angle| {
        try args.append("angle");
        try args.append(angle);
    }

    try args.append(opts.latitude);
    try args.append(opts.longitude);

    return try std.process.Child.run(.{
        .allocator = allocator,
        .argv = args.items,
        .env_map = &env,
    });
}

test {
    const legacy = try poll(std.testing.allocator, .{
        .bin = config.legacy_bin,
    });
    defer std.testing.allocator.free(legacy.stderr);
    defer std.testing.allocator.free(legacy.stdout);

    const new = try poll(std.testing.allocator, .{
        .bin = config.new_bin,
    });
    defer std.testing.allocator.free(new.stderr);
    defer std.testing.allocator.free(new.stdout);

    try std.testing.expectEqual(legacy.term.Exited, new.term.Exited);
    try std.testing.expectEqualStrings(legacy.stderr, new.stderr);
    try std.testing.expectEqualStrings(legacy.stdout, new.stdout);
}

test {
    const regular = try poll(std.testing.allocator, .{
        .bin = config.new_bin,
    });
    defer std.testing.allocator.free(regular.stderr);
    defer std.testing.allocator.free(regular.stdout);

    var env = std.process.EnvMap.init(std.testing.allocator);
    defer env.deinit();

    try env.put("TZ", "UTC");

    const debug = try std.process.Child.run(.{
        .allocator = std.testing.allocator,
        .argv = &.{ config.new_bin, "--debug", "poll", "29.977435N", "31.132484E" },
        .env_map = &env,
    });
    defer std.testing.allocator.free(debug.stderr);
    defer std.testing.allocator.free(debug.stdout);

    // Original sunwait prints debug log to stdout.
    // TODO: Print to stderr
    try std.testing.expect(debug.stdout.len > regular.stdout.len);
}

test "From original USAGE.txt: Example 3" {
    // Indicate by program exit-code if is Day or Night using a custom twilight angle of
    // 10 degrees above horizon. Washington, UK.
    const legacy = try poll(std.testing.allocator, .{
        .bin = config.legacy_bin,
        .twilight_angle = "10",
        .latitude = "54.897786N",
        // Original example uses -1.517536E but it does not make sense.
        .longitude = "1.517536W",
    });
    defer std.testing.allocator.free(legacy.stderr);
    defer std.testing.allocator.free(legacy.stdout);

    const new = try poll(std.testing.allocator, .{
        .bin = config.new_bin,
        .twilight_angle = "10",
        .latitude = "54.897786N",
        .longitude = "1.517536W",
    });
    defer std.testing.allocator.free(new.stderr);
    defer std.testing.allocator.free(new.stdout);

    try std.testing.expectEqual(legacy.term.Exited, new.term.Exited);
    try std.testing.expectEqualStrings(legacy.stderr, new.stderr);
    try std.testing.expectEqualStrings(legacy.stdout, new.stdout);
}