Configurable Redirect URL (#190)

* add callback url config param * changeset
burnt-labs · Jun 25, 2024 · bcc35c9 · bcc35c9
1 parent 7556cda
commit bcc35c9
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 2 deletions.
diff --git a/.changeset/eleven-ladybugs-enjoy.md b/.changeset/eleven-ladybugs-enjoy.md
@@ -0,0 +1,6 @@
+---
+"@burnt-labs/abstraxion-core": minor
+"@burnt-labs/abstraxion": minor
+---
+
+Introduce configurable redirect url param
diff --git a/packages/abstraxion-core/src/AbstraxionAuth.ts b/packages/abstraxion-core/src/AbstraxionAuth.ts
@@ -11,6 +11,7 @@ export class AbstraxionAuth {
   grantContracts?: ContractGrantDescription[];
   stake?: boolean;
   bank?: SpendLimit[];
+  callbackUrl?: string;
 
   // Signer
   private client?: GranteeSignerClient;
@@ -36,19 +37,22 @@ export class AbstraxionAuth {
    * @param {ContractGrantDescription[]} [grantContracts] - Contracts for granting permissions.
    * @param {boolean} [stake] - Indicates whether staking is enabled.
    * @param {SpendLimit[]} [bank] - The spend limits for the user.
+   * @param {string} callbackUrl - preferred callback url to override default
    */
   configureAbstraxionInstance(
     rpc: string,
     restUrl?: string,
     grantContracts?: ContractGrantDescription[],
     stake?: boolean,
     bank?: SpendLimit[],
+    callbackUrl?: string,
   ) {
     this.rpcUrl = rpc;
     this.restUrl = restUrl;
     this.grantContracts = grantContracts;
     this.stake = stake;
     this.bank = bank;
+    this.callbackUrl = callbackUrl;
   }
 
   /**
@@ -235,7 +239,7 @@ export class AbstraxionAuth {
     userAddress: string,
   ): void {
     if (typeof window !== "undefined") {
-      const currentUrl = window.location.href;
+      const currentUrl = this.callbackUrl || window.location.href;
       const urlParams = new URLSearchParams();
 
       if (this.bank) {

diff --git a/packages/abstraxion/src/components/Abstraxion/index.tsx b/packages/abstraxion/src/components/Abstraxion/index.tsx
@@ -65,6 +65,7 @@ export interface AbstraxionConfig {
   restUrl?: string;
   stake?: boolean;
   bank?: SpendLimit[];
+  callbackUrl?: string;
 }
 
 export function AbstraxionProvider({
@@ -81,6 +82,7 @@ export function AbstraxionProvider({
       restUrl={config.restUrl}
       stake={config.stake}
       bank={config.bank}
+      callbackUrl={config.callbackUrl}
     >
       {children}
     </AbstraxionContextProvider>

diff --git a/packages/abstraxion/src/components/AbstraxionContext/index.tsx b/packages/abstraxion/src/components/AbstraxionContext/index.tsx
@@ -47,6 +47,7 @@ export function AbstraxionContextProvider({
   restUrl = testnetChainInfo.rest,
   stake = false,
   bank,
+  callbackUrl,
 }: {
   children: ReactNode;
   contracts?: ContractGrantDescription[];
@@ -55,6 +56,7 @@ export function AbstraxionContextProvider({
   restUrl?: string;
   stake?: boolean;
   bank?: SpendLimit[];
+  callbackUrl?: string;
 }): JSX.Element {
   const [abstraxionError, setAbstraxionError] = useState("");
   const [isConnected, setIsConnected] = useState(false);
@@ -74,8 +76,9 @@ export function AbstraxionContextProvider({
       contracts,
       stake,
       bank,
+      callbackUrl,
     );
-  }, [rpcUrl, restUrl, contracts, stake, bank]);
+  }, [rpcUrl, restUrl, contracts, stake, bank, callbackUrl]);
 
   useEffect(() => {
     const searchParams = new URLSearchParams(window.location.search);