Skip to content

Commit

Permalink
[FIX] Improved build folder detection
Browse files Browse the repository at this point in the history
  • Loading branch information
@Sasikuttan2163
Sasikuttan2163 committed Feb 23, 2025
1 parent 9f2f4d0 commit 295ff2a
Showing 1 changed file with 125 additions and 55 deletions.
180 changes: 125 additions & 55 deletions frontend/source.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,23 @@
import subprocess
from urllib.parse import urlparse


def find_project_root(start_path=None, marker=".git"):
"""Traverse directories upward to find project root"""
if start_path is None:
start_path = os.path.abspath(os.path.dirname(__file__))

current_path = os.path.abspath(start_path)

while True:
if os.path.exists(os.path.join(current_path, marker)):
return current_path
parent_path = os.path.dirname(current_path)
if parent_path == current_path:
raise FileNotFoundError(f"Project root not found with marker '{marker}'")
current_path = parent_path


gi.require_version("Gtk", "3.0")
from gi.repository import Gtk, GObject, Pango, Gdk

Expand All @@ -24,49 +41,62 @@
provider = Gtk.CssProvider()
provider.load_from_data(css)
screen = Gdk.Screen.get_default()
Gtk.StyleContext.add_provider_for_screen(screen, provider, Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION)
Gtk.StyleContext.add_provider_for_screen(
screen, provider, Gtk.STYLE_PROVIDER_PRIORITY_APPLICATION
)


class WavyApp(Gtk.Window):
def __init__(self):
super().__init__(title="Wavy")
self.set_default_size(500, 400)
# Find project root and build directory
try:
self.project_root = find_project_root(marker=".wavy_root")
except FileNotFoundError:
try:
self.project_root = find_project_root(marker=".git")
except FileNotFoundError:
self.show_error("Could not locate project root directory!")
sys.exit(1)

self.build_dir = os.path.join(self.project_root, "build")
self.stack = Gtk.Stack()
self.add(self.stack)

# Page 1: File Selection
self.file_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=20)
self.file_box.set_border_width(30)

self.file_label = Gtk.Label(label="Select File to Stream.")
self.file_label.set_xalign(0)

self.select_button = Gtk.Button(label="Select...")
self.select_button.connect("clicked", self.on_select_file)

self.file_box.pack_start(self.file_label, False, False, 0)
self.file_box.pack_start(self.select_button, False, False, 0)

self.stack.add_named(self.file_box, "file_selection")

# Page 2: Directory Selection
self.dir_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=20)
self.dir_box.set_border_width(30)

self.dir_label = Gtk.Label(label="Select Output Directory")
self.dir_label.set_xalign(0)

self.dir_select_button = Gtk.Button(label="Choose Folder...")
self.dir_select_button.connect("clicked", self.on_select_directory)

self.dir_warning_label = Gtk.Label() # For warnings if directory is not empty.
self.dir_warning_label.set_xalign(0)
self.dir_warning_label.set_markup("<span foreground='red'></span>")

self.dir_box.pack_start(self.dir_label, False, False, 0)
self.dir_box.pack_start(self.dir_select_button, False, False, 0)
self.dir_box.pack_start(self.dir_warning_label, False, False, 0)

self.stack.add_named(self.dir_box, "directory_selection")

# Page 3: URL Input
Expand All @@ -75,13 +105,13 @@ def __init__(self):

self.url_title = Gtk.Label(label="Enter URL of the Server")
self.url_title.set_xalign(0)

# File display with a frame (uses default theme colours)
self.file_frame = Gtk.Frame()
self.file_frame.set_shadow_type(Gtk.ShadowType.IN)
self.file_display = Gtk.Label()
self.file_frame.add(self.file_display)

# Directory display with a frame (uses default theme colours)
self.dir_frame = Gtk.Frame()
self.dir_frame.set_shadow_type(Gtk.ShadowType.IN)
Expand All @@ -98,101 +128,123 @@ def __init__(self):

entry_box.pack_start(self.entry, True, True, 0)
entry_box.pack_start(self.forward_button, False, False, 0)

