Skip to content

feat: video export #11

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
122 changes: 122 additions & 0 deletions src/app/recorder.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,122 @@
'use client';

import { useStopWatch } from '@/hooks/use-stop-watch';
import { useState, useRef, type JSX } from 'react';
import { toast } from 'sonner';

type RecordingControls = {
stop: () => void;
};

function record(canvasRef: React.RefObject<HTMLCanvasElement | null>): RecordingControls | null {
const canvas = canvasRef.current;
console.log('Attempting to record canvas:', canvas);
if (!canvas) {
console.warn('No canvas element found');
return null;
}

try {
console.log('Capturing stream from canvas');
const stream = canvas.captureStream(30);
console.log('Stream obtained:', stream);
const mediaRecorder = new MediaRecorder(stream, {
mimeType: 'video/webm;codecs=vp9',
videoBitsPerSecond: 2_500_000
});

const chunks: Blob[] = [];

mediaRecorder.ondataavailable = (event) => {
if (event.data.size > 0) chunks.push(event.data);
};

mediaRecorder.onstop = () => {
const blob = new Blob(chunks, { type: 'video/webm' });
const url = URL.createObjectURL(blob);
const a = document.createElement('a');
a.href = url;
a.download = `recording-${Date.now()}.webm`;
a.click();
URL.revokeObjectURL(url);
};

mediaRecorder.start(1000);
return { stop: () => mediaRecorder.stop() };
} catch (error) {
toast.error(`Recording failed: ${error}`);
return null;
}
}

interface RecorderProps {
className?: string;
canvasRef: React.RefObject<HTMLCanvasElement | null>;
}

function Recorder({ className, canvasRef }: RecorderProps): JSX.Element {
const [isRecording, setIsRecording] = useState(false);
const { timeElapsed, startStopWatch, stopStopWatch } = useStopWatch();
const recordingRef = useRef<RecordingControls | null>(null);

const toggleRecording = () => {
console.log(canvasRef.current);
if (!canvasRef.current) return;

if (isRecording) {
stopStopWatch();
recordingRef.current?.stop();
} else {
startStopWatch();
recordingRef.current = record(canvasRef);
}
setIsRecording(!isRecording);
};

const formatTime = (seconds: number): string => {
const minutes = Math.floor(seconds / 60);
const remainingSeconds = seconds % 60;
return `${minutes.toString().padStart(2, '0')}:${remainingSeconds
.toString()
.padStart(2, '0')}`;
};

return (
<div className={`flex items-center gap-20 overflow-hidden ${className || ''}`}>
<div className={`text-2xl font-mono font-bold text-white w-40 ${isRecording ? 'visible' : 'invisible'}`}>
{formatTime(timeElapsed)}
</div>
<RecordButton isRecording={isRecording} toggleRecording={toggleRecording} />
</div>
);
}

function RecordButton({
isRecording,
toggleRecording,
}: {
isRecording: boolean;
toggleRecording: () => void;
}): JSX.Element {
return (
<button
className="relative flex h-35 w-35 items-center justify-center rounded-full bg-white transition-all duration-300"
onClick={toggleRecording}
aria-label={isRecording ? 'Stop recording' : 'Start recording'}
>
<div
className={`flex h-[calc(100%-4px)] w-[calc(100%-4px)] items-center justify-center rounded-full bg-button transition-all duration-300 ${
isRecording ? 'scale-[0.6]' : ''
}`}
></div>
{isRecording && (
<div className="absolute -top-1 -right-1 h-2 w-2">
<div className="bg-red-500 animate-ping absolute h-full w-full rounded-full opacity-75"></div>
<div className="bg-red-500 absolute h-full w-full rounded-full"></div>
</div>
)}
</button>
);
}

export default Recorder;
11 changes: 9 additions & 2 deletions src/hero/canvas.tsx
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
'use client';

import { liquidFragSource } from '@/app/hero/liquid-frag';
import { useEffect, useRef, useState } from 'react';
import { useEffect, useRef, useState, type Ref, type RefObject} from 'react';
import { toast } from 'sonner';

