From 27f414ca9dadc7430831c7bb608e3ff123ece566 Mon Sep 17 00:00:00 2001
From: Andrew Glago <andrewglago1@gmail.com>
Date: Fri, 21 Jun 2024 16:00:54 +0000
Subject: [PATCH] feat: add helpers for fetching site data

---
 packages/core/src/helpers/convenience.ts |  9 +++++++++
 packages/core/src/helpers/index.ts       |  1 +
 packages/core/src/helpers/site.ts        | 14 ++++++++++++++
 packages/core/src/index.ts               |  1 +
 packages/core/src/lib/gql.ts             | 14 ++++++++++++++
 packages/core/src/types/index.ts         | 10 ++++++++++
 packages/react-sdk/src/hooks/index.ts    |  4 ++++
 packages/react-sdk/src/server/index.ts   |  2 ++
 packages/vue-sdk/src/hooks/index.ts      |  2 ++
 9 files changed, 57 insertions(+)
 create mode 100644 packages/core/src/helpers/site.ts

diff --git a/packages/core/src/helpers/convenience.ts b/packages/core/src/helpers/convenience.ts
index dceade7d..8d3f9c05 100644
--- a/packages/core/src/helpers/convenience.ts
+++ b/packages/core/src/helpers/convenience.ts
@@ -11,6 +11,7 @@ import {
   getArticlesWithSummary,
 } from "./articles";
 import { getAllTags } from "./metadata";
+import { getSite as _getSite } from "./site";
 
 const config = {
   // eslint-disable-next-line turbo/no-undeclared-env-vars
@@ -141,6 +142,13 @@ async function getRecommendedArticles(id: number | string) {
   return article.data.recommendedArticles as Article[];
 }
 
+async function getSite() {
+  const client = buildPantheonClient({ isClientSide: false });
+  const site = await _getSite(client, client.siteId);
+
+  return site;
+}
+
 export const PCCConvenienceFunctions = {
   buildPantheonClient,
   getAllArticles,
@@ -149,4 +157,5 @@ export const PCCConvenienceFunctions = {
   getRecommendedArticles,
   getPaginatedArticles,
   getTags,
+  getSite,
 };
diff --git a/packages/core/src/helpers/index.ts b/packages/core/src/helpers/index.ts
index 6194d6ff..8d7dd2ec 100644
--- a/packages/core/src/helpers/index.ts
+++ b/packages/core/src/helpers/index.ts
@@ -3,3 +3,4 @@ export * from "./convenience";
 export * from "./errors";
 export * from "./metadata";
 export * from "./validator";
+export * from "./site";
diff --git a/packages/core/src/helpers/site.ts b/packages/core/src/helpers/site.ts
new file mode 100644
index 00000000..e281db76
--- /dev/null
+++ b/packages/core/src/helpers/site.ts
@@ -0,0 +1,14 @@
+import { PantheonClient } from "../core/pantheon-client";
+import { GET_SITE_QUERY } from "../lib/gql";
+import { Site } from "../types";
+
+export async function getSite(client: PantheonClient, id: string) {
+  const site = await client.apolloClient.query({
+    query: GET_SITE_QUERY,
+    variables: {
+      id,
+    },
+  });
+
+  return site.data.site as Site;
+}
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
index 58bc7de2..ab6b7aaa 100644
--- a/packages/core/src/index.ts
+++ b/packages/core/src/index.ts
@@ -3,6 +3,7 @@ export * from "./core/pantheon-api";
 
 export * from "./lib/apollo-client";
 export * from "./lib/gql";
+export * as GQL from "./lib/gql";
 export * from "./lib/jwt";
 
 export * from "./helpers";
diff --git a/packages/core/src/lib/gql.ts b/packages/core/src/lib/gql.ts
index 7dd0a58e..821eea49 100644
--- a/packages/core/src/lib/gql.ts
+++ b/packages/core/src/lib/gql.ts
@@ -204,3 +204,17 @@ export const GET_RECOMMENDED_ARTICLES_QUERY = gql`
     }
   }
 `;
+
+export const GET_SITE_QUERY = gql`
+  query GetSite($id: String!) {
+    site(id: $id) {
+      id
+      name
+      url
+      domain
+      contentStructure
+      tags
+      metadataFields
+    }
+  }
+`;
diff --git a/packages/core/src/types/index.ts b/packages/core/src/types/index.ts
index 819c7644..149fcf04 100644
--- a/packages/core/src/types/index.ts
+++ b/packages/core/src/types/index.ts
@@ -48,6 +48,16 @@ export enum SortOrder {
   DESC = "DESC",
 }
 
+export interface Site {
+  id: string;
+  name: string;
+  url: string;
+  domain: string;
+  tags: string[];
+  contentStructure: Record<string, unknown>;
+  metadataFields: Record<string, unknown>;
+}
+
 export interface TreePantheonContent {
   tag: string;
   attrs: {
diff --git a/packages/react-sdk/src/hooks/index.ts b/packages/react-sdk/src/hooks/index.ts
index eec97e08..cea824e3 100644
--- a/packages/react-sdk/src/hooks/index.ts
+++ b/packages/react-sdk/src/hooks/index.ts
@@ -1,3 +1,7 @@
 export { useArticle } from "./use-article";
 export { useArticles } from "./use-articles";
 export { usePagination } from "./use-pagination";
+
+export { useQuery } from "@apollo/client/react/hooks/useQuery.js";
+export { useMutation } from "@apollo/client/react/hooks/useMutation.js";
+export { useSubscription } from "@apollo/client/react/hooks/useSubscription.js";
diff --git a/packages/react-sdk/src/server/index.ts b/packages/react-sdk/src/server/index.ts
index 416413a0..fb905f0b 100644
--- a/packages/react-sdk/src/server/index.ts
+++ b/packages/react-sdk/src/server/index.ts
@@ -13,6 +13,8 @@ export {
   getArticleBySlugOrId,
   PCCConvenienceFunctions,
   updateConfig,
+  GQL,
+  getSite,
 } from "@pantheon-systems/pcc-sdk-core";
 export * from "@pantheon-systems/pcc-sdk-core/types";
 
diff --git a/packages/vue-sdk/src/hooks/index.ts b/packages/vue-sdk/src/hooks/index.ts
index 466831e1..96f3eb9e 100644
--- a/packages/vue-sdk/src/hooks/index.ts
+++ b/packages/vue-sdk/src/hooks/index.ts
@@ -1,2 +1,4 @@
 export { useArticle } from "./use-article";
 export { useArticles } from "./use-articles";
+
+export { useQuery } from "@vue/apollo-composable";