-- SQL Script to add missing columns to Locations table
-- Run this in SQL Server Management Studio

USE [db_a7141e_seahavenindustry]
GO

-- Check and add AccountId column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'AccountId')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [AccountId] int NULL;
    PRINT 'Added AccountId column';
END
ELSE
    PRINT 'AccountId column already exists';

-- Check and add Address column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'Address')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [Address] nvarchar(max) NULL;
    PRINT 'Added Address column';
END
ELSE
    PRINT 'Address column already exists';

-- Check and add City column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'City')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [City] nvarchar(max) NULL;
    PRINT 'Added City column';
END
ELSE
    PRINT 'City column already exists';

-- Check and add Contact column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'Contact')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [Contact] nvarchar(max) NULL;
    PRINT 'Added Contact column';
END
ELSE
    PRINT 'Contact column already exists';

-- Check and add ContactEmail column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'ContactEmail')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [ContactEmail] nvarchar(max) NULL;
    PRINT 'Added ContactEmail column';
END
ELSE
    PRINT 'ContactEmail column already exists';

-- Check and add Phone column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'Phone')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [Phone] nvarchar(max) NULL;
    PRINT 'Added Phone column';
END
ELSE
    PRINT 'Phone column already exists';

-- Check and add State column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'State')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [State] nvarchar(max) NULL;
    PRINT 'Added State column';
END
ELSE
    PRINT 'State column already exists';

-- Check and add Status column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'Status')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [Status] nvarchar(max) NULL;
    PRINT 'Added Status column';
END
ELSE
    PRINT 'Status column already exists';

-- Check and add ZipCode column
IF NOT EXISTS (SELECT * FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_NAME = 'Locations' AND COLUMN_NAME = 'ZipCode')
BEGIN
    ALTER TABLE [dbo].[Locations] ADD [ZipCode] nvarchar(max) NULL;
    PRINT 'Added ZipCode column';
END
ELSE
    PRINT 'ZipCode column already exists';

-- Check and add foreign key index
IF NOT EXISTS (SELECT * FROM sys.indexes WHERE name = 'IX_Locations_AccountId' AND object_id = OBJECT_ID('dbo.Locations'))
BEGIN
    CREATE NONCLUSTERED INDEX [IX_Locations_AccountId] ON [dbo].[Locations] ([AccountId] ASC);
    PRINT 'Added IX_Locations_AccountId index';
END
ELSE
    PRINT 'IX_Locations_AccountId index already exists';

-- Check and add foreign key constraint if Accounts table exists
IF EXISTS (SELECT * FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_NAME = 'Accounts')
    AND NOT EXISTS (SELECT * FROM sys.foreign_keys WHERE name = 'FK_Locations_Accounts_AccountId')
BEGIN
    ALTER TABLE [dbo].[Locations] WITH CHECK ADD CONSTRAINT [FK_Locations_Accounts_AccountId] 
        FOREIGN KEY([AccountId]) REFERENCES [dbo].[Accounts] ([Id]);
    PRINT 'Added FK_Locations_Accounts_AccountId foreign key';
END
ELSE
    PRINT 'FK_Locations_Accounts_AccountId foreign key already exists or Accounts table does not exist';

PRINT 'Schema fix completed!';
GO