self.url_box.pack_start(self.url_title, False, False, 0)
self.url_box.pack_start(self.file_frame, False, False, 0)
self.url_box.pack_start(self.dir_frame, False, False, 0)
self.url_box.pack_start(entry_box, False, False, 0)

self.stack.add_named(self.url_box, "url_input")

# Page 4: Log Output
self.log_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=10)
self.log_box.set_border_width(30)

self.log_view = Gtk.TextView()
self.log_view.set_editable(False)
self.log_view.set_cursor_visible(False)
self.log_view.set_wrap_mode(Gtk.WrapMode.WORD)
self.log_view.modify_font(Pango.FontDescription("monospace"))
self.log_buffer = self.log_view.get_buffer()

log_scroll = Gtk.ScrolledWindow()
log_scroll.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
log_scroll.add(self.log_view)

self.log_box.pack_start(log_scroll, True, True, 0)

self.stack.add_named(self.log_box, "log_output")

self.stack.set_visible_child_name("file_selection")

def on_select_file(self, widget):
dialog = Gtk.FileChooserDialog(
title="Select an MP3 or FLAC file", parent=self, action=Gtk.FileChooserAction.OPEN,
buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN, Gtk.ResponseType.OK)
title="Select an MP3 or FLAC file",
parent=self,
action=Gtk.FileChooserAction.OPEN,
buttons=(
Gtk.STOCK_CANCEL,
Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN,
Gtk.ResponseType.OK,
),
)
file_filter = Gtk.FileFilter()
file_filter.set_name("Audio Files")
file_filter.add_pattern("*.mp3")
file_filter.add_pattern("*.flac")
dialog.add_filter(file_filter)

response = dialog.run()
if response == Gtk.ResponseType.OK:
self.selected_file = dialog.get_filename()
self.file_display.set_text(self.selected_file)
self.stack.set_visible_child_name("directory_selection")
dialog.destroy()

def on_select_directory(self, widget):
dialog = Gtk.FileChooserDialog(
title="Select Output Directory", parent=self, action=Gtk.FileChooserAction.SELECT_FOLDER,
buttons=(Gtk.STOCK_CANCEL, Gtk.ResponseType.CANCEL, Gtk.STOCK_OPEN, Gtk.ResponseType.OK)
title="Select Output Directory",
parent=self,
action=Gtk.FileChooserAction.SELECT_FOLDER,
buttons=(
Gtk.STOCK_CANCEL,
Gtk.ResponseType.CANCEL,
Gtk.STOCK_OPEN,
Gtk.ResponseType.OK,
),
)
response = dialog.run()
if response == Gtk.ResponseType.OK:
self.selected_directory = dialog.get_filename()
self.dir_display.set_text(self.selected_directory)
# Check if directory is not empty and show warning if needed.
if os.listdir(self.selected_directory):
self.dir_warning_label.set_markup("<span foreground='red'>Warning: Directory is not empty. Contents will be deleted.</span>")
self.dir_warning_label.set_markup(
"<span foreground='red'>Warning: Directory is not empty. Contents will be deleted.</span>"
)
else:
self.dir_warning_label.set_text("") # Clear warning if empty.
self.stack.set_visible_child_name("url_input")
dialog.destroy()

def on_forward(self, widget):
url = self.entry.get_text().strip()
# Prepend http:// if missing
if not (url.startswith("http://") or url.startswith("https://")):
url = "http://" + url

# Validate URL: must include a port
from urllib.parse import urlparse

parsed = urlparse(url)
if parsed.port is None:
self.show_error("URL must include a port number (e.g., http://example.com:8080).")
self.show_error(
"URL must include a port number (e.g., http://example.com:8080)."
)
return

self.entry.set_text(url) # update the entry field with the full URL

# If the directory is not empty, show a disclaimer before deletion.
if os.listdir(self.selected_directory):
disclaimer_dialog = Gtk.MessageDialog(parent=self, flags=0,
message_type=Gtk.MessageType.WARNING,
buttons=Gtk.ButtonsType.YES_NO,
text="Warning: The selected directory is not empty. All contents will be deleted.\nDo you want to proceed?")
disclaimer_dialog = Gtk.MessageDialog(
parent=self,
flags=0,
message_type=Gtk.MessageType.WARNING,
buttons=Gtk.ButtonsType.YES_NO,
text="Warning: The selected directory is not empty. All contents will be deleted.\nDo you want to proceed?",
)
response = disclaimer_dialog.run()
disclaimer_dialog.destroy()
if response != Gtk.ResponseType.YES:
return
else:
self.delete_directory_contents(self.selected_directory)

