mahmad/local-etc-hosts-updater
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add additional mapping for local domain

Browse source
This commit is contained in:
Muaz Ahmad 2024-05-13 01:34:30 +05:00
parent f755e0aeec
commit db1fbeb92a
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/hosts.zig
View file

@ -5,8 +5,8 @@ const hosts_header = "# local-etc-hosts-updater";
const FILE_LINE_BUFF_SIZE = 100; const FILE_LINE_BUFF_SIZE = 100;
pub fn update_hosts(ip: util.IP) !void { pub fn update_hosts(ip: util.IP, domain: util.Domain) !void {
try create_tmp_hosts(ip); try create_tmp_hosts(ip, domain);
try move_tmp_hosts(); try move_tmp_hosts();
} }
@ -61,7 +61,7 @@ fn purge_existing(old_hosts: std.fs.File, tmp_hosts: std.fs.File) !void {
} }
} }
fn append_new(tmp_hosts: std.fs.File, addr: std.net.Address) !void { fn append_new(tmp_hosts: std.fs.File, addr: std.net.Address, local_domain: util.Domain) !void {
const writer = tmp_hosts.writer(); const writer = tmp_hosts.writer();
var buff = [_]u8{0x00} ** FILE_LINE_BUFF_SIZE; var buff = [_]u8{0x00} ** FILE_LINE_BUFF_SIZE;
@ -85,6 +85,7 @@ fn append_new(tmp_hosts: std.fs.File, addr: std.net.Address) !void {
while (subdomains.next()) |subdomain| { while (subdomains.next()) |subdomain| {
try std.fmt.format(writer, "{s} {s}.{s}\n", .{ addr_str, subdomain, domain }); try std.fmt.format(writer, "{s} {s}.{s}\n", .{ addr_str, subdomain, domain });
} }
try std.fmt.format(writer, "{s} {s}\n", .{ addr_str, local_domain.name });
// add extra final newline // add extra final newline
try writer.writeByte('\n'); try writer.writeByte('\n');
@ -95,7 +96,7 @@ fn get_subdomains() !std.mem.SplitIterator(u8, .scalar) {
return std.mem.splitScalar(u8, std.mem.span(subdomain_list), ' '); return std.mem.splitScalar(u8, std.mem.span(subdomain_list), ' ');
} }
fn create_tmp_hosts(ip: util.IP) !void { fn create_tmp_hosts(ip: util.IP, domain: util.Domain) !void {
var old_hosts = try std.fs.openFileAbsoluteZ(try util.getenv("OLD_HOSTS_PATH"), .{ .mode = .read_only }); var old_hosts = try std.fs.openFileAbsoluteZ(try util.getenv("OLD_HOSTS_PATH"), .{ .mode = .read_only });
defer old_hosts.close(); defer old_hosts.close();
var tmp_hosts = try std.fs.createFileAbsoluteZ(try util.getenv("TMP_HOSTS_PATH"), .{ .truncate = true }); var tmp_hosts = try std.fs.createFileAbsoluteZ(try util.getenv("TMP_HOSTS_PATH"), .{ .truncate = true });
@ -103,7 +104,7 @@ fn create_tmp_hosts(ip: util.IP) !void {
try purge_existing(old_hosts, tmp_hosts); try purge_existing(old_hosts, tmp_hosts);
if (ip) |addr| { if (ip) |addr| {
try append_new(tmp_hosts, addr); try append_new(tmp_hosts, addr, domain);
} }
} }

2
src/main.zig
View file

@ -7,5 +7,5 @@ pub fn main() !void {
try util.check_perms(); try util.check_perms();
const domain, const ip_info = try util.get_input(); const domain, const ip_info = try util.get_input();
const ip = try mdns.get_mdns(domain, ip_info); const ip = try mdns.get_mdns(domain, ip_info);
try hosts.update_hosts(ip); try hosts.update_hosts(ip, domain);
} }