Changes for commercial use (#138)

src-tauri/yaak_license/index.ts

@@ -0,0 +1,33 @@
+import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { invoke } from '@tauri-apps/api/core';
+import { useListenToTauriEvent } from '@yaakapp/app/hooks/useListenToTauriEvent';
+import { LicenseCheckStatus } from './bindings/license';
+
+export * from './bindings/license';
+
+export function useLicense() {
+  const queryClient = useQueryClient();
+  const activate = useMutation<void, string, { licenseKey: string }>({
+    mutationKey: ['license.activate'],
+    mutationFn: (payload) => invoke('plugin:yaak-license|activate', payload),
+    onSuccess: async () => {
+      await queryClient.invalidateQueries({ queryKey: CHECK_QUERY_KEY });
+    },
+  });
+
+  // Check the license again after a license is activated
+  useListenToTauriEvent('license-activated', async () => {
+    await queryClient.invalidateQueries({ queryKey: CHECK_QUERY_KEY });
+  });
+
+  const CHECK_QUERY_KEY = ['license.check'];
+  const check = useQuery<void, string, LicenseCheckStatus>({
+    queryKey: CHECK_QUERY_KEY,
+    queryFn: () => invoke('plugin:yaak-license|check'),
+  });
+
+  return {
+    activate,
+    check,
+  } as const;
+}