Merge branch '18874-merge-wb2'

[arvados.git] / services / workbench2 / src / services / api / order-builder.ts

diff --git a/services/workbench2/src/services/api/order-builder.ts b/services/workbench2/src/services/api/order-builder.ts
new file mode 100644 (file)
index 0000000..3fc4900
--- /dev/null
+++ b/services/workbench2/src/services/api/order-builder.ts
@@ -0,0 +1,30 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
+import { snakeCase } from "lodash";
+import { Resource } from "models/resource";
+
+export enum OrderDirection { ASC, DESC }
+
+export class OrderBuilder<T extends Resource = Resource> {
+
+    constructor(private order: string[] = []) {}
+
+    addOrder(direction: OrderDirection, attribute: keyof T, prefix?: string) {
+        this.order.push(`${prefix ? prefix + "." : ""}${snakeCase(attribute.toString())} ${direction === OrderDirection.ASC ? "asc" : "desc"}`);
+        return this;
+    }
+
+    addAsc(attribute: keyof T, prefix?: string) {
+        return this.addOrder(OrderDirection.ASC, attribute, prefix);
+    }
+
+    addDesc(attribute: keyof T, prefix?: string) {
+        return this.addOrder(OrderDirection.DESC, attribute, prefix);
+    }
+
+    getOrder() {
+        return this.order.join(",");
+    }
+}