From c46cf8d97ec44dfcc6b2527489dd122754430c19 Mon Sep 17 00:00:00 2001
From: Emil Lerch <emil@lerch.org>
Date: Tue, 17 Oct 2023 20:26:14 -0700
Subject: [PATCH] steps 3 and 4

---
 src/main.zig | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 52 insertions(+), 2 deletions(-)

diff --git a/src/main.zig b/src/main.zig
index 1512820..0cfbc14 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -15,8 +15,6 @@ pub fn main() !void {
     //
     const accountid = @embedFile("accountid.txt");
     const worker_name = @embedFile("worker_name.txt");
-    const get_subdomain = "https://api.cloudflare.com/client/v4/accounts/{s}/workers/scripts/{s}/subdomain";
-    _ = get_subdomain;
 
     // stdout is for the actual output of your application, for example if you
     // are implementing gzip, then only the compressed bytes should be sent to
@@ -36,8 +34,60 @@ pub fn main() !void {
     );
 
     try putNewWorker(allocator, &client, accountid, worker_name);
+    try stdout.print("{s}", .{try getSubdomain(allocator, &client, accountid)});
+    if (!worker_exists)
+        try enableWorker(allocator, &client, accountid, worker_name);
     try bw.flush(); // don't forget to flush!
 }
+
+fn enableWorker(allocator: std.mem.Allocator, client: *std.http.Client, account_id: []const u8, name: []const u8) !void {
+    const enable_script = "https://api.cloudflare.com/client/v4/accounts/{s}/workers/scripts/{s}/subdomain";
+    const url = try std.fmt.allocPrint(allocator, enable_script, .{ account_id, name });
+    defer allocator.free(url);
+    var headers = std.http.Headers.init(allocator);
+    defer headers.deinit();
+    try headers.append("X-Auth-Email", x_auth_email);
+    try headers.append("X-Auth-Key", x_auth_key);
+    try headers.append("Content-Type", "application/json; charset=UTF-8");
+    var req = try client.request(.POST, try std.Uri.parse(url), headers, .{});
+    defer req.deinit();
+
+    const request_payload =
+        \\{ "enabled": true }
+    ;
+    req.transfer_encoding = .{ .content_length = @as(u64, request_payload.len) };
+    try req.start();
+    try req.writeAll(request_payload);
+    try req.finish();
+    try req.wait();
+    if (req.response.status != .ok) {
+        std.debug.print("Status is {}\n", .{req.response.status});
+        return error.RequestFailed;
+    }
+}
+
+/// Gets the subdomain for a worker. Caller owns memory
+fn getSubdomain(allocator: std.mem.Allocator, client: *std.http.Client, account_id: []const u8) ![]const u8 {
+    const get_subdomain = "https://api.cloudflare.com/client/v4/accounts/{s}/workers/subdomain";
+    const url = try std.fmt.allocPrint(allocator, get_subdomain, .{account_id});
+    defer allocator.free(url);
+
+    var headers = std.http.Headers.init(allocator);
+    defer headers.deinit();
+    try headers.append("X-Auth-Email", x_auth_email);
+    try headers.append("X-Auth-Key", x_auth_key);
+    var req = try client.request(.GET, try std.Uri.parse(url), headers, .{});
+    defer req.deinit();
+    try req.start();
+    try req.wait();
+    if (req.response.status != .ok) return error.RequestNotOk;
+    var json_reader = std.json.reader(allocator, req.reader());
+    defer json_reader.deinit();
+    var body = try std.json.parseFromTokenSource(std.json.Value, allocator, &json_reader, .{});
+    defer body.deinit();
+    return try allocator.dupe(u8, body.value.object.get("result").?.object.get("subdomain").?.string);
+}
+
 fn putNewWorker(allocator: std.mem.Allocator, client: *std.http.Client, account_id: []const u8, name: []const u8) !void {
     const put_script = "https://api.cloudflare.com/client/v4/accounts/{s}/workers/scripts/{s}?include_subdomain_availability=true&excludeScript=true";
     const url = try std.fmt.allocPrint(allocator, put_script, .{ account_id, name });