#!/usr/bin/env python3

import httpx
from bs4 import BeautifulSoup as bs
import pyperclip as clip

import os
import subprocess
from os.path import expanduser

headers = {
    "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:101.0) Gecko/20100101 Firefox/101.0"
}

client = httpx.Client(headers=headers, follow_redirects=True)

home = expanduser("~")
download_path = f"{home}/pix/hart-cli"
os.system(f"mkdir -p {download_path}")

item = 0
page_num = 1

url = f"https://yande.re/post?page={page_num}"
page = client.get(url)

links_arr_full = []
links_arr_preview = []

def get_new_urls():
    global url
    global page
    global page_num
    global soup
    global main_content
    global links_full
    global links_arr_full
    global links_preview
    global links_arr_preview
    
    os.system("clear")

    links_arr_full.clear
    links_arr_preview.clear
    
    soup = bs(page.content, "html.parser")
    main_content = soup.find(id="post-list-posts")
    main_content = str(main_content)
    main_content = main_content.replace("smallimg", "largeimg")
    main_content = bs(main_content, features="lxml")
    main_content = main_content.find(id="post-list-posts")

    links_full = main_content.find_all_next("a", class_="directlink largeimg")
    links_arr_full = []
    links_preview = main_content.find_all_next("img",  class_="preview")
    links_arr_preview = []
    for link in links_full:
        link_url = link["href"]
        links_arr_full.append(link_url)
    for link in links_preview:
        link_url = link["src"]
        links_arr_preview.append(link_url)

def next():
    global item
    global page_num
    
    os.system("clear")
    if item != len(links_arr_preview)-1:
        item += 1
    else:
        page_num += 1
        item = 1
        get_new_urls()

def previous():
    global item
    global page_num
    global links_arr_preview

    os.system("clear")
    if item != 1:
        item -= 1
    else:
        page_num -= 1
        get_new_urls()
        item = len(links_arr_preview)-1

def download():
    global item
    global links_arr_full
    global download_path

    command = 'echo ' + links_arr_full[item] + ' | cut -d "%" -f 2 |cut -b 3-8'
    name = subprocess.check_output(command, shell=True, text=True, encoding='utf_8')
    name = name.strip('\n')
    name = str(name)+".jpg"
    command = "curl -s -o " + download_path + "/" + name + " " + links_arr_full[item]
    os.system(command)
    os.system("clear")

get_new_urls()

while True:
    command = "curl -s -o /tmp/hart-preview.jpg " + links_arr_preview[item]
    os.system(command)
    command = "convert /tmp/hart-preview.jpg -resize 500x500 /tmp/hart-preview.jpg"
    os.system(command)
    command = "kitty +icat --place 100x100@0x0 /tmp/hart-preview.jpg"
    os.system(command)
    print("next:\t\tn")
    print("previous:\tp")
    print("download:\td")
    print("copy URL:\tc")
    print("quit:\t\tq")
    choice= input()
    if choice == "n":
        next()
    elif choice == "p":
        previous()
    elif choice == "d":
        download()
    elif choice == "c":
        clip.copy(links_arr_full[item])
        os.system('clear')
    elif choice == "q":
        os.system('clear')
        exit()
    else:
        print("invaled awnser")
        exit(0)