mirror of
https://github.com/bellingcat/gesara-entity-viz.git
synced 2026-06-12 21:48:33 +03:00
initial commit
This commit is contained in:
102
src/views/SearchField.tsx
Normal file
102
src/views/SearchField.tsx
Normal file
@@ -0,0 +1,102 @@
|
||||
import React, { KeyboardEvent, ChangeEvent, FC, useEffect, useState } from "react";
|
||||
import { useSigma } from "react-sigma-v2";
|
||||
import { Attributes } from "graphology-types";
|
||||
import { BsSearch } from "react-icons/bs";
|
||||
|
||||
import { FiltersState } from "../types";
|
||||
|
||||
/**
|
||||
* This component is basically a fork from React-sigma-v2's SearchControl
|
||||
* component, to get some minor adjustments:
|
||||
* 1. We need to hide hidden nodes from results
|
||||
* 2. We need custom markup
|
||||
*/
|
||||
const SearchField: FC<{ filters: FiltersState }> = ({ filters }) => {
|
||||
const sigma = useSigma();
|
||||
|
||||
const [search, setSearch] = useState<string>("");
|
||||
const [values, setValues] = useState<Array<{ id: string; label: string }>>([]);
|
||||
const [selected, setSelected] = useState<string | null>(null);
|
||||
|
||||
const refreshValues = () => {
|
||||
const newValues: Array<{ id: string; label: string }> = [];
|
||||
const lcSearch = search.toLowerCase();
|
||||
if (!selected && search.length > 1) {
|
||||
sigma.getGraph().forEachNode((key: string, attributes: Attributes): void => {
|
||||
if (!attributes.hidden && attributes.label && attributes.label.toLowerCase().indexOf(lcSearch) === 0)
|
||||
newValues.push({ id: key, label: attributes.label });
|
||||
});
|
||||
}
|
||||
setValues(newValues);
|
||||
};
|
||||
|
||||
// Refresh values when search is updated:
|
||||
useEffect(() => refreshValues(), [search]);
|
||||
|
||||
// Refresh values when filters are updated (but wait a frame first):
|
||||
useEffect(() => {
|
||||
requestAnimationFrame(refreshValues);
|
||||
}, [filters]);
|
||||
|
||||
useEffect(() => {
|
||||
if (!selected) return;
|
||||
|
||||
sigma.getGraph().setNodeAttribute(selected, "highlighted", true);
|
||||
const nodeDisplayData = sigma.getNodeDisplayData(selected);
|
||||
|
||||
if (nodeDisplayData)
|
||||
sigma.getCamera().animate(
|
||||
{ ...nodeDisplayData, ratio: 0.05 },
|
||||
{
|
||||
duration: 600,
|
||||
},
|
||||
);
|
||||
|
||||
return () => {
|
||||
sigma.getGraph().setNodeAttribute(selected, "highlighted", false);
|
||||
};
|
||||
}, [selected]);
|
||||
|
||||
const onInputChange = (e: ChangeEvent<HTMLInputElement>) => {
|
||||
const searchString = e.target.value;
|
||||
const valueItem = values.find((value) => value.label === searchString);
|
||||
if (valueItem) {
|
||||
setSearch(valueItem.label);
|
||||
setValues([]);
|
||||
setSelected(valueItem.id);
|
||||
} else {
|
||||
setSelected(null);
|
||||
setSearch(searchString);
|
||||
}
|
||||
};
|
||||
|
||||
const onKeyPress = (e: KeyboardEvent<HTMLInputElement>) => {
|
||||
if (e.key === "Enter" && values.length) {
|
||||
setSearch(values[0].label);
|
||||
setSelected(values[0].id);
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="search-wrapper">
|
||||
<input
|
||||
type="search"
|
||||
placeholder="Search in nodes..."
|
||||
list="nodes"
|
||||
value={search}
|
||||
onChange={onInputChange}
|
||||
onKeyPress={onKeyPress}
|
||||
/>
|
||||
<BsSearch className="icon" />
|
||||
<datalist id="nodes">
|
||||
{values.map((value: { id: string; label: string }) => (
|
||||
<option key={value.id} value={value.label}>
|
||||
{value.label}
|
||||
</option>
|
||||
))}
|
||||
</datalist>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
export default SearchField;
|
||||
Reference in New Issue
Block a user