-- Настройки торговой стратегии
local class = "QJSIM" 
local sec = "SBER" 
local account = "NL0011100043" 
local LotSize = 1 
local step = 0.02 --> Нужно проверить не уверен что так?
local limits = 10 

-- Служебные переменные
local tid = 0 
local trz_comment = "FIBO-2 --> " 
local uprice = nil
local dprice = nil
local TradeNums = {}
local do_main = true
local climits = 0

-- Пути к файлам
local csv_path = getScriptPath().."/FiboTrades.csv"
local secondfile = getScriptPath().."/fibobuys.csv"
local CSV = nil

function OnInit()
    CSV = io.open(csv_path, "a+")
end

-- Округление
function math_round(roundIn, roundDig)
    local mul = 10^roundDig
    return math.floor((roundIn * mul) + 0.5) / mul
end

-- Отправка транзакции -->  Проверьте параметры?
function LimitSpot(operation, SpotPrice)
    tid = tid + 1
    local transaction = {
        ["TRANS_ID"]   = tostring(tid),
        ["ACTION"]     = "NEW_ORDER",
        ["CLASSCODE"]  = class,
        ["SECCODE"]    = sec,
        ["OPERATION"]  = operation,
        ["QUANTITY"]   = tostring(LotSize),
        ["PRICE"]      = tostring(SpotPrice),
        ["ACCOUNT"]    = tostring(account),
        ["EXECUTION_CONDITION"] = "PUT_IN_QUEUE",
    }
    local res = sendTransaction(transaction)
    if res ~= '' then
        message(trz_comment..'Transaction Error! '..res)
    else
        message(trz_comment..'Transaction Success! ID = '..tid)
    end
end

-- Проверка файла цен для продажи или покупки
function CheckLine(number)
    local current_line = 1
    local same = 0
    local is_sold = false
    
    local CheckFile = io.open(secondfile, "r")
    if CheckFile then
        for line in CheckFile:lines() do
            local val = tonumber(line)
            if val then
                if val == number then same = same + 1 end
                -- Если цена в файле ниже или равна текущей — продаем по этой фиксации
                if val <= number then 
                    LimitSpot('S', number)
                    is_sold = true
                    break 
                end
            end
            current_line = current_line + 1
        end
        CheckFile:close()
    end

    -- Если совпадений нет, лимит не исчерпан и мы не продали — покупаем
    if not is_sold and same == 0 and climits < limits then 
        LimitSpot('B', number) 
    end
end

-- Удаление строки из CSV (безопасное переписывание)
function remove_line_from_csv(filename, line_to_remove)
    local lines = {}
    local input_file = io.open(filename, "r")
    
    if input_file then
        for line in input_file:lines() do
            lines[#lines + 1] = line
        end
        input_file:close()
    end

    local output_file = io.open(filename, "w")
    if not output_file then return end

    for i, line in ipairs(lines) do
        if i ~= line_to_remove then
            output_file:write(line .. "\n")
        end
    end
    output_file:close()
    message(trz_comment.."Строка "..line_to_remove.." удалена из базы покупок.")
end

-- Функция поиска строки по цене для удаления при OnTrade
function find_line_by_price(filename, price_to_find)
    local input_file = io.open(filename, "r")
    if not input_file then return nil end
    
    local line_number = 1
    for line in input_file:lines() do
        if tonumber(line) == price_to_find then
            input_file:close()
            return line_number
        end
        line_number = line_number + 1
    end
    input_file:close()
    return nil
end

function main()
    message(trz_comment..'Start.')
    while do_main do
        sleep(1000)
    end
end

function OnStop()
    do_main = false
    if CSV then CSV:close() end
    return 1000
end

function OnParam(class1, sec1)
    if class1 == class and sec1 == sec then
        local last_param = getParamEx(class, sec, "last")
        if not last_param or last_param.param_value == "" then return end
        
        local price = math_round(tonumber(last_param.param_value), 2)
        
        if uprice ~= nil then
            if price <= dprice then
                dprice = dprice - step
                uprice = uprice - step
                message(trz_comment..'Price-- -> '..price)
            elseif price >= uprice then
                CheckLine(price)
                dprice = dprice + step
                uprice = uprice + step
                message(trz_comment.."Price++ -> "..price)
            end
        end
    end
end

function OnTrade(trade)
    if trade.sec_code == sec and trade.class_code == class then
        -- Инициализация сетки по первой сделке
        if uprice == nil then
            uprice = trade.price + step
            dprice = trade.price - step
            message(trz_comment..'Grid placed success!')
        end

        -- Проверка на дубликаты сделок
        for i = #TradeNums, 1, -1 do
            if TradeNums[i] == trade.trade_num then return end
        end
        TradeNums[#TradeNums + 1] = trade.trade_num

        local Op = ""
        local t = bit.band(tonumber(trade.flags), 4) -- 4 = sell, 0 = buy
        
        if t == 0 then 
            -- ЛОГИКА ПОКУПКИ
            Op = "Buy"
            climits = climits + 1
            message(trz_comment.."buy "..trade.price)
            
            local wr = io.open(secondfile, 'a+')
            if wr then
                wr:write(trade.price.."\n")
                wr:flush()
                wr:close()
            end
        else 
            -- ЛОГИКА ПРОДАЖИ
            Op = "Sell"
            climits = climits - 1
            message(trz_comment.."Sell "..trade.price)
            
            -- Ищем строку с ценой, которая послужила триггером для продажи
            local line_idx = find_line_by_price(secondfile, trade.price)
            if line_idx then
                remove_line_from_csv(secondfile, line_idx)
            else
                -- Если точной цены нет, удаляем первую попавшуюся строку (FIFO)
                remove_line_from_csv(secondfile, 1)
            end
        end

        -- Логирование в общий файл FiboTrades.csv
        if CSV then
            local TradeLine = os.date("%x %X", os.time(trade.datetime))..";"..
            trade.class_code..";"..
            trade.sec_code..";"..
            trade.trade_num..";"..
            trade.order_num..";"..
            Op..";"..
            trade.price..";"..
            trade.qty.."\n"
            
            CSV:write(TradeLine)
            CSV:flush()
        end
    end
end