Split up code into single-object files

Author: erbridge

index.js

@@ -1,3 +1,5 @@
-import Client from "./lib/client.js";
-const replicate = new Client({});
+import ReplicateClient from "./lib/ReplicateClient.js";
+
+const replicate = new ReplicateClient({});
+
 export default replicate;

lib/Prediction.js

@@ -0,0 +1,8 @@
+export default class Prediction {
+  static fromJSON(json) {
+    const prediction = new Prediction();
+    Object.assign(prediction, json);
+    // TODO: make prediction.version a Version object
+    return prediction;
+  }
+}

lib/PredictionCollection.js

@@ -0,0 +1,38 @@
+import Prediction from "./Prediction.js";
+
+export default class PredictionCollection {
+  constructor({ client }) {
+    this.client = client;
+  }
+
+  async create({ version, input }) {
+    const response = await this.client._request({
+      method: "POST",
+      path: `/v1/predictions`,
+      body: {
+        version,
+        input,
+      },
+    });
+    return Prediction.fromJSON(response);
+  }
+
+  async get({ id }) {
+    const response = await this.client._request({
+      method: "GET",
+      path: `/v1/predictions/${id}`,
+    });
+    return Prediction.fromJSON(response);
+  }
+
+  async list() {
+    const response = await this.client._request({
+      method: "GET",
+      path: `/v1/predictions`,
+    });
+    // TODO pagination
+    return response.results.map((prediction) =>
+      Prediction.fromJSON(prediction)
+    );
+  }
+}

lib/ReplicateClient.js

@@ -0,0 +1,26 @@
+import fetch from "node-fetch";
+import pkg from "../package.json" assert { type: "json" };
+import PredictionCollection from "./PredictionCollection.js";
+
+export default class ReplicateClient {
+  constructor({ token }) {
+    this.token = token || process.env.REPLICATE_API_TOKEN;
+    this.predictions = new PredictionCollection({ client: this });
+  }
+
+  async _request({ method, path, body }) {
+    const response = await fetch(`https://api.replicate.com${path}`, {
+      method: method,
+      headers: this._headers(),
+      body: JSON.stringify(body),
+    });
+    return await response.json();
+  }
+
+  _headers() {
+    return {
+      Authorization: `Token ${this.token}`,
+      "User-Agent": `replicate-js@${pkg.version}`,
+    };
+  }
+}