local dataBase = dbConnect("sqlite", "vehicles.db" )
local vehiclesTable = dbExec( dataBase, "CREATE TABLE IF NOT EXISTS vehicles( model, x, y, z, rx, ry, rz, r, g, b, owner, code )")

local vehicles = {}

function setVehicleData( player, command, code )
    if (dataBase) then
        if ( isPedInVehicle(player) or vehicles[client] ) then
            local account = getAccountName(getPlayerAccount(player))
            local vehicle = getPedOccupiedVehicle(player)
            local model = getElementModel(vehicle)
            local x, y, z = getElementPosition(vehicle)
            local rx, ry, rz = getElementRotation(vehicle)
            local r, g, b = getVehicleColor(vehicle)
            setElementData(vehicle,"owner",account)
            local query = dbQuery(dataBase, "SELECT * FROM vehicles WHERE owner=?", account)
            local poll = dbPoll(query, -1)

            if (#poll == 0) then
                dbExec(dataBase, "INSERT INTO vehicles( model, x, y, z, rx, ry, rz, r, g, b, owner, code ) VALUES(?,?,?,?,?,?,?,?,?,?,?,?)", model, x, y, z, rx, ry, rz, r, g, b, account, code )
                destroyElement(vehicle)
            else
                local checkCode = dbPoll(dbQuery(dataBase, "SELECT code FROM vehicles WHERE owner=? AND code=?", account,code), -1)
                if ( #checkCode == 0 ) then
                    dbExec(dataBase, "INSERT INTO vehicles( model, x, y, z, rx, ry, rz, r, g, b, owner, code ) VALUES(?,?,?,?,?,?,?,?,?,?,?,?)", model, x, y, z, rx, ry, rz, r, g, b, account, code )
                    destroyElement(vehicle)
                else
                    outputChatBox("The code is already in use.",player,255,255,255)
                end
            end
        end
    else
        outputChatBox("Vehicle System Failure !")
    end
end
addCommandHandler("set",setVehicleData)

function showInformationVehicle(player, command , code)
    if not isGuestAccount (getPlayerAccount(player)) then
        local account = getAccountName(getPlayerAccount(player))
        local pollquery = dbPoll(dbQuery(dataBase, "SELECT owner, model FROM vehicles WHERE owner=? AND code=?", account,code), -1)
        for i=1, #pollquery do
            outputChatBox( "Owner: "..pollquery[i].owner.." | Model: "..pollquery[i].model , player)
        end
    end
end
addCommandHandler("show", showInformationVehicle)

function showInformationVehicle(player)
    if not isGuestAccount (getPlayerAccount(player)) then
        local account = getAccountName(getPlayerAccount(player))
        local pollquery = dbPoll(dbQuery(dataBase, "SELECT owner, model, code FROM vehicles WHERE owner=?", account), -1)
        for i=1, #pollquery do
            outputChatBox( "Owner: "..pollquery[i].owner.." | Model: "..pollquery[i].model.." | Code: "..pollquery[i].code, player)
        end
    end
end
addCommandHandler("showall", showInformationVehicle)

function createPersonalVehicle ( code )
    local account = getAccountName(getPlayerAccount(client))
    local getDetails = dbPoll(dbQuery(dataBase, "SELECT model, x, y, z, rx, ry, rz, r, g, b FROM vehicles WHERE owner=? AND code=?", account,code), -1)
    if isElement(vehicles[client]) then destroyElement(vehicles[client]) end
    for k=1,#getDetails do
        vehicles[client] = createVehicle(getDetails[k].model,getDetails[k].x,getDetails[k].y,getDetails[k].z,getDetails[k].rx,getDetails[k].ry,getDetails[k].rz)
        setElementData(vehicles[client],"hasOwner","yes")
        setElementData(vehicles[client],"code",code)
    end
    --setVehicleColor(vehicles[client],getDetails[k].r,getDetails[k].g,getDetails[k].b)
end
addEvent("onVehicleCreate",true)
addEventHandler("onVehicleCreate",resourceRoot,createPersonalVehicle)

function hidePersonalVehicle ( code )
    if isElement(vehicles[client]) then
        local account = getAccountName(getPlayerAccount(client))
        local x, y, z = getElementPosition(vehicles[client])
        dbExec(dataBase,"UPDATE vehicles SET x=?,y=?,z=? WHERE owner=? AND code=?",x,y,z, account,code)
        destroyElement(vehicles[client])
    end
end
addEvent("onVehicleHide",true)
addEventHandler("onVehicleHide",resourceRoot,hidePersonalVehicle)

function sellPersonalVehicle ( code )
    if isElement(vehicles[client]) then destroyElement(vehicles[client]) end
    local account = getAccountName(getPlayerAccount(client))
    dbExec(dataBase,"DELETE FROM vehicles WHERE owner=? AND code=?", account,code)
end
addEvent("onVehicleSell",true)
addEventHandler("onVehicleSell",resourceRoot,sellPersonalVehicle)

addEvent("onGUIOpenS",true)
addEventHandler("onGUIOpenS",resourceRoot,
function ()
    local account = getAccountName(getPlayerAccount(client))
    local sendDetails = dbPoll(dbQuery(dataBase, "SELECT model,code FROM vehicles WHERE owner=?", account), -1)
    for j=1,#sendDetails do
        triggerClientEvent(client,"onGUIOpenC",resourceRoot,sendDetails[j].model,sendDetails[j].code)
    end
end
)

addEventHandler("onPlayerQuit", resourceRoot,
function( )
    if isElement(vehicles[source]) then
        local account = getAccountName(getPlayerAccount(source))
        local codeCheck = getElementData(vehicles[source],"code")
        local x, y, z = getElementPosition(vehicles[source])
        dbExec(dataBase,"UPDATE vehicles SET x=?,y=?,z=? WHERE owner=? AND code=?",x,y,z, account,codeCheck)
        destroyElement(vehicles[source])
        vehicles[source] = nil
    end
end
)