-- ui
local u_get = ui.get
local u_set = ui.set
-- client
local c_random_int = client.random_int
local c_draw = client.draw_text
-- entity
local get_local_player, get_prop = entity.get_local_player, entity.get_prop
-- globals
local g_tickcount = globals.tickcount

local old_yaw = 0
local mici = false
local deg = 0
local active = false
local _,yaw = ui.reference("AA", "anti-aimbot angles", "yaw")
local slowmotion = ui.reference("AA", "Other", "slow motion")

local sw, sh = client.screen_size()
local final_sw, final_sh = sw / 2, sh - 200

local menu = {
  enable_jitter = ui.new_checkbox("AA", "Other", "Freestand jitter"),
  jitter = ui.new_combobox("AA", "Other", "Trigger", "Always", "Standing", "Moving"),
  jittermode = ui.new_combobox("AA", "Other", "Mode", "Center", "Random", "Z", "Z (2)"),
  range = ui.new_slider("AA", "Other", "Range", 0, 60, 0, true, "°"),
  moving_jittermode = ui.new_combobox("AA", "Other", "Mode", "Center", "Random", "Z", "Z (2)"),
  moving_range = ui.new_slider("AA", "Other", "Range", 0, 60, 0, true, "°")
}

local function handle_menu()
  if ui.get(menu.enable_jitter) == true then
    ui.set_visible(menu.jitter, true)
    if ui.get(menu.jitter) == "Always" then
      ui.set_visible(menu.jittermode, true)
      ui.set_visible(menu.range, true)
      ui.set_visible(menu.moving_jittermode, true)
      ui.set_visible(menu.moving_range, true)
    elseif ui.get(menu.jitter) == "Standing" then
      ui.set_visible(menu.jittermode, true)
      ui.set_visible(menu.range, true)
      ui.set_visible(menu.moving_jittermode, false)
      ui.set_visible(menu.moving_range, false)
      ui.set(menu.moving_range, 0)
    elseif ui.get(menu.jitter) == "Moving" then
      ui.set_visible(menu.jittermode, false)
      ui.set_visible(menu.range, false)
      ui.set(menu.range, 0)
      ui.set_visible(menu.moving_jittermode, true)
      ui.set_visible(menu.moving_range, true)
    end
  else
    ui.set_visible(menu.jitter, false)
    ui.set_visible(menu.jittermode, false)
    ui.set_visible(menu.range, false)
    ui.set_visible(menu.moving_jittermode, false)
    ui.set_visible(menu.moving_range, false)
  end
end

handle_menu()
ui.set_callback(menu.enable_jitter, handle_menu)
ui.set_callback(menu.jitter, handle_menu)

local function on_paint(c)
  if ui.get(menu.enable_jitter) == false then return end

  if old_yaw ~= u_get(yaw) and not mici then
    old_yaw = u_get(yaw)
  end

  local vel_x, vel_y, vel_z = get_prop(get_local_player(), "m_vecVelocity")
  local vel = math.sqrt(vel_x * vel_x + vel_y * vel_y + vel_z * vel_z)

  local rand = c_random_int(-u_get(menu.range), u_get(menu.range))
  local movingrand = c_random_int(-u_get(menu.moving_range), u_get(menu.moving_range))

  local r, g, b, a = nil

  if ui.get(menu.jitter) == "Standing" then
    if ui.get(menu.range) > 0 then
      r, g, b, a = 124, 195, 13, 255
    else
      r, g, b, a = 255, 0, 0, 255
    end
    if vel > 50 then
      r, g, b, a = 255, 0, 0, 255
    end
  elseif ui.get(menu.jitter) == "Moving" then
    if vel > 50 then
      if ui.get(menu.moving_range) > 0 then
        r, g, b, a = 124, 195, 13, 255
      else
        r, g, b, a = 255, 0, 0, 255
      end
    else
      r, g, b, a = 255, 0, 0, 255
    end
  elseif ui.get(menu.jitter) == "Always" then
    if ui.get(menu.range) > 0 then
      r, g, b, a = 124, 195, 13, 255
    else
      r, g, b, a = 255, 0, 0, 255
    end
    if vel > 50 then
      r, g, b, a = 255, 0, 0, 255
    end
    if vel > 50 then
      if ui.get(menu.moving_range) > 0 then
        r, g, b, a = 124, 195, 13, 255
      else
        r, g, b, a = 255, 0, 0, 255
      end
    end
  end

  client.draw_indicator(c, r, g, b, a, "FJ")

  if ui.get(menu.enable_jitter) == true then
    if ui.get(menu.jitter) == "Standing" then end
    if vel > 50 then
      if u_get(menu.moving_jittermode) == "Center" then
        if(u_get(yaw) == u_get(menu.moving_range)) then
          u_set(yaw, -u_get(menu.moving_range))
        else
          u_set(yaw, u_get(menu.moving_range))
        end
      elseif u_get(menu.moving_jittermode) == "Random" then
        u_set(yaw, movingrand)
      elseif u_get(menu.moving_jittermode) == "Z" then
        local tickcount = (g_tickcount()%60)
        u_set(yaw, -u_get(menu.moving_range)+tickcount)
      elseif u_get(menu.moving_jittermode) == "Z (2)" then
        deg = deg + 1
        if(deg == 60) then
          deg = 0
        else
          u_set(yaw, -u_get(menu.moving_range)+deg)
        end
      end
    else
      if ui.get(menu.jitter) == "Moving" then end
      if u_get(menu.jittermode) == "Center" then
        if(u_get(yaw) == u_get(menu.range)) then
          u_set(yaw, -u_get(menu.range))
        else
          u_set(yaw, u_get(menu.range))
        end
      elseif u_get(menu.jittermode) == "Random" then
        u_set(yaw, rand)
      elseif u_get(menu.jittermode) == "Z" then
        local tickcount = (g_tickcount()%60)
        u_set(yaw, -u_get(menu.range)+tickcount)
      elseif u_get(menu.jittermode) == "Z (2)" then
        deg = deg + 1
        if(deg == 60) then
          deg = 0
        else
          u_set(yaw, -u_get(menu.range)+deg)
        end
      end
    end
  end
end

client.set_event_callback("paint", on_paint)