add: save last cursor pos

.gitignore

@@ -0,0 +1,3 @@
+database.json
+.env/
+__pycache__/

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "terminal_menu"]
+	path = terminal_menu
+	url = git@git.chauvet.pro:starnakin/terminal_menu.git

README.md

@@ -2,6 +2,11 @@
 
 This project is a simple python programme to travel in http repository and open all file in VLC 
 
+## Screenshoot
+
+![preview](https://media.discordapp.net/attachments/576080802882781194/1107765749273083925/image.png)
+![server selector](https://media.discordapp.net/attachments/501841539161522176/1107767265430089828/image.png)
+
 ## Installation
 
 1. Clone the repository:
@@ -15,6 +20,7 @@ cd VLC_HTTP_LAUNCHER
 3. Install the dependencies:
 ```bash
 pip install -r requirements.txt
+pip install -r terminal_menu/requirements.txt
 ```
 
 ## Usage

database.py

@@ -0,0 +1,63 @@
+from tinydb import TinyDB, Query
+from tinydb.operations import set, delete
+
+class Database():
+    def __init__(self):
+        self.db = TinyDB("database.json", indent=4)
+        self.query = Query()
+        self.cookies_table = self.db.table("cookies")
+        self.viewing_table = self.db.table("viewing")
+        self.sites_table = self.db.table("sites")
+    
+    def get_viewing_data(self, url:str ):
+        return (self.viewing_table.get(self.query.url == url))
+
+    def get_sites(self):
+        return (self.sites_table.all())
+
+    def get_site_by_id(self, id: int):
+        return (self.sites_table.get(self.query.id == id))
+    
+    def get_sites_table_len(self):
+        return (len(self.sites_table.all()))
+
+    def get_cookies_table_len(self):
+        return (len(self.cookies_table.all()))
+
+    def add_cookies(self, arg: dict):
+        self.cookies_table.insert(arg)
+
+    def get_cookies(self):
+        return (self.cookies_table.get(self.query.id == "0"))
+
+    def get_last_site_id(self):
+        cookies = self.get_cookies()
+        last_site_id = cookies.get("last_site")
+        return (last_site_id)
+    
+    def get_last_site(self):
+        last_site_id = self.get_last_site_id()
+        last_site = self.get_site_by_id(last_site_id)
+        return (last_site)
+    
+    def get_last_path(self):
+        cookies = self.get_cookies()
+        last_path = cookies.get("last_path")
+        return (last_path)
+
+    def get_last_cursor_pos(self):
+        cookies = self.get_cookies()
+        last_path = cookies.get("last_cursor_pos")
+        return (last_path)
+
+    def set_last_site(self, value):
+        self.cookies_table.update(set("last_site", value), self.query.id == "0")
+
+    def set_last_path(self, value):
+        self.cookies_table.update(set("last_path", value), self.query.id == "0")
+
+    def set_last_cursor_pos(self, value):
+        self.cookies_table.update(set("last_cursor_pos", value), self.query.id == "0")
+
+    def add_site(self, site):
+        self.sites_table.insert(site)

main.py

@@ -1,51 +1,110 @@
-from tinydb import TinyDB, Query
-from bs4 import BeautifulSoup
-import requests
-from simple_term_menu import TerminalMenu
-import os
-import subprocess
+from urllib.parse import unquote, quote
+from scrapper import get_uri, get_files
+from urllib.parse import quote, unquote
+import player
+from database import Database
+from utils import get_url
+import menu
 
-db = TinyDB("./database.json", indent=4)
-query = Query()
-
-sites = db.table("sites")
-nb_sites = len(sites.all());
-site_choose = 0;
-if (nb_sites == 0):
-    print("add a site:")
-    site = {}
-    site.update({"url": input("url:")})
-    site.update({"user": input("user(leave blank):")})
-    site.update({"password": input("password(leave blank):")})
-    name = input("name[url] :")
-    if (name == ""):
-        name = site.get("url")
-    site.update({"name": name})
-    sites.insert(site)
-elif (nb_sites != 1):
-    lst = []
-    for site in sites.all():
-        append(sites.get("name"))
-    terminal_menu = TerminalMenu(lst)
-    site_choose = terminal_menu.show()
-site = sites.all()[site_choose]
-if (site.get("user") and site.get("password")):
-    url = f"http://{site.get('user')}:{site.get('password')}@{site.get('url')}/"
-else:
-    url = f"http://{site.get('url')}/";
-while True:
-    response = requests.get(url)
-    print("connection:", response.reason)
-    if (response.status_code != 200):
-        exit
-    soup = BeautifulSoup(response.text, 'html.parser')
-    lst = []
-    for element in soup.findAll("a"):
-        lst.append(element.text)
-    terminal_menu = TerminalMenu(lst)
-    file = soup.findAll("a")[terminal_menu.show()]["href"]
-    if (not file.endswith("/")):
-        with open(os.devnull, 'wb') as devnull:
-            subprocess.check_call(['vlc', url + file], stdout=devnull, stderr=subprocess.STDOUT)
+def files_navigator(site: dict):
+    if site.get("id") == database.get_last_site_id():
+        path = database.get_last_path()
     else:
-        url = url + file
+        path = '/'
+    database.set_last_site(site.get("id"))
+    while True:
+        pos = database.get_last_cursor_pos()
+        path = get_uri(get_url(site) + path)
+        database.set_last_path(path)
+        file = menu.files(site, path, database, pos)
+        if (file == None):
+            return 
+        elif (file == "../" and path == "/"):
+            return 
+        if (not file.endswith("/")):
+            player.play(get_url(site) + quote(path + file), database)
+        else:
+            path = path + file
+
+def add_site(database: Database):
+    site = menu.add_site(database)
+    database.add_site(site)
+
+def config_preview(site_name:str):
+    id = int(site_name.split(": ")[0])
+    site = database.get_site_by_id(id)
+    str = f"""
+    url: {site.get('url')}
+    user: {site.get('user')}
+    password: {site.get('password')}
+    """
+    return (str)
+
+def sites_deleter(database: Database):
+    lst = []
+    for i in database.get_sites():
+        lst.append(f"{str(i.get('id'))}: {i.get('name')}")
+    choose = menu.site_deleter(lst, database)
+    if (choose == None):
+        return (1)
+    site_name = lst[choose]
+    id = int(site_name.split(": ")[0])
+    sites_table.remove(query.id == id)
+    if (database.get_last_site() == id):
+        database.set_last_path("")
+        database.set_last_site("")
+
+def sites_editor(database: Database):
+    lst = []
+    for i in database.get_sites():
+        lst.append(f"{str(i.get('id'))}: {i.get('name')}")
+    choose = menu.site_editor(lst, database)
+    if (choose == None):
+        return (1)
+    site_name = lst[choose]
+    id = int(site_name.split(": ")[0])
+    site = database.get_site_by_id(id)
+    lst = [f"name: {site.get('name')}", f"url: {site.get('url')}", f"user: {site.get('user')}", f"password: {site.get('password')}"]
+    choose = menu.param_editor(lst)
+    if (choose == None):
+        return (1);
+    for i in lst:
+        print(i);
+    if (choose == 0):
+        sites_table.update(set("name", input("name: ")), query.id == id)
+    elif (choose == 1):
+        sites_table.update(set("url", input("url: ")), query.id == id)
+    elif (choose == 2):
+        sites_table.update(set("user", input("user: ")), query.id == id)
+    elif (choose == 3):
+        sites_table.update(set("password", input("password: ")), query.id == id)
+
+def sites_navigator(database: Database):
+    nb_site = database.get_sites_table_len()
+    if (nb_site == 0):
+        add_site(database)
+        nb_site = 1
+    last_site = database.get_last_site()
+    if (last_site != None):
+        files_navigator(last_site)
+    while True:
+        nb_site = database.get_sites_table_len()
+        choose = menu.sites(database.get_sites())
+        if (choose == None):
+            return (1)
+        if (choose == nb_site + 1):
+            add_site(database);
+        elif (choose == nb_site + 2):
+            sites_editor(database)
+        elif (choose == nb_site + 3):
+            sites_deleter(database)
+            sites_navigator(database)
+            return
+        else:
+            files_navigator(database.get_sites()[choose])
+
+database = Database()
+
+if (database.get_cookies_table_len() == 0):
+    database.add_cookies({"last_path": "", "last_site": "", "last_cursor_pos": 0, "id": "0"})    
+sites_navigator(database);

menu.py

@@ -0,0 +1,98 @@
+from database import Database
+from scrapper import get_files, get_uri
+from utils import get_url, get
+import platform
+from terminal_menu.Menu import Menu
+
+
+uri = ""
+preview_site = {}
+preview_database = ""
+
+def get_user_choice_by_menu(options: list, title = None, preview_command=None, preview_size: float = 0.3, show_search_hint: bool = False, skip_empty_entries = False, cursor_pos: int = 0):
+    menu = Menu(options, title, preview_body_function=preview_command, preview_ratio=preview_size, skip_empty_option=skip_empty_entries, cursor_pos=cursor_pos
+            )
+    return (menu.show());
+
+def add_site(database: Database) -> dict:
+    print("add a site:")
+    site = {}
+    site.update({"url": input("url without protocol (ip:port):")})
+    site.update({"user": input("user(leave blank):")})
+    site.update({"password": input("password(leave blank):")})
+    site.update({"id": database.get_sites_table_len()})
+    name = input(f"name[{str(database.get_sites_table_len())}]")
+    if (name == ""):
+        name = database.get_sites_table_len()
+    site.update({"name": name})
+    return (site)
+
+def files_preview(filename: str) -> str:
+    if (not filename.endswith("/")):
+        return (None)
+    files = get_files(preview_site, uri + filename)
+    return ("\n".join(files))
+
+def files(site:dict, path:str, database:Database, pos:int):
+    global uri
+    global preview_site
+
+    uri = path
+    preview_site = site
+    files = get_files(site, path)
+    choose = get_user_choice_by_menu(files,
+                                 preview_command=files_preview,
+                                 preview_size=0.3,
+                                 show_search_hint=True,
+                                 title=f"Index of {path}",
+                                 cursor_pos = pos)
+    if (choose == None):
+        database.set_last_cursor_pos(0)
+        return (None)
+    database.set_last_cursor_pos(choose)
+    return (files[choose])
+
+def config_preview(site_name:str):
+    id = int(site_name.split(": ")[0])
+    site = preview_database.get_site_by_id(id)
+    if (site == None):
+        return
+    str = f"""
+    url: {site.get('url')}
+    user: {site.get('user')}
+    password: {site.get('password')}
+    """
+    return (str)
+
+def site_deleter(sites, database: Database):
+    global preview_database
+
+    preview_database = database
+    choose = get_user_choice_by_menu(sites,
+                                 show_search_hint=True,
+                                 preview_title="delete",
+                                 preview_command=config_preview)
+    return (choose)
+
+def param_editor(lst: list):
+    choose = get_user_choice_by_menu(lst)
+    return (choose)
+
+def site_editor(lst: list, database: Database):
+    global preview_database
+
+    preview_database = database
+    choose = get_user_choice_by_menu(lst + ["", "quit"],
+                                 skip_empty_entries=True,
+                                 show_search_hint=True,
+                                 title="Edit")
+    if (choose == len(lst) + 1):
+        return (None)
+    return (choose)
+
+def sites(sites: list):
+    nb_site = len(sites)
+    choose = get_user_choice_by_menu(get(sites, "name") + ["", "add", "edit", "delete"],
+                                 skip_empty_entries=True,
+                                 show_search_hint=True)
+    return (choose)

player.py

@@ -0,0 +1,31 @@
+from database import Database
+import menu
+from time import sleep
+import os, subprocess
+
+def _play(url: str, database: Database):
+    start_pos = 0
+    viewing_data = database.get_viewing_data(url)
+    if (viewing_data != None):
+        response = menu.start_pos([f"go back to {str(viewing_data.get('last_pos'))}", "restart from 0:00"])
+        if (response == None):
+            return (1);
+        elif (response == 0):
+            start_pos = viewing_data.get("last_pos")
+    vlc_instance = vlc.Instance()
+    player = vlc_instance.media_player_new()
+    media = vlc_instance.media_new(url)
+    player.set_media(media)
+    player.play()
+    sleep(1)
+    player.video_set_mouse_input(True)
+    player.set_position(start_pos / player.get_length())
+    player.set_fullscreen(True)
+    while True:
+        print("duration:", player.get_time(),":",player.get_length())
+
+
+def play(url: str, database: Database):
+    with open(os.devnull, 'wb') as devnull:
+        subprocess.check_call(['vlc', url], stdout=devnull, stderr=subprocess.STDOUT)
+

requirements.txt

@@ -1,13 +1,7 @@
 beautifulsoup4==4.12.2
 bs4==0.0.1
-certifi==2023.5.7
-charset-normalizer==3.1.0
-idna==3.4
-install==1.3.5
-keyboard==0.13.5
+getch==1.0
 requests==2.30.0
-simple-term-menu==1.6.1
 soupsieve==2.4.1
-sty==1.0.4
 tinydb==4.7.1
 urllib3==2.0.2

scrapper.py

@@ -0,0 +1,31 @@
+from bs4 import BeautifulSoup
+from urllib.parse import quote, unquote
+from utils import get_url
+import requests
+
+def get_files(site:dict, path:str) -> []:
+    url = get_url(site) + quote(path)
+    if (url in ["/../", "../"]):
+        return ([])
+    response = requests.get(url)
+    if (response.status_code != 200):
+        print("connection:", response.reason)
+        sys.exit(1)
+    soup = BeautifulSoup(response.text, 'html.parser')
+    files = []
+    for element in soup.findAll("a"):
+        files.append(unquote(element["href"]))
+    return (files)
+
+def get_uri(url: str) -> []:
+    if (url in ["/../", "../"]):
+        return ([])
+    try:
+        response = requests.get(url)
+        if (response.status_code != 200):
+            print("connection:", response.reason)
+            sys.exit(1)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        return(soup.find("h1").text[9:])
+    except:
+        return ("")

utils.py

@@ -0,0 +1,12 @@
+
+def get_url(site: dict):
+    if (site.get("user") == "" and site.get("password") == ""):
+        return (f"http://{site.get('url')}")
+    else:
+        return (f"http://{site.get('user')}:{site.get('password')}@{site.get('url')}")
+
+def get(dictionnarys: list, key: str):
+    lst = []
+    for dictionnary in dictionnarys:
+        lst.append(dictionnary.get(key))
+    return (lst)