import obspython as obs
import random


# Internal variables
direction_x = random.choice([-1, 1])
direction_y = random.choice([-1, 1])

# Initialize global position
position_x = 0.0
position_y = 0.0

# Global variables holding the values of data settings / properties
source_name = "Spaceship"  # Name of the source to shake
speed_x = 5.0  # Speed of movement in the X direction
speed_y = 3.0  # Speed of movement in the Y direction

# Called on script load
def script_load(settings):
    global position_x, position_y
    source = obs.obs_get_source_by_name(source_name)
    if source is not None:
        scene_item = get_scene_item_from_source(source)
        if scene_item is not None:
            pos = obs.vec2()
            obs.obs_sceneitem_get_pos(scene_item, pos)
            position_x = pos.x
            position_y = pos.y
        obs.obs_source_release(source)

    # Set up timer
    #obs.timer_add(update_position, 16)  # Around 60 FPS

# Function to get scene item
def get_scene_item_from_source(source):
    current_scene = obs.obs_frontend_get_current_scene()
    scene = obs.obs_scene_from_source(current_scene)
    obs.obs_source_release(current_scene)

    scene_item = obs.obs_scene_find_source(scene, source_name)
    return scene_item

# Function to update position
def update_position():
    global position_x, position_y, direction_x, direction_y

    source = obs.obs_get_source_by_name(source_name)
    if source is None:
        return

    scene_item = get_scene_item_from_source(source)
    if scene_item is None:
        obs.obs_source_release(source)
        return

    # Get canvas size
    width = 490 #obs.obs_source_get_width(source)
    height = 275 #obs.obs_source_get_height(source)
    canvas_width = 1920
    canvas_height = 1080
    print(width, height, canvas_width, canvas_height)
    # Calculate new position
    position_x += direction_x * speed_x
    position_y += direction_y * speed_y
    print(position_x, position_y)
    # Bounce off edges
    if position_x <= 0 or position_x + width >= canvas_width:
        direction_x *= -1
    if position_y <= 0 or position_y  + height >= canvas_height:
        direction_y *= -1

    # Apply new position
    pos = obs.vec2()
    pos.x = position_x
    pos.y = position_y
    obs.obs_sceneitem_set_pos(scene_item, pos)

    obs.obs_source_release(source)

# Called to set default values of data settings
def script_defaults(settings):
  obs.obs_data_set_default_string(settings, "source_name", "")
  obs.obs_data_set_default_double(settings, "speed", 2)

# Fills the given list property object with the names of all sources plus an empty one
def populate_list_property_with_source_names(list_property):
  sources = obs.obs_enum_sources()
  obs.obs_property_list_clear(list_property)
  obs.obs_property_list_add_string(list_property, "", "")
  for source in sources:
    name = obs.obs_source_get_name(source)
    obs.obs_property_list_add_string(list_property, name, name)
  obs.source_list_release(sources)


# Called to display the properties GUI
def script_properties():
  props = obs.obs_properties_create()

  # Drop-down list of sources
  list_property = obs.obs_properties_add_list(props, "source_name", "Source name",
                                              obs.OBS_COMBO_TYPE_LIST, obs.OBS_COMBO_FORMAT_STRING)
  populate_list_property_with_source_names(list_property)
  # obs.obs_properties_add_text(props, "source_name", "Source name", obs.OBS_TEXT_DEFAULT)

  # Button to refresh the drop-down list
  obs.obs_properties_add_button(props, "button", "Refresh list of sources",
                                lambda props, prop: True if populate_list_property_with_source_names(list_property) else True)

  obs.obs_properties_add_float_slider(
      props, "speed_x", "Speed in X direction", 0.1, 20, 0.1)
  obs.obs_properties_add_float_slider(
      props, "speed_y", "Speed in Y direction", 0.1, 20, 0.1)
  return props

# Called after change of settings including once after script load


def script_update(settings):
  global source_name, speed_x, speed_y
  source_name = obs.obs_data_get_string(settings, "source_name")
  speed_x = obs.obs_data_get_double(settings, "speed_x")
  speed_y = obs.obs_data_get_double(settings, "speed_y")


# Global animation activity flag
is_active = False

# Called every frame
def script_tick(seconds):
  if is_active:
    update_position()

# Callback for the hotkey
def on_dvd_hotkey(pressed):
  global is_active
  is_active = pressed


# Identifier of the hotkey set by OBS
hotkey_id = obs.OBS_INVALID_HOTKEY_ID

# Called at script load
def script_load(settings):
  global hotkey_id
  hotkey_id = obs.obs_hotkey_register_frontend(
      script_path(), "Source DVD", on_dvd_hotkey)
  hotkey_save_array = obs.obs_data_get_array(settings, "dvd_hotkey")
  obs.obs_hotkey_load(hotkey_id, hotkey_save_array)
  obs.obs_data_array_release(hotkey_save_array)

# Called before data settings are saved


def script_save(settings):
  
  obs.obs_save_sources()

  # Hotkey save
  hotkey_save_array = obs.obs_hotkey_save(hotkey_id)
  obs.obs_data_set_array(settings, "dvd_hotkey", hotkey_save_array)
  obs.obs_data_array_release(hotkey_save_array)