// uniform sampler2D u_image_texture;
// uniform float u_time;
// uniform float u_ratio;
Expand Down Expand Up @@ -38,10 +37,12 @@ export function Canvas({
imageData,
params,
processing,
ref
}: {
imageData: ImageData;
params: ShaderParams;
processing: boolean;
ref: RefObject<HTMLCanvasElement | null>
}) {
const canvasRef = useRef<HTMLCanvasElement>(null);
const [gl, setGl] = useState<WebGL2RenderingContext | null>(null);
Expand Down Expand Up @@ -251,5 +252,11 @@ export function Canvas({
};
}, [gl, uniforms, imageData]);

useEffect(() => {
if(ref && canvasRef.current){
ref.current = canvasRef.current;
}
},[canvasRef,ref])

return <canvas ref={canvasRef} className="block h-full w-full object-contain" />;
}
38 changes: 23 additions & 15 deletions src/hero/hero.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import { toast } from 'sonner';
import { parseLogoImage } from './parse-logo-image';
import { uploadImage } from '@/hero/upload-image';
import isEqual from 'lodash-es/isEqual';
import Recorder from '@/app/recorder';

interface HeroProps {
imageId: string;
Expand All @@ -35,8 +36,10 @@ export function Hero({ imageId }: HeroProps) {
const [imageData, setImageData] = useState<ImageData | null>(null);
const [processing, setProcessing] = useState<boolean>(true);

const canvasRef = useRef<HTMLCanvasElement>(null);
// Check URL for image ID on mount
useEffect(() => {
console.log(canvasRef);
setProcessing(true);

async function updateImageData() {
Expand All @@ -61,7 +64,7 @@ export function Hero({ imageId }: HeroProps) {
}

updateImageData();
}, [imageId]);
}, [imageId, canvasRef]);

useEffect(() => {
stateRef.current = state;
Expand Down Expand Up @@ -189,20 +192,25 @@ export function Hero({ imageId }: HeroProps) {
handleFiles(files);
}}
>
<div
className="flex aspect-square w-full items-center justify-center rounded-10"
style={{
background: (() => {
switch (state.background) {
case 'metal':
return 'linear-gradient(to bottom, #eee, #b8b8b8)';
}
return state.background;
})(),
}}
>
<div className="aspect-square w-400">
{imageData && <Canvas imageData={imageData} params={state} processing={processing} />}
<div className='flex flex-col relative h-full'>
<div className='-top-45 right-0 absolute'>
<Recorder canvasRef={canvasRef} className=""></Recorder>
</div>
<div
className="flex aspect-square w-full items-center justify-center rounded-10 h-full"
style={{
background: (() => {
switch (state.background) {
case 'metal':
return 'linear-gradient(to bottom, #eee, #b8b8b8)';
}
return state.background;
})(),
}}
>
<div className="aspect-square w-400">
{imageData && <Canvas ref={canvasRef} imageData={imageData} params={state} processing={processing} />}
</div>
</div>
</div>

Expand Down
51 changes: 51 additions & 0 deletions src/hooks/use-stop-watch.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,51 @@
import { useCallback, useEffect, useRef, useState } from "react";

interface UseStopWatchResult {
timeElapsed: number;
isRunning: boolean;
startStopWatch: () => void;
stopStopWatch: () => void;
}

export function useStopWatch() {
const [timeElapsed, setTimeElapsed] = useState<number>(0);
const [isRunning, setIsRunning] = useState<boolean>(false);
const rafRef = useRef<number | null>(null);
const startTimeRef = useRef<number | null>(null);

const updateTime = useCallback(() => {
if(startTimeRef.current !== null) {
const elapsedMS = Date.now() - startTimeRef.current;
setTimeElapsed(Math.floor(elapsedMS / 1000));
}
setTimeElapsed((prev) => prev + 1);
rafRef.current = requestAnimationFrame(updateTime);
},[])

const startStopWatch = useCallback(() => {
if (typeof window === "undefined") return;
if(!isRunning) {
setIsRunning(true);
rafRef.current = requestAnimationFrame(updateTime);
startTimeRef.current = Date.now();
}
}, [isRunning, updateTime])

const stopStopWatch = useCallback(() => {
if(rafRef.current !== null) {
cancelAnimationFrame(rafRef.current);
rafRef.current = null;
setIsRunning(false);
}
}, []);

useEffect(() => {
return () => {
if(rafRef.current !== null){
cancelAnimationFrame(rafRef.current);
}
}
},[])

return {timeElapsed, isRunning,startStopWatch, stopStopWatch}
}