vzcode 1.5.0 → 1.6.0

package/src/client/vzReducer/searchReducer.ts

@@ -0,0 +1,114 @@
+import { VZAction, VZState } from '.';
+import {
+  SearchFile,
+  SearchFileVisibility,
+  SearchResult,
+  ShareDBDoc,
+  VZCodeContent,
+} from '../../types';
+
+function searchPattern(
+  shareDBDoc: ShareDBDoc<VZCodeContent>,
+  pattern: string,
+): SearchResult {
+  const files = shareDBDoc.data.files;
+  const fileIds = Object.keys(shareDBDoc.data.files);
+  let results: { [id: string]: SearchFile } = {};
+
+  for (let i = 0; i < fileIds.length; i++) {
+    const file = files[fileIds[i]];
+
+    if (files[fileIds[i]].text) {
+      const fileName = file.name;
+      const lines = file.text.split('\n');
+      const matches = [];
+
+      for (let j = 0; j < lines.length; j++) {
+        const index = lines[j].indexOf(pattern);
+
+        if (index !== -1) {
+          matches.push({
+            line: j + 1,
+            index: index,
+            text: lines[j],
+          });
+        }
+      }
+
+      if (matches.length > 0) {
+        results[fileIds[i]] = {
+          name: fileName,
+          matches: matches,
+          visibility: 'open',
+        };
+      }
+    }
+  }
+
+  return results;
+}
+
+function updateSearchFileVisibility(
+  results: SearchResult,
+  id: string,
+  visibility: SearchFileVisibility,
+): SearchResult {
+  return {
+    ...results,
+    [id]: { ...results[id], visibility: visibility },
+  };
+}
+
+export const setIsSearchOpenReducer = (
+  state: VZState,
+  action: VZAction,
+): VZState =>
+  action.type === 'set_is_search_open'
+    ? { ...state, isSearchOpen: action.value }
+    : state;
+
+export const setSearchReducer = (
+  state: VZState,
+  action: VZAction,
+): VZState =>
+  action.type === 'set_search'
+    ? {
+        ...state,
+        search: { pattern: action.value, results: {} },
+      }
+    : state;
+
+export const setSearchResultsReducer = (
+  state: VZState,
+  action: VZAction,
+): VZState =>
+  action.type === 'set_search_results'
+    ? {
+        ...state,
+        search: {
+          pattern: state.search.pattern,
+          results: searchPattern(
+            action.files,
+            state.search.pattern,
+          ),
+        },
+      }
+    : state;
+
+export const setSearchFileVisibilityReducer = (
+  state: VZState,
+  action: VZAction,
+): VZState =>
+  action.type === 'set_search_file_visibility'
+    ? {
+        ...state,
+        search: {
+          pattern: state.search.pattern,
+          results: updateSearchFileVisibility(
+            state.search.results,
+            action.id,
+            action.visibility,
+          ),
+        },
+      }
+    : state;

package/src/types.ts

@@ -61,6 +61,26 @@ export interface FileTreeFile {
   file: File;
   fileId: FileId;
 }
+export type SearchMatch = Array<{
+  line: number;
+  index: number;
+  text: string;
+}>;
+export type SearchFileVisibility =
+  | 'open'
+  | 'flattened'
+  | 'closed';
+export type SearchResult = { [id: string]: SearchFile };
+
+export interface SearchFile {
+  name: string;
+  matches: SearchMatch;
+  visibility: SearchFileVisibility;
+}
+export interface SearchResults {
+  pattern: string;
+  results: SearchResult;
+}
 
 export type JSONOp = any;