87 lines
2.8 KiB
TypeScript
87 lines
2.8 KiB
TypeScript
|
import { create } from 'zustand';
|
||
|
|
import { immer } from 'zustand/middleware/immer';
|
||
|
|
|
||
|
|
interface AisleStore {
|
||
|
|
aisles: Aisles;
|
||
|
|
setAisles: (aisles: Aisles) => void;
|
||
|
|
addAisle: (aisle: Aisle) => void;
|
||
|
|
updateAisle: (uuid: string, updated: Partial<Aisle>) => void;
|
||
|
|
removeAisle: (uuid: string) => void;
|
||
|
|
setPosition: (pointUuid: string, position: [number, number, number]) => void;
|
||
|
|
setLayer: (pointUuid: string, layer: number) => void;
|
||
|
|
setMaterial: (aisleUuid: string, material: string) => void;
|
||
|
|
setColor: (aisleUuid: string, color: string) => void;
|
||
|
|
setWidth: (aisleUuid: string, width: number) => void;
|
||
|
|
getAisleById: (uuid: string) => Aisle | undefined;
|
||
|
|
}
|
||
|
|
|
||
|
|
export const useAisleStore = create<AisleStore>()(
|
||
|
|
immer((set, get) => ({
|
||
|
|
aisles: [],
|
||
|
|
|
||
|
|
setAisles: (aisles) => set((state) => {
|
||
|
|
state.aisles = aisles;
|
||
|
|
}),
|
||
|
|
|
||
|
|
addAisle: (aisle) => set((state) => {
|
||
|
|
state.aisles.push(aisle);
|
||
|
|
}),
|
||
|
|
|
||
|
|
updateAisle: (uuid, updated) => set((state) => {
|
||
|
|
const aisle = state.aisles.find((a) => a.uuid === uuid);
|
||
|
|
if (aisle) {
|
||
|
|
Object.assign(aisle, updated);
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
removeAisle: (uuid) => set((state) => {
|
||
|
|
state.aisles = state.aisles.filter((a) => a.uuid !== uuid);
|
||
|
|
}),
|
||
|
|
|
||
|
|
setPosition: (pointUuid: string, position: [number, number, number]) => set((state) => {
|
||
|
|
for (const aisle of state.aisles) {
|
||
|
|
const point = aisle.points.find(p => p.uuid === pointUuid);
|
||
|
|
if (point) {
|
||
|
|
point.position = position;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
setLayer: (pointUuid: string, layer: number) => set((state) => {
|
||
|
|
for (const aisle of state.aisles) {
|
||
|
|
const point = aisle.points.find(p => p.uuid === pointUuid);
|
||
|
|
if (point) {
|
||
|
|
point.layer = layer;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
setMaterial: (aisleUuid: string, material: string) => set((state) => {
|
||
|
|
const aisle = state.aisles.find(a => a.uuid === aisleUuid);
|
||
|
|
if (aisle) {
|
||
|
|
aisle.type.material = material;
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
setColor: (aisleUuid: string, color: string) => set((state) => {
|
||
|
|
const aisle = state.aisles.find(a => a.uuid === aisleUuid);
|
||
|
|
if (aisle) {
|
||
|
|
aisle.type.color = color;
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
setWidth: (aisleUuid: string, width: number) => set((state) => {
|
||
|
|
const aisle = state.aisles.find(a => a.uuid === aisleUuid);
|
||
|
|
if (aisle) {
|
||
|
|
aisle.type.width = width;
|
||
|
|
}
|
||
|
|
}),
|
||
|
|
|
||
|
|
getAisleById: (uuid) => {
|
||
|
|
return get().aisles.find((a) => a.uuid === uuid);
|
||
|
|
},
|
||
|
|
}))
|
||
|
|
);
|