feat: Add kiosk setup and API testing scripts, alongside database function fixes for API key generation and short IDs.

2026-03-15 11:24:26 +00:00 · 2026-01-16 16:23:43 -06:00
parent cf2d8f9b4d
commit 9bb0caaecf
11 changed files with 1099 additions and 0 deletions
--- a/scripts/fix-kiosk-func.js
+++ b/scripts/fix-kiosk-func.js
@@ -0,0 +1,88 @@
+require('dotenv').config({ path: '.env.local' });
+const { createClient } = require('@supabase/supabase-js');
+
+const supabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL;
+const supabaseServiceKey = process.env.SUPABASE_SERVICE_ROLE_KEY;
+
+if (!supabaseUrl || !supabaseServiceKey) {
+  console.error('Missing env vars');
+  process.exit(1);
+}
+
+const supabase = createClient(supabaseUrl, supabaseServiceKey);
+
+async function fixDb() {
+  console.log('Applying DB fix...');
+  
+  const sql = `
+CREATE OR REPLACE FUNCTION generate_kiosk_api_key()
+RETURNS VARCHAR(64) AS $$
+DECLARE
+    chars VARCHAR(62) := 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+    v_api_key VARCHAR(64);
+    attempts INT := 0;
+    max_attempts INT := 10;
+BEGIN
+    LOOP
+        v_api_key := '';
+        FOR i IN 1..64 LOOP
+            v_api_key := v_api_key || substr(chars, floor(random() * 62 + 1)::INT, 1);
+        END LOOP;
+
+        IF NOT EXISTS (SELECT 1 FROM kiosks WHERE api_key = v_api_key) THEN
+            RETURN v_api_key;
+        END IF;
+
+        attempts := attempts + 1;
+        IF attempts >= max_attempts THEN
+            RAISE EXCEPTION 'Failed to generate unique api_key after % attempts', max_attempts;
+        END IF;
+    END LOOP;
+END;
+$$ LANGUAGE plpgsql SECURITY DEFINER;
+  `;
+
+  const { error } = await supabase.rpc('exec_sql', { sql_query: sql });
+  
+  // Wait, I might not have 'exec_sql' RPC function available unless I added it.
+  // Standard Supabase doesn't have it by default.
+  // I can try to use standard pg connection if I have the connection string.
+  // But I only have the URL and Key.
+  // BUT: The error happened in a function that is part of my migration.
+  
+  // Alternative: If I don't have direct SQL execution, I can't easily patch the DB function 
+  // without a migration tool.
+  // However, I can try to see if I can drop/recreate via some available mechanism or 
+  // maybe the 'exec_sql' exists (some starters have it).
+  
+  if (error) {
+    console.error('RPC exec_sql failed (might not exist):', error);
+    
+    // Fallback: If I can't execute SQL, I'm stuck unless I have a way to run migrations.
+    // I noticed `db/migrate.sh` in package.json. Maybe I can run that?
+    // But I don't have `db` folder locally in the listing.
+    // I only have `supabase` folder.
+    
+    console.log('Trying to use direct postgres connection if connection string available...');
+    // I don't have the connection string in .env.example, only the URL.
+    // Usually the URL is http...
+    
+    // Let's assume I CANNOT run raw SQL easily.
+    // BUT I can try to "re-apply" a migration if I had the tool.
+    
+    process.exit(1);
+  } else {
+    console.log('Fix applied successfully!');
+  }
+}
+
+// Actually, let's check if I can use a simpler approach.
+// I can CREATE a new migration file with the fix and ask the user to run it?
+// Or I can use the `postgres` package if I can derive the connection string.
+// But I don't have the DB password.
+// The service role key allows me to use the API as superuser (sort of), but not run DDL 
+// unless I have an RPC for it.
+
+// Let's check if there is an `exec_sql` or similar function. 
+// I'll try to run a simple query.
+fixDb();
--- a/scripts/fix-short-id.sql
+++ b/scripts/fix-short-id.sql
@@ -0,0 +1,26 @@
+-- Fix short_id variable name collision
+CREATE OR REPLACE FUNCTION generate_short_id()
+RETURNS VARCHAR(6) AS $$
+DECLARE
+    chars VARCHAR(36) := '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ';
+    v_short_id VARCHAR(6);
+    attempts INT := 0;
+    max_attempts INT := 10;
+BEGIN
+    LOOP
+        v_short_id := '';
+        FOR i IN 1..6 LOOP
+            v_short_id := v_short_id || substr(chars, floor(random() * 36 + 1)::INT, 1);
+        END LOOP;
+
+        IF NOT EXISTS (SELECT 1 FROM bookings WHERE short_id = v_short_id) THEN
+            RETURN v_short_id;
+        END IF;
+
+        attempts := attempts + 1;
+        IF attempts >= max_attempts THEN
+            RAISE EXCEPTION 'Failed to generate unique short_id after % attempts', max_attempts;
+        END IF;
+    END LOOP;
+END;
+$$ LANGUAGE plpgsql SECURITY DEFINER;
--- a/scripts/setup-kiosk.js
+++ b/scripts/setup-kiosk.js
@@ -0,0 +1,73 @@
+require('dotenv').config({ path: '.env.local' });
+const { createClient } = require('@supabase/supabase-js');
+
+const supabaseUrl = process.env.NEXT_PUBLIC_SUPABASE_URL;
+const supabaseServiceKey = process.env.SUPABASE_SERVICE_ROLE_KEY;
+
+if (!supabaseUrl || !supabaseServiceKey) {
+  console.error('Missing SUPABASE_URL or SUPABASE_SERVICE_ROLE_KEY');
+  process.exit(1);
+}
+
+const supabase = createClient(supabaseUrl, supabaseServiceKey);
+
+async function setupKiosk() {
+  console.log('Setting up Kiosk...');
+  
+  // 1. Get a location
+  const { data: locations, error: locError } = await supabase
+    .from('locations')
+    .select('id, name')
+    .limit(1);
+
+  if (locError || !locations || locations.length === 0) {
+    console.error('Error fetching locations or no locations found:', locError);
+    return;
+  }
+
+  const location = locations[0];
+  console.log(`Using location: ${location.name} (${location.id})`);
+
+  // 2. Check if kiosk exists
+  const { data: existingKiosks, error: kioskError } = await supabase
+    .from('kiosks')
+    .select('id, api_key')
+    .eq('location_id', location.id)
+    .eq('device_name', 'TEST_KIOSK_DEVICE');
+
+  if (existingKiosks && existingKiosks.length > 0) {
+    console.log('Test Kiosk already exists.');
+    console.log('API_KEY=' + existingKiosks[0].api_key);
+    return existingKiosks[0].api_key;
+  }
+
+  // 3. Create Kiosk (Direct Insert to bypass broken SQL function)
+  const chars = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+  let apiKey = '';
+  for (let i = 0; i < 64; i++) {
+    apiKey += chars.charAt(Math.floor(Math.random() * chars.length));
+  }
+
+  const { data: newKiosk, error: createError } = await supabase
+    .from('kiosks')
+    .insert({
+      location_id: location.id,
+      device_name: 'TEST_KIOSK_DEVICE',
+      display_name: 'Test Kiosk Display',
+      api_key: apiKey,
+      ip_address: '127.0.0.1'
+    })
+    .select()
+    .single();
+
+  if (createError) {
+    console.error('Error creating kiosk:', createError);
+    return;
+  }
+
+  console.log('Kiosk created successfully!');
+  console.log('API_KEY=' + newKiosk.api_key);
+  return newKiosk.api_key;
+}
+
+setupKiosk();
--- a/scripts/setup-staff-availability.js
+++ b/scripts/setup-staff-availability.js
@@ -0,0 +1,48 @@
+require('dotenv').config({ path: '.env.local' });
+const { createClient } = require('@supabase/supabase-js');
+
+const supabase = createClient(
+  process.env.NEXT_PUBLIC_SUPABASE_URL,
+  process.env.SUPABASE_SERVICE_ROLE_KEY
+);
+
+async function setupStaffAvailability() {
+  console.log('Setting up staff availability...');
+
+  const locationId = '90d200c5-55dd-4726-bc23-e32ca0c5655b';
+
+  const { data: staff, error: staffError } = await supabase
+    .from('staff')
+    .select('id, display_name')
+    .eq('location_id', locationId)
+    .is('is_active', true);
+
+  if (staffError || !staff || staff.length === 0) {
+    console.error('Error fetching staff:', staffError);
+    return;
+  }
+
+  console.log(`Found ${staff.length} staff members`);
+
+  for (const member of staff) {
+    const { error: updateError } = await supabase
+      .from('staff')
+      .update({
+        work_hours_start: '09:00:00',
+        work_hours_end: '20:00:00',
+        work_days: 'MON,TUE,WED,THU,FRI,SAT',
+        is_available_for_booking: true
+      })
+      .eq('id', member.id);
+
+    if (updateError) {
+      console.error(`Error updating ${member.display_name}:`, updateError);
+    } else {
+      console.log(`✓ Updated ${member.display_name}`);
+    }
+  }
+
+  console.log('\n✅ Staff availability setup complete!');
+}
+
+setupStaffAvailability();
--- a/scripts/test-kiosk-complete.js
+++ b/scripts/test-kiosk-complete.js
@@ -0,0 +1,165 @@
+const API_KEY = process.argv[2];
+const BASE_URL = 'http://localhost:3000';
+
+if (!API_KEY) {
+  console.error('Please provide API KEY as argument');
+  process.exit(1);
+}
+
+require('dotenv').config({ path: '.env.local' });
+const { createClient } = require('@supabase/supabase-js');
+const supabase = createClient(process.env.NEXT_PUBLIC_SUPABASE_URL, process.env.SUPABASE_SERVICE_ROLE_KEY);
+
+async function runTests() {
+  console.log('Starting Complete Kiosk API Tests...\n');
+
+  let createdBooking = null;
+
+  try {
+    // Test 1: Authenticate
+    console.log('--- Test 1: Authenticate ---');
+    const authRes = await fetch(`${BASE_URL}/api/kiosk/authenticate`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ api_key: API_KEY })
+    });
+
+    console.log('Status:', authRes.status);
+    const authData = await authRes.json();
+    console.log('Response:', JSON.stringify(authData, null, 2));
+
+    if (authRes.status !== 200) throw new Error('Authentication failed');
+    console.log('✅ Auth Test Passed\n');
+
+    // Test 2: Get Bookings for today
+    console.log('--- Test 2: Get Bookings for Today ---');
+    const today = new Date().toISOString().split('T')[0];
+    const getBookingsRes = await fetch(
+      `${BASE_URL}/api/kiosk/bookings?date=${today}`,
+      { headers: { 'x-kiosk-api-key': API_KEY } }
+    );
+
+    console.log('Status:', getBookingsRes.status);
+    const getBookingsData = await getBookingsRes.json();
+    console.log(`Found ${getBookingsData.bookings?.length || 0} bookings today`);
+
+    if (getBookingsRes.status !== 200) throw new Error('Get bookings failed');
+    console.log('✅ Get Bookings Test Passed\n');
+
+    // Test 3: Get Available Resources
+    console.log('--- Test 3: Get Available Resources ---');
+    const now = new Date();
+    const oneHourFromNow = new Date(now.getTime() + 60 * 60 * 1000);
+
+    const resourcesRes = await fetch(
+      `${BASE_URL}/api/kiosk/resources/available?start_time=${now.toISOString()}&end_time=${oneHourFromNow.toISOString()}`,
+      { headers: { 'x-kiosk-api-key': API_KEY } }
+    );
+
+    console.log('Status:', resourcesRes.status);
+    const resourcesData = await resourcesRes.json();
+    console.log(`Available resources: ${resourcesData.total_available}`);
+
+    if (resourcesRes.status !== 200) throw new Error('Get resources failed');
+    console.log('✅ Get Resources Test Passed\n');
+
+    // Test 4: Create Booking (Scheduled)
+    console.log('--- Test 4: Create Scheduled Booking ---');
+
+    const { data: services } = await supabase
+      .from('services')
+      .select('id, name, duration_minutes')
+      .eq('is_active', true)
+      .limit(1);
+
+    const { data: staff } = await supabase
+      .from('staff')
+      .select('id, display_name')
+      .eq('is_active', true)
+      .limit(1);
+
+    if (!services || services.length === 0 || !staff || staff.length === 0) {
+      console.error('No services or staff available');
+      return;
+    }
+
+    const scheduledStartTime = new Date();
+    scheduledStartTime.setHours(scheduledStartTime.getHours() + 2);
+
+    const createBookingRes = await fetch(`${BASE_URL}/api/kiosk/bookings`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'x-kiosk-api-key': API_KEY
+      },
+      body: JSON.stringify({
+        customer_email: `scheduled_${Date.now()}@example.com`,
+        customer_name: 'Scheduled Customer',
+        customer_phone: '+525512345678',
+        service_id: services[0].id,
+        staff_id: staff[0].id,
+        start_time_utc: scheduledStartTime.toISOString(),
+        notes: 'Scheduled booking test'
+      })
+    });
+
+    console.log('Status:', createBookingRes.status);
+    const createBookingData = await createBookingRes.json();
+    console.log('Response:', JSON.stringify(createBookingData, null, 2));
+
+    if (createBookingRes.status !== 201) throw new Error('Create booking failed');
+    createdBooking = createBookingData.booking;
+    console.log('✅ Create Booking Test Passed\n');
+
+    // Test 5: Confirm Booking
+    console.log('--- Test 5: Confirm Booking ---');
+    if (createdBooking && createdBooking.short_id) {
+      const confirmRes = await fetch(
+        `${BASE_URL}/api/kiosk/bookings/${createdBooking.short_id}/confirm`,
+        {
+          method: 'POST',
+          headers: { 'x-kiosk-api-key': API_KEY }
+        }
+      );
+
+      console.log('Status:', confirmRes.status);
+      const confirmData = await confirmRes.json();
+      console.log('Response:', JSON.stringify(confirmData, null, 2));
+
+      if (confirmRes.status !== 200) throw new Error('Confirm booking failed');
+      console.log('✅ Confirm Booking Test Passed\n');
+    }
+
+    // Test 6: Walk-in Booking
+    console.log('--- Test 6: Walk-in Booking ---');
+    const walkinRes = await fetch(`${BASE_URL}/api/kiosk/walkin`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'x-kiosk-api-key': API_KEY
+      },
+      body: JSON.stringify({
+        customer_email: `walkin_${Date.now()}@example.com`,
+        customer_name: 'Walk-in Customer',
+        service_id: services[0].id,
+        notes: 'Automated Walk-in Test'
+      })
+    });
+
+    console.log('Status:', walkinRes.status);
+    const walkinData = await walkinRes.json();
+    console.log('Response:', JSON.stringify(walkinData, null, 2));
+
+    if (walkinRes.status !== 201) throw new Error('Walk-in failed');
+    console.log('✅ Walk-in Test Passed\n');
+
+    console.log('✅✅✅ ALL TESTS PASSED ✅✅✅');
+    process.exit(0);
+
+  } catch (e) {
+    console.error('\n❌ Test Failed:', e.message);
+    process.exit(1);
+  }
+}
+
+runTests();
--- a/scripts/test-kiosk.js
+++ b/scripts/test-kiosk.js
@@ -0,0 +1,119 @@
+const API_KEY = process.argv[2];
+const BASE_URL = 'http://localhost:3000';
+
+if (!API_KEY) {
+  console.error('Please provide API KEY as argument');
+  process.exit(1);
+}
+
+async function runTests() {
+  console.log('Starting Kiosk API Tests...');
+
+  // 1. Authenticate
+  console.log('\n--- Test 1: Authenticate ---');
+  try {
+    const authRes = await fetch(`${BASE_URL}/api/kiosk/authenticate`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ api_key: API_KEY })
+    });
+    
+    console.log('Status:', authRes.status);
+    const authData = await authRes.json();
+    console.log('Response:', authData);
+
+    if (authRes.status !== 200) throw new Error('Authentication failed');
+  } catch (e) {
+    console.error('Auth Test Failed:', e);
+    process.exit(1);
+  }
+
+  // 2. Get Resources
+  console.log('\n--- Test 2: Get Available Resources ---');
+  
+  // Generate time window for today
+  const now = new Date();
+  const oneHourFromNow = new Date(now.getTime() + 60 * 60 * 1000);
+  const startTime = now.toISOString();
+  const endTime = oneHourFromNow.toISOString();
+  
+  console.log(`Time window: ${startTime} to ${endTime}`);
+  
+  try {
+    const resRes = await fetch(`${BASE_URL}/api/kiosk/resources/available?start_time=${encodeURIComponent(startTime)}&end_time=${encodeURIComponent(endTime)}`, {
+      headers: { 
+        'x-kiosk-api-key': API_KEY 
+      }
+    });
+    
+    console.log('Status:', resRes.status);
+    const resData = await resRes.json();
+    console.log('Response:', resData);
+
+    if (resRes.status !== 200) throw new Error('Get Resources failed');
+  } catch (e) {
+    console.error('Resources Test Failed:', e);
+    process.exit(1);
+  }
+
+  // 3. Walk-in Booking
+  console.log('\n--- Test 3: Walk-in Booking ---');
+  
+  // Need a service ID. I'll fetch it from the DB via a public endpoint if available, 
+  // or assuming I can't reach DB here easily without duplicating setup logic. 
+  // Wait, I can use the same setup logic or just try a known ID if I knew one. 
+  // But I don't.
+  // I will use `setup-kiosk.js` to also print a service ID or just fetch it here if I use supabase client.
+  // But this script is meant to test HTTP API.
+  // I will assume I can find a service if I query the DB.
+  
+  // Better approach: Since I am running this locally, I can use the supabase client here too just to get a service ID.
+  require('dotenv').config({ path: '.env.local' });
+  const { createClient } = require('@supabase/supabase-js');
+  const supabase = createClient(process.env.NEXT_PUBLIC_SUPABASE_URL, process.env.SUPABASE_SERVICE_ROLE_KEY);
+  
+  const { data: services } = await supabase
+    .from('services')
+    .select('id, name')
+    .limit(1);
+    
+  if (!services || services.length === 0) {
+    console.error('No services found to test walk-in');
+    return;
+  }
+  
+  const serviceId = services[0].id;
+  console.log(`Using Service: ${services[0].name} (${serviceId})`);
+
+  try {
+    const walkinRes = await fetch(`${BASE_URL}/api/kiosk/walkin`, {
+      method: 'POST',
+      headers: { 
+        'Content-Type': 'application/json',
+        'x-kiosk-api-key': API_KEY
+      },
+      body: JSON.stringify({
+        customer_email: `walkin_test_${Date.now()}@example.com`,
+        customer_name: 'Test Walkin',
+        service_id: serviceId,
+        notes: 'Automated Test'
+      })
+    });
+    
+    console.log('Status:', walkinRes.status);
+    const walkinData = await walkinRes.json();
+    console.log('Response:', JSON.stringify(walkinData, null, 2));
+
+    if (walkinRes.status !== 201) throw new Error('Walk-in failed: ' + walkinData.error);
+    
+    console.log('✅ WALK-IN TEST PASSED');
+  } catch (e) {
+    console.error('Walk-in Test Failed:', e);
+    process.exit(1);
+  }
+  
+  console.log('\n✅ ALL TESTS PASSED');
+  process.exit(0);
+}
+
+runTests();