more AI stuff - not reviewed

This commit is contained in:
Emil Lerch 2025-12-18 09:48:57 -08:00
parent bb25def875
commit 68dbe7d7dd
Signed by: lobo
GPG key ID: A7B62D657EF764F8
8 changed files with 329 additions and 37 deletions

View file

@ -4,6 +4,9 @@ const Cache = @import("../cache/cache.zig").Cache;
const WeatherProvider = @import("../weather/provider.zig").WeatherProvider;
const ansi = @import("../render/ansi.zig");
const line = @import("../render/line.zig");
const json = @import("../render/json.zig");
const v2 = @import("../render/v2.zig");
const custom = @import("../render/custom.zig");
const help = @import("help.zig");
pub const HandleWeatherOptions = struct {
@ -77,15 +80,26 @@ fn handleWeatherInternal(
const query = try req.query();
const format = query.get("format");
const output = if (format) |fmt|
try line.render(allocator, weather, fmt)
else
try ansi.render(allocator, weather, .{});
const output = if (format) |fmt| blk: {
if (std.mem.eql(u8, fmt, "j1")) {
res.content_type = .JSON;
break :blk try json.render(allocator, weather);
} else if (std.mem.eql(u8, fmt, "v2")) {
break :blk try v2.render(allocator, weather);
} else if (std.mem.startsWith(u8, fmt, "%")) {
break :blk try custom.render(allocator, weather, fmt);
} else {
break :blk try line.render(allocator, weather, fmt);
}
} else try ansi.render(allocator, weather, .{});
const ttl = 1000 + std.crypto.random.intRangeAtMost(u64, 0, 1000);
try opts.cache.put(cache_key, output, ttl);
res.content_type = .TEXT;
if (res.content_type != .JSON) {
res.content_type = .TEXT;
}
res.body = output;
}

88
zig/src/http/query.zig Normal file
View file

@ -0,0 +1,88 @@
const std = @import("std");
pub const QueryParams = struct {
format: ?[]const u8 = null,
lang: ?[]const u8 = null,
units: ?Units = null,
transparency: ?u8 = null,
pub const Units = enum {
metric,
uscs,
};
pub fn parse(allocator: std.mem.Allocator, query_string: []const u8) !QueryParams {
var params = QueryParams{};
var iter = std.mem.splitScalar(u8, query_string, '&');
while (iter.next()) |pair| {
if (pair.len == 0) continue;
var kv = std.mem.splitScalar(u8, pair, '=');
const key = kv.next() orelse continue;
const value = kv.next();
if (std.mem.eql(u8, key, "format")) {
params.format = if (value) |v| try allocator.dupe(u8, v) else null;
} else if (std.mem.eql(u8, key, "lang")) {
params.lang = if (value) |v| try allocator.dupe(u8, v) else null;
} else if (std.mem.eql(u8, key, "u")) {
params.units = .uscs;
} else if (std.mem.eql(u8, key, "m")) {
params.units = .metric;
} else if (std.mem.eql(u8, key, "transparency")) {
if (value) |v| {
params.transparency = try std.fmt.parseInt(u8, v, 10);
}
} else if (std.mem.eql(u8, key, "t")) {
params.transparency = 150;
}
}
return params;
}
};
test "parse empty query" {
const allocator = std.testing.allocator;
const params = try QueryParams.parse(allocator, "");
try std.testing.expect(params.format == null);
try std.testing.expect(params.lang == null);
try std.testing.expect(params.units == null);
}
test "parse format parameter" {
const allocator = std.testing.allocator;
const params = try QueryParams.parse(allocator, "format=j1");
defer if (params.format) |f| allocator.free(f);
try std.testing.expect(params.format != null);
try std.testing.expectEqualStrings("j1", params.format.?);
}
test "parse units parameters" {
const allocator = std.testing.allocator;
const params_m = try QueryParams.parse(allocator, "m");
try std.testing.expectEqual(QueryParams.Units.metric, params_m.units.?);
const params_u = try QueryParams.parse(allocator, "u");
try std.testing.expectEqual(QueryParams.Units.uscs, params_u.units.?);
}
test "parse multiple parameters" {
const allocator = std.testing.allocator;
const params = try QueryParams.parse(allocator, "format=3&lang=de&m");
defer if (params.format) |f| allocator.free(f);
defer if (params.lang) |l| allocator.free(l);
try std.testing.expectEqualStrings("3", params.format.?);
try std.testing.expectEqualStrings("de", params.lang.?);
try std.testing.expectEqual(QueryParams.Units.metric, params.units.?);
}
test "parse transparency" {
const allocator = std.testing.allocator;
const params_t = try QueryParams.parse(allocator, "t");
try std.testing.expectEqual(@as(u8, 150), params_t.transparency.?);
const params_custom = try QueryParams.parse(allocator, "transparency=200");
try std.testing.expectEqual(@as(u8, 200), params_custom.transparency.?);
}

View file

@ -74,7 +74,7 @@ pub const Resolver = struct {
}
// Format IP address
const ip_str = try std.fmt.allocPrint(self.allocator, "{}", .{addr_list.addrs[0].any});
const ip_str = try std.fmt.allocPrint(self.allocator, "{any}", .{addr_list.addrs[0].any});
defer self.allocator.free(ip_str);
return self.resolveIP(ip_str);

View file

@ -70,6 +70,9 @@ test {
_ = @import("http/query.zig");
_ = @import("http/help.zig");
_ = @import("render/line.zig");
_ = @import("render/json.zig");
_ = @import("render/v2.zig");
_ = @import("render/custom.zig");
_ = @import("location/geoip.zig");
_ = @import("location/resolver.zig");
}

137
zig/src/render/custom.zig Normal file
View file

@ -0,0 +1,137 @@
const std = @import("std");
const types = @import("../weather/types.zig");
const weather_icons = [_][]const u8{
"", // 0-99
"🌑", // 100-199
"", // 200-299
"☁️", // 300-399
};
fn getWeatherIcon(code: u16) []const u8 {
const idx = @min(code / 100, weather_icons.len - 1);
return weather_icons[idx];
}
pub fn render(allocator: std.mem.Allocator, weather: types.WeatherData, format: []const u8) ![]const u8 {
var output: std.ArrayList(u8) = .empty;
errdefer output.deinit(allocator);
const writer = output.writer(allocator);
var i: usize = 0;
while (i < format.len) {
if (format[i] == '%' and i + 1 < format.len) {
const code = format[i + 1];
switch (code) {
'c' => try writer.print("{s}", .{getWeatherIcon(weather.current.weather_code)}),
'C' => try writer.print("{s}", .{weather.current.condition}),
'h' => try writer.print("{d}%", .{weather.current.humidity}),
't' => try writer.print("{d:.0}°C", .{weather.current.temp_c}),
'f' => try writer.print("{d:.0}°C", .{weather.current.temp_c}), // Feels like (same as temp for now)
'w' => try writer.print("{d:.0} km/h {s}", .{ weather.current.wind_kph, weather.current.wind_dir }),
'l' => try writer.print("{s}", .{weather.location}),
'p' => try writer.print("{d:.1} mm", .{weather.current.precip_mm}),
'P' => try writer.print("{d:.0} hPa", .{weather.current.pressure_mb}),
'm' => try writer.print("🌕", .{}), // Moon phase placeholder
'M' => try writer.print("15", .{}), // Moon day placeholder
'o' => try writer.print("0%", .{}), // Probability of precipitation placeholder
'D' => try writer.print("06:00", .{}), // Dawn placeholder
'S' => try writer.print("07:30", .{}), // Sunrise placeholder
'z' => try writer.print("12:00", .{}), // Zenith placeholder
's' => try writer.print("18:30", .{}), // Sunset placeholder
'd' => try writer.print("20:00", .{}), // Dusk placeholder
'%' => try writer.print("%", .{}),
'n' => try writer.print("\n", .{}),
else => {
try writer.print("%{c}", .{code});
},
}
i += 2;
} else {
try writer.writeByte(format[i]);
i += 1;
}
}
return output.toOwnedSlice(allocator);
}
test "render custom format with location and temp" {
const allocator = std.testing.allocator;
const weather = types.WeatherData{
.location = "London",
.current = .{
.temp_c = 7.0,
.temp_f = 44.6,
.condition = "Overcast",
.weather_code = 122,
.humidity = 76,
.wind_kph = 11.0,
.wind_dir = "NNE",
.pressure_mb = 1019.0,
.precip_mm = 0.0,
},
.forecast = &[_]types.ForecastDay{},
.allocator = allocator,
};
const output = try render(allocator, weather, "%l: %c %t");
defer allocator.free(output);
try std.testing.expect(std.mem.indexOf(u8, output, "London") != null);
try std.testing.expect(std.mem.indexOf(u8, output, "7°C") != null);
}
test "render custom format with newline" {
const allocator = std.testing.allocator;
const weather = types.WeatherData{
.location = "Paris",
.current = .{
.temp_c = 10.0,
.temp_f = 50.0,
.condition = "Clear",
.weather_code = 113,
.humidity = 65,
.wind_kph = 8.0,
.wind_dir = "E",
.pressure_mb = 1020.0,
.precip_mm = 0.0,
},
.forecast = &[_]types.ForecastDay{},
.allocator = allocator,
};
const output = try render(allocator, weather, "%l%n%C");
defer allocator.free(output);
try std.testing.expect(std.mem.indexOf(u8, output, "Paris\nClear") != null);
}
test "render custom format with humidity and pressure" {
const allocator = std.testing.allocator;
const weather = types.WeatherData{
.location = "Berlin",
.current = .{
.temp_c = 5.0,
.temp_f = 41.0,
.condition = "Cloudy",
.weather_code = 119,
.humidity = 85,
.wind_kph = 12.0,
.wind_dir = "W",
.pressure_mb = 1012.0,
.precip_mm = 0.2,
},
.forecast = &[_]types.ForecastDay{},
.allocator = allocator,
};
const output = try render(allocator, weather, "Humidity: %h, Pressure: %P");
defer allocator.free(output);
try std.testing.expect(std.mem.indexOf(u8, output, "85%") != null);
try std.testing.expect(std.mem.indexOf(u8, output, "1012") != null);
}

View file

@ -2,10 +2,7 @@ const std = @import("std");
const types = @import("../weather/types.zig");
pub fn render(allocator: std.mem.Allocator, weather: types.WeatherData) ![]const u8 {
var output = std.ArrayList(u8).init(allocator);
errdefer output.deinit();
try std.json.stringify(.{
const data = .{
.current_condition = .{
.temp_C = weather.current.temp_c,
.temp_F = weather.current.temp_f,
@ -17,31 +14,10 @@ pub fn render(allocator: std.mem.Allocator, weather: types.WeatherData) ![]const
.pressure = weather.current.pressure_mb,
.precipMM = weather.current.precip_mm,
},
.weather = blk: {
var forecast_array = std.ArrayList(struct {
date: []const u8,
maxtempC: f32,
mintempC: f32,
weatherCode: u16,
weatherDesc: []const u8,
}).init(allocator);
defer forecast_array.deinit();
.weather = weather.forecast,
};
for (weather.forecast) |day| {
try forecast_array.append(.{
.date = day.date,
.maxtempC = day.max_temp_c,
.mintempC = day.min_temp_c,
.weatherCode = day.weather_code,
.weatherDesc = day.condition,
});
}
break :blk try forecast_array.toOwnedSlice();
},
}, .{}, output.writer());
return output.toOwnedSlice();
return try std.fmt.allocPrint(allocator, "{any}", .{std.json.fmt(data, .{})});
}
test "render json format" {

View file

@ -9,30 +9,36 @@ pub fn render(allocator: std.mem.Allocator, data: types.WeatherData, format: []c
data.current.temp_c,
});
} else if (std.mem.eql(u8, format, "2")) {
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C 🌬️{s}{d:.0}km/h", .{
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C {s}{s}{d:.0}km/h", .{
data.location,
getConditionEmoji(data.current.weather_code),
data.current.temp_c,
"🌬️",
data.current.wind_dir,
data.current.wind_kph,
});
} else if (std.mem.eql(u8, format, "3")) {
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C 🌬️{s}{d:.0}km/h 💧{d}%%", .{
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C {s}{s}{d:.0}km/h {s}{d}%%", .{
data.location,
getConditionEmoji(data.current.weather_code),
data.current.temp_c,
"🌬️",
data.current.wind_dir,
data.current.wind_kph,
"💧",
data.current.humidity,
});
} else if (std.mem.eql(u8, format, "4")) {
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C 🌬️{s}{d:.0}km/h 💧{d}%% ☀️", .{
return std.fmt.allocPrint(allocator, "{s}: {s} {d:.0}°C {s}{s}{d:.0}km/h {s}{d}%% {s}", .{
data.location,
getConditionEmoji(data.current.weather_code),
data.current.temp_c,
"🌬️",
data.current.wind_dir,
data.current.wind_kph,
"💧",
data.current.humidity,
"☀️",
});
} else {
return renderCustom(allocator, data, format);

68
zig/src/render/v2.zig Normal file
View file

@ -0,0 +1,68 @@
const std = @import("std");
const types = @import("../weather/types.zig");
pub fn render(allocator: std.mem.Allocator, weather: types.WeatherData) ![]const u8 {
var output: std.ArrayList(u8) = .empty;
errdefer output.deinit(allocator);
const writer = output.writer(allocator);
// Header with location
try writer.print("Weather report: {s}\n\n", .{weather.location});
// Current conditions
try writer.print(" Current conditions\n", .{});
try writer.print(" {s}\n", .{weather.current.condition});
try writer.writeAll(" 🌡️ ");
try writer.print("{d:.1}°C ({d:.1}°F)\n", .{ weather.current.temp_c, weather.current.temp_f });
try writer.writeAll(" 💧 ");
try writer.print("{d}%\n", .{weather.current.humidity});
try writer.writeAll(" 🌬️ ");
try writer.print("{d:.1} km/h {s}\n", .{ weather.current.wind_kph, weather.current.wind_dir });
try writer.writeAll(" 🔽 ");
try writer.print("{d:.1} hPa\n", .{weather.current.pressure_mb});
try writer.writeAll(" 💦 ");
try writer.print("{d:.1} mm\n\n", .{weather.current.precip_mm});
// Forecast
if (weather.forecast.len > 0) {
try writer.print(" Forecast\n", .{});
for (weather.forecast) |day| {
try writer.print(" {s}: {s}\n", .{ day.date, day.condition });
try writer.writeAll("");
try writer.print("{d:.1}°C ", .{day.max_temp_c});
try writer.writeAll("");
try writer.print("{d:.1}°C\n", .{day.min_temp_c});
}
}
return output.toOwnedSlice(allocator);
}
test "render v2 format" {
const allocator = std.testing.allocator;
const weather = types.WeatherData{
.location = "Munich",
.current = .{
.temp_c = 12.0,
.temp_f = 53.6,
.condition = "Overcast",
.weather_code = 122,
.humidity = 80,
.wind_kph = 15.0,
.wind_dir = "NW",
.pressure_mb = 1015.0,
.precip_mm = 0.5,
},
.forecast = &[_]types.ForecastDay{},
.allocator = allocator,
};
const output = try render(allocator, weather);
defer allocator.free(output);
try std.testing.expect(output.len > 0);
try std.testing.expect(std.mem.indexOf(u8, output, "Munich") != null);
try std.testing.expect(std.mem.indexOf(u8, output, "Current conditions") != null);
try std.testing.expect(std.mem.indexOf(u8, output, "12.0°C") != null);
}