self.stack.set_visible_child_name("log_output")
self.run_subprocess(self.selected_file, self.selected_directory, url)

def delete_directory_contents(self, directory):
for filename in os.listdir(directory):
file_path = os.path.join(directory, filename)
Expand All @@ -203,10 +255,19 @@ def delete_directory_contents(self, directory):
shutil.rmtree(file_path)
except Exception as e:
self.append_log(f"Failed to delete {file_path}: {e}")

def run_subprocess(self, file, directory, url):
hls_segmenter_path = os.path.abspath("../build/hls_segmenter")
hls_dispatcher_path = os.path.abspath("../build/hls_dispatcher")
# Get absolute paths using project root
hls_segmenter_path = os.path.join(self.build_dir, "hls_segmenter")
hls_dispatcher_path = os.path.join(self.build_dir, "hls_dispatcher")

# Verify binaries exist
if not os.path.exists(hls_segmenter_path):
self.show_error(f"hls_segmenter not found at {hls_segmenter_path}")
return
if not os.path.exists(hls_dispatcher_path):
self.show_error(f"hls_dispatcher not found at {hls_dispatcher_path}")
return

command = [hls_segmenter_path, file, directory, file.split(".")[-1]]
parsed_url = urlparse(url)
Expand All @@ -222,13 +283,13 @@ def run_in_sequence():
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
text=True,
cwd=directory # Run in target directory
cwd=directory, # Run in target directory
)

# Capture segmenter output
for line in iter(seg_process.stdout.readline, ''):
for line in iter(seg_process.stdout.readline, ""):
GObject.idle_add(self.append_log, f"[Segmenter] {line.strip()}")

seg_process.wait() # Wait for segmenter to finish
except Exception as e:
GObject.idle_add(self.append_log, f"[ERROR] Segmenter failed: {str(e)}")
Expand All @@ -237,7 +298,9 @@ def run_in_sequence():
# Verify playlist was created
playlist_path = os.path.join(directory, "index.m3u8")
if not os.path.exists(playlist_path):
GObject.idle_add(self.append_log, "[ERROR] Segmenter did not create index.m3u8")
GObject.idle_add(
self.append_log, "[ERROR] Segmenter did not create index.m3u8"
)
return

# Then run dispatcher
Expand All @@ -247,32 +310,39 @@ def run_in_sequence():
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
text=True,
cwd=directory # Run in target directory
cwd=directory, # Run in target directory
)

# Capture dispatcher output
for line in iter(disp_process.stdout.readline, ''):
for line in iter(disp_process.stdout.readline, ""):
GObject.idle_add(self.append_log, f"[Dispatcher] {line.strip()}")
except Exception as e:
GObject.idle_add(self.append_log, f"[ERROR] Dispatcher failed: {str(e)}")
GObject.idle_add(
self.append_log, f"[ERROR] Dispatcher failed: {str(e)}"
)

# Run the entire sequence in a background thread
import threading

threading.Thread(target=run_in_sequence, daemon=True).start()

def append_log(self, text):
end_iter = self.log_buffer.get_end_iter()
self.log_buffer.insert(end_iter, text + "\n")
return False

def show_error(self, message):
dialog = Gtk.MessageDialog(parent=self, flags=0,
message_type=Gtk.MessageType.ERROR,
buttons=Gtk.ButtonsType.OK,
text=message)
dialog = Gtk.MessageDialog(
parent=self,
flags=0,
message_type=Gtk.MessageType.ERROR,
buttons=Gtk.ButtonsType.OK,
text=message,
)
dialog.run()
dialog.destroy()


app = WavyApp()
app.connect("destroy", Gtk.main_quit)
app.show_all()
Expand Down

0 comments on commit 295ff2a

Please sign in to comment.