FEATURE NMH-168: Hide licensed models from spark-nlp search [skip-test]

JohnSnowLabs · Apr 7, 2023 · 34afcc5 · 34afcc5
1 parent 0e60c0b
commit 34afcc5
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/.github/workflows/create_search_index.yml b/.github/workflows/create_search_index.yml
@@ -75,6 +75,7 @@ jobs:
         working-directory: docs/_frontend
         env:
           SEARCH_ORIGIN: ${{ secrets.SEARCH_ORIGIN }}
+          ORIGIN: ${{ secrets.ORIGIN }}
         run: |
           yarn
           yarn run build

diff --git a/docs/_frontend/src/models/App/App.js b/docs/_frontend/src/models/App/App.js
@@ -116,7 +116,7 @@ const useFilterQuery = () => {
     window.history.pushState(
       nextParams,
       window.title,
-      '/models' + toSearchString(nextParams)
+      '/models' + toSearchString(nextParams, false)
     );
     setParams(nextParams);
   };

diff --git a/docs/_frontend/src/models/common.js b/docs/_frontend/src/models/common.js
@@ -1,6 +1,7 @@
 export const SEARCH_ORIGIN = process.env.SEARCH_ORIGIN;
+export const ORIGIN = process.env.ORIGIN;
 
-export const toSearchString = (params) => {
+export const toSearchString = (params, includeOrigin = true) => {
   const searchParams = Object.keys(params).reduce((acc, k) => {
     if (params[k]) {
       switch (k) {
@@ -34,6 +35,9 @@ export const toSearchString = (params) => {
     return acc;
   }, new URLSearchParams());
 
+  if (includeOrigin && ORIGIN) {
+    searchParams.append('origin', ORIGIN);
+  }
   const search = searchParams.toString();
   return search ? '?' + search : '';
 };
diff --git a/docs/_frontend/webpack.config.js b/docs/_frontend/webpack.config.js
@@ -111,6 +111,7 @@ module.exports = (env, argv) => {
         SEARCH_ORIGIN:
           process.env.SEARCH_ORIGIN ||
           'https://search.modelshub.johnsnowlabs.com',
+        ORIGIN: process.env.ORIGIN,
       }),
       isProduction &&
         new MiniCssExtractPlugin({