initial public release.

This commit is contained in:
Xiaohan-Tian
2025-08-11 18:37:21 -07:00
commit de51967b49
186 changed files with 32322 additions and 0 deletions
+659
View File
@@ -0,0 +1,659 @@
import React, { useState, useEffect, useRef, useCallback } from 'react';
import { createPortal } from 'react-dom';
import { useProjectStore } from '../stores/projectStore';
import { KGCore } from '../core/KGCore';
import { KGTrack } from '../core/track/KGTrack';
import { KGMidiRegion } from '../core/region/KGMidiRegion';
import TrackInfoPanel from './track/TrackInfoPanel';
import TrackGridPanel from './track/TrackGridPanel';
import PianoRoll from './piano-roll/PianoRoll';
import type { RegionUI } from './interfaces';
import { DEBUG_MODE } from '../constants';
import { useRegionOperations } from '../hooks/useRegionOperations';
import { regionDeleteManager } from '../util/regionDeleteUtil';
interface MainContentProps {
onTrackClick?: () => void;
}
const MainContent: React.FC<MainContentProps> = ({
onTrackClick = () => {} // Default to empty function if not provided
}) => {
const {
tracks,
maxBars,
reorderTracks,
updateTrack,
updateTrackProperties,
timeSignature,
setPlayheadPosition,
clearAllSelections,
setSelectedTrack,
showPianoRoll,
activeRegionId,
setShowPianoRoll,
setActiveRegionId
} = useProjectStore();
// State to store regions
const [regions, setRegions] = useState<RegionUI[]>([]);
// Drag state for track grid highlighting
const [draggedTrackIndex, setDraggedTrackIndex] = useState<number | null>(null);
const [dragOverTrackIndex, setDragOverTrackIndex] = useState<number | null>(null);
// Piano roll state is now managed by the store - removed local state
// Region selection state
const [selectedRegionId, setSelectedRegionId] = useState<string | null>(null);
// Use the region operations hook
const { deleteSelectedRegions } = useRegionOperations({
tracks,
updateTrack,
setRegions,
selectedRegionId,
setSelectedRegionId,
showPianoRoll,
setShowPianoRoll,
activeRegionId,
setActiveRegionId
});
// Register the delete function with the global manager
useEffect(() => {
regionDeleteManager.registerDeleteCallback(deleteSelectedRegions);
// Cleanup on unmount
return () => {
regionDeleteManager.unregisterDeleteCallback();
};
}, [deleteSelectedRegions]);
// Refs to track pending updates for verification
const pendingUpdates = useRef<Map<string, { trackId: string, regionId: string, startBeat: number, length: number }>>(new Map());
// Refs for bar numbers drag functionality
const isDraggingRef = useRef(false);
const barNumbersRef = useRef<HTMLDivElement | null>(null);
// Effect to verify track updates
useEffect(() => {
// Check for pending updates
if (pendingUpdates.current.size > 0) {
// Create a copy of the pending updates
const updates = new Map(pendingUpdates.current);
// Clear pending updates
pendingUpdates.current.clear();
// Check each update
updates.forEach((update, key) => {
const { trackId, regionId, startBeat, length } = update;
// Find the track
const track = tracks.find(t => t.getId().toString() === trackId);
if (track) {
// Find the region
const regions = track.getRegions();
const region = regions.find(r => r.getId() === regionId);
if (region && DEBUG_MODE.MAIN_CONTENT) {
console.log(`Verification - Region ${regionId} in track ${trackId}:`);
console.log(` Expected: startBeat=${startBeat}, length=${length}`);
console.log(` Actual: startBeat=${region.getStartFromBeat()}, length=${region.getLength()}, trackId=${region.getTrackId()}, trackIndex=${region.getTrackIndex()}`);
// Check if the update was successful
const success = region.getStartFromBeat() === startBeat && region.getLength() === length && region.getTrackId() === trackId;
console.log(` Update successful: ${success}`);
}
}
});
}
}, [tracks]);
// Effect to update regions when tracks change
useEffect(() => {
// Create a new array of RegionUI objects based on the current tracks
const updatedRegions: RegionUI[] = [];
// Iterate through all tracks
tracks.forEach(track => {
const trackId = track.getId().toString();
const trackIndex = track.getTrackIndex();
// Iterate through all regions in the track
track.getRegions().forEach(region => {
if (region instanceof KGMidiRegion) {
// Calculate bar number and length from beats
const beatsPerBar = timeSignature.numerator;
const barNumber = Math.floor(region.getStartFromBeat() / beatsPerBar) + 1;
const length = region.getLength() / beatsPerBar;
// Create a RegionUI object
updatedRegions.push({
id: region.getId(),
trackId,
trackIndex,
barNumber,
length,
name: region.getName()
});
}
});
});
// Update the regions state
setRegions(updatedRegions);
}, [tracks, timeSignature]);
// Handle track name edit
const handleTrackNameEdit = (track: KGTrack, newName: string) => {
// Use the command pattern to update track name with undo support
updateTrackProperties(track.getId(), { name: newName });
};
// Handle track reordering
const handleTracksReordered = (fromIndex: number, toIndex: number) => {
// Reorder tracks in the store - this will also update trackIndex in each KGTrack
reorderTracks(fromIndex, toIndex);
// Update regions to match the new track order
setRegions(prevRegions => {
return prevRegions.map(region => {
// If the region belongs to the dragged track, update its trackIndex
if (region.trackIndex === fromIndex) {
return { ...region, trackIndex: toIndex };
}
// If the region belongs to a track that was shifted due to the drag operation
else if (
(fromIndex < toIndex &&
region.trackIndex > fromIndex &&
region.trackIndex <= toIndex)
) {
// Shift up by 1
return { ...region, trackIndex: region.trackIndex - 1 };
}
else if (
(fromIndex > toIndex &&
region.trackIndex < fromIndex &&
region.trackIndex >= toIndex)
) {
// Shift down by 1
return { ...region, trackIndex: region.trackIndex + 1 };
}
// Otherwise leave it unchanged
return region;
});
});
// Update the grid drag state to match
setDraggedTrackIndex(null);
setDragOverTrackIndex(null);
};
// Handle region creation from TrackGridPanel
const handleRegionCreated = (trackIndex: number, regionUI: RegionUI, midiRegion: KGMidiRegion) => {
// Note: The region model is already created by the CreateRegionCommand
// We just need to update the UI state and handle selection
// Get the track for store updates
const track = tracks[trackIndex];
// Update the track in the store to reflect the command changes
updateTrack(track);
// Select the track that contains the new region
setSelectedTrack(track.getId().toString());
// Add the new region to the UI state and select it immediately
setRegions(prevRegions => {
const updatedRegions = [...prevRegions, regionUI];
// Select the region using the updated regions array
selectRegion(regionUI.id, updatedRegions);
// Manually trigger selection sync to ensure UI updates immediately
const { syncSelectionFromCore } = useProjectStore.getState();
syncSelectionFromCore();
// If piano roll is visible, set this region as the active region
if (showPianoRoll) {
setActiveRegionId(regionUI.id);
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Newly created region ${regionUI.id} set as active region in piano roll`);
}
}
return updatedRegions;
});
};
// Handle region updates (resize, move, etc.)
const handleRegionUpdated = (
regionId: string,
updates: Partial<RegionUI>,
expectedModelUpdates?: { startBeat: number, length: number }
) => {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Updating region ${regionId} with:`, updates);
}
// Select the region when it's being updated (resize or move)
selectRegion(regionId);
// Find the region to determine which track to select
const updatedRegion = regions.find(r => r.id === regionId);
if (updatedRegion) {
// Use the updated trackId if available, otherwise use the existing trackId
const trackId = updates.trackId || updatedRegion.trackId;
const track = tracks.find(t => t.getId().toString() === trackId);
if (track) {
setSelectedTrack(track.getId().toString());
}
}
// Update the region in the UI state
setRegions(prevRegions => {
return prevRegions.map(region => {
if (region.id === regionId) {
return { ...region, ...updates };
}
return region;
});
});
// Find the region that was updated
const region = regions.find(r => r.id === regionId);
if (!region) return;
// Check if the track ID is being updated (region moved to different track)
if (updates.trackId && updates.trackId !== region.trackId) {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Region ${regionId} moved from track ${region.trackId} to track ${updates.trackId}`);
}
// Get the original track
const originalTrack = tracks.find(t => t.getId().toString() === region.trackId);
// Get the target track
const targetTrack = tracks.find(t => t.getId().toString() === updates.trackId);
if (originalTrack && targetTrack) {
// Select the target track that now contains the region
setSelectedTrack(targetTrack.getId().toString());
// Update both tracks in the store
updateTrack(originalTrack);
updateTrack(targetTrack);
// Add to pending updates for verification
if (expectedModelUpdates) {
const key = `${updates.trackId}-${regionId}-${Date.now()}`;
pendingUpdates.current.set(key, {
trackId: updates.trackId,
regionId,
startBeat: expectedModelUpdates.startBeat,
length: expectedModelUpdates.length
});
}
}
} else {
// Find the track that contains this region
const track = tracks.find(t => t.getId().toString() === region.trackId);
if (track) {
// Log the track's regions before updating the store
const trackRegions = track.getRegions();
const midiRegion = trackRegions.find(r => r.getId() === regionId) as KGMidiRegion | undefined;
if (midiRegion) {
// If we have expected model updates, use those
if (expectedModelUpdates) {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`MainContent - Expected model updates: startBeat=${expectedModelUpdates.startBeat}, length=${expectedModelUpdates.length}`);
}
// Add to pending updates for verification
const key = `${track.getId()}-${regionId}-${Date.now()}`;
pendingUpdates.current.set(key, {
trackId: track.getId().toString(),
regionId,
startBeat: expectedModelUpdates.startBeat,
length: expectedModelUpdates.length
});
} else {
// Otherwise use the current values (for backward compatibility)
const startBeat = midiRegion.getStartFromBeat();
const length = midiRegion.getLength();
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`MainContent - Region before store update: startBeat=${startBeat}, length=${length}`);
}
// Add to pending updates for verification
const key = `${track.getId()}-${regionId}-${Date.now()}`;
pendingUpdates.current.set(key, {
trackId: track.getId().toString(),
regionId,
startBeat,
length
});
}
}
// Update the track in the store to persist changes
updateTrack(track);
}
}
// If piano roll is visible, set this region as the active region
if (showPianoRoll) {
setActiveRegionId(regionId);
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Updated region ${regionId} set as active region in piano roll`);
}
}
};
// Helper function to select a region (clears previous selections)
const selectRegion = (regionId: string, regionsToSearch?: RegionUI[]) => {
// Clear any existing selections using store method
clearAllSelections();
// Find the region in the UI state (use provided regions or current state)
const regionsToUse = regionsToSearch || regions;
const region = regionsToUse.find(r => r.id === regionId);
if (!region) {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Region not found in UI state: ${regionId}`);
}
return;
}
// Find the track that contains this region
const track = tracks.find(t => t.getId().toString() === region.trackId);
if (!track) {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Track not found for region: ${regionId}`);
}
return;
}
// Find the region in the track's model
const trackRegions = track.getRegions();
const midiRegion = trackRegions.find(r => r.getId() === regionId) as KGMidiRegion | undefined;
if (!midiRegion) {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`MIDI region not found in track model: ${regionId}`);
}
return;
}
// Add the region to KGCore's selection
const core = KGCore.instance();
core.addSelectedItem(midiRegion);
// Update the region's internal selection state
midiRegion.select();
// Set the selected region (this might be redundant now, but keeping for compatibility)
setSelectedRegionId(regionId);
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Selected region: ${regionId} (added to KGCore selection)`);
}
};
// Handle region single click: selection only (no piano roll opening)
const handleRegionClick = (regionId: string) => {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Region clicked in MainContent (selection only): ${regionId}`);
}
// Select the region
selectRegion(regionId);
// Also select the containing track
const region = regions.find(r => r.id === regionId);
if (!region) return;
const track = tracks.find(t => t.getId().toString() === region.trackId);
if (!track) return;
setSelectedTrack(track.getId().toString());
};
// Handle explicit pencil action: select region and open piano roll
const handleOpenPianoRoll = (regionId: string) => {
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Open piano roll via pencil for region: ${regionId}`);
}
// Reuse selection logic
handleRegionClick(regionId);
// Activate and show piano roll
setActiveRegionId(regionId);
setShowPianoRoll(true);
};
// Handle piano roll close
const handlePianoRollClose = () => {
setShowPianoRoll(false);
setActiveRegionId(null);
};
/**
* Add keyboard event listener for region deletion
* Handles Backspace (Windows) and Delete (Mac) keys to delete selected regions
* Only processes deletion when not in the piano roll (piano roll has its own delete handler)
*/
useEffect(() => {
const handleKeyDown = (event: KeyboardEvent) => {
// Skip if user is typing in an input field (including ChatBox)
const target = event.target as HTMLElement;
if (target && (
target.tagName === 'INPUT' ||
target.tagName === 'TEXTAREA' ||
target.contentEditable === 'true' ||
target.hasAttribute('data-chatbox-input') ||
target.closest('.chatbox-input')
)) {
return;
}
// Handle delete key for selected regions (Backspace on Windows, Delete on Mac)
if (event.key === 'Backspace' || event.key === 'Delete') {
// Only handle if we're not in the piano roll (piano roll has its own delete handler)
const isInPianoRoll = document.querySelector('.piano-roll')?.contains(event.target as Node);
const isPianoRollOpen = showPianoRoll;
if (!isInPianoRoll && !isPianoRollOpen) {
const deleted = deleteSelectedRegions();
if (deleted) {
// Prevent default behavior only if regions were actually deleted
event.preventDefault();
}
}
}
};
// Add event listener
window.addEventListener('keydown', handleKeyDown);
// Remove event listener on cleanup
return () => {
window.removeEventListener('keydown', handleKeyDown);
};
}, [deleteSelectedRegions, showPianoRoll]); // Dependencies for the effect
// Utility function to calculate playhead position from mouse coordinates (bar-level snapping)
const calculatePlayheadFromMouse = useCallback((clientX: number): number | null => {
if (!barNumbersRef.current) return null;
const rect = barNumbersRef.current.getBoundingClientRect();
const relativeX = clientX - rect.left;
// Calculate the width of each bar
const barWidth = parseInt(
getComputedStyle(document.documentElement).getPropertyValue('--track-grid-bar-width')
) || 40;
// Find the closest bar start (using Math.round for nearest bar)
const barIndex = Math.round(relativeX / barWidth);
// Ensure we don't go below 0
const clampedBarIndex = Math.max(0, barIndex);
// Calculate destination beat position (start of the bar)
const beatsPerBar = timeSignature.numerator;
const destinationBeatPosition = clampedBarIndex * beatsPerBar;
return destinationBeatPosition;
}, [timeSignature]);
// Handle mouse down to start dragging
const handleBarNumbersMouseDown = (e: React.MouseEvent<HTMLDivElement>) => {
// Only handle left mouse button
if (e.button !== 0) return;
isDraggingRef.current = true;
// Calculate and set initial playhead position
const newPosition = calculatePlayheadFromMouse(e.clientX);
if (newPosition !== null) {
setPlayheadPosition(newPosition);
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Bar numbers drag started - Initial position: ${newPosition} (bar ${Math.floor(newPosition / timeSignature.numerator) + 1})`);
}
}
// Prevent text selection during drag
e.preventDefault();
};
// Handle click on bar numbers to move playhead (when not dragging)
const handleBarNumbersClick = (e: React.MouseEvent<HTMLDivElement>) => {
// If we were dragging, don't process as a click
if (isDraggingRef.current) {
return;
}
const newPosition = calculatePlayheadFromMouse(e.clientX);
if (newPosition !== null) {
const core = KGCore.instance();
const currentPlayheadPosition = core.getPlayheadPosition();
const beatsPerBar = timeSignature.numerator;
const currentBarNumber = Math.floor(currentPlayheadPosition / beatsPerBar) + 1; // 1-indexed
const destinationBarNumber = Math.floor(newPosition / beatsPerBar) + 1; // 1-indexed
// Debug logging
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Bar numbers click - Position: ${newPosition}`);
console.log(`Current bar: ${currentBarNumber} (beat ${currentPlayheadPosition})`);
console.log(`Destination bar: ${destinationBarNumber} (beat ${newPosition})`);
}
setPlayheadPosition(newPosition);
}
};
// Global mouse move and mouse up handlers for bar numbers drag functionality
useEffect(() => {
const handleMouseMove = (e: MouseEvent) => {
if (!isDraggingRef.current) return;
const newPosition = calculatePlayheadFromMouse(e.clientX);
if (newPosition !== null) {
setPlayheadPosition(newPosition);
if (DEBUG_MODE.MAIN_CONTENT) {
console.log(`Bar numbers drag - Position: ${newPosition} (bar ${Math.floor(newPosition / timeSignature.numerator) + 1})`);
}
}
};
const handleMouseUp = () => {
if (isDraggingRef.current) {
isDraggingRef.current = false;
if (DEBUG_MODE.MAIN_CONTENT) {
console.log('Bar numbers drag ended');
}
}
};
// Add global event listeners for drag functionality
document.addEventListener('mousemove', handleMouseMove);
document.addEventListener('mouseup', handleMouseUp);
// Cleanup event listeners on unmount
return () => {
document.removeEventListener('mousemove', handleMouseMove);
document.removeEventListener('mouseup', handleMouseUp);
};
}, [calculatePlayheadFromMouse, setPlayheadPosition, timeSignature]);
const { showInstrumentSelection } = useProjectStore();
return (
<div className={`main-content${showInstrumentSelection ? ' has-left-instrument' : ''}`}>
<div className="main-content-wrapper">
{/* Top-left spacer */}
<div className="top-left-spacer"></div>
{/* Bar numbers at the top */}
<div
className="bar-numbers"
ref={barNumbersRef}
onMouseDown={handleBarNumbersMouseDown}
onClick={handleBarNumbersClick}
>
{Array.from({ length: maxBars }, (_, i) => (
<div key={i} className="bar-number-cell">{i + 1}</div>
))}
</div>
<div className="main-content-body">
{/* Fixed left panel with track info */}
<TrackInfoPanel
tracks={tracks}
onTrackClick={onTrackClick}
onTrackNameEdit={handleTrackNameEdit}
onTracksReordered={handleTracksReordered}
/>
{/* Scrollable grid area */}
<TrackGridPanel
tracks={tracks}
regions={regions}
maxBars={maxBars}
timeSignature={timeSignature}
draggedTrackIndex={draggedTrackIndex}
dragOverTrackIndex={dragOverTrackIndex}
selectedRegionId={selectedRegionId}
onRegionCreated={handleRegionCreated}
onRegionUpdated={handleRegionUpdated}
onRegionClick={handleRegionClick}
onOpenPianoRoll={handleOpenPianoRoll}
/>
</div>
</div>
{/* Piano Roll - render using portal */}
{showPianoRoll && createPortal(
<PianoRoll
onClose={handlePianoRollClose}
regionId={activeRegionId}
/>,
document.body
)}
</div>
);
};
export default MainContent;