(first): initial commit

src/App.tsx

@@ -0,0 +1,130 @@
+import { useState, useEffect } from "react";
+import "./index.css";
+import { invoke } from "@tauri-apps/api/tauri";
+import { listen } from '@tauri-apps/api/event';
+import { appWindow } from '@tauri-apps/api/window';
+import { ThemeProvider } from "@/components/theme-provider";
+import { Button } from "@/components/ui/button";
+import { InstalledPrograms, WebSocketMessage, } from "./types";
+import { extract_version, is_installed, sendStreamInfo } from "./lib/utils";
+
+function App() {
+  useEffect(() => {
+    const handleCloseRequested = (event: any) => {
+      event.preventDefault();
+      appWindow.hide();
+    };
+
+    appWindow.onCloseRequested(handleCloseRequested);
+  }, []);
+  
+  const [installedPrograms, setInstalledPrograms] = useState<InstalledPrograms>({
+    python: {
+      installed: false,
+      version: null,
+    },
+    ffmpeg: {
+      installed: false,
+      version: null,
+    },
+    pytubefix: {
+      installed: false,
+      version: null,
+    },
+    pytubepp: {
+      installed: false,
+      version: null,
+    },
+  });
+
+  useEffect(() => {
+    const unlisten = listen<WebSocketMessage>('websocket-message', (event) => {
+      if(event.payload.command === 'send-stream-info') {
+        sendStreamInfo(event.payload.url);
+      } else if(event.payload.command === 'download-stream') {
+        const startDownload = async () => {
+          try {
+            await invoke('download_stream',  { url: event.payload.url, stream: event.payload.argument });
+            await invoke('receive_frontend_response',  { response: 'Download started' });
+          } catch (error) {
+            console.error(error);
+          }
+        };
+        startDownload();
+      } else if (event.payload.command === 'autostart') {
+        const handleAppAutostart = async () => {
+          appWindow.hide();
+          await invoke('receive_frontend_response',  { response: 'Appwindow Hidden' });
+        };
+        handleAppAutostart();
+      }
+    });
+
+    return () => {
+      unlisten.then(f => f());
+    };
+  }, []);
+
+  function check_all_programs() {
+    is_installed('python', '--version').then((result) => {
+      setInstalledPrograms((prevState) => ({
+        ...prevState,
+        python: {
+          installed: result.installed,
+          version: result.output ? extract_version(result.output) : null,
+        }
+      }));
+    });
+    is_installed('ffmpeg', '-version').then((result) => {
+      setInstalledPrograms((prevState) => ({
+        ...prevState,
+        ffmpeg: {
+          installed: result.installed,
+          version: result.output ? extract_version(result.output) : null,
+        }
+      }));
+    });
+    is_installed('pytubefix', '--version').then((result) => {
+      setInstalledPrograms((prevState) => ({
+        ...prevState,
+        pytubefix: {
+          installed: result.installed,
+          version: result.output ? extract_version(result.output) : null,
+        }
+      }));
+    });
+    is_installed('pytubepp', '--version').then((result) => {
+      setInstalledPrograms((prevState) => ({
+        ...prevState,
+        pytubepp: {
+          installed: result.installed,
+          version: result.output ? extract_version(result.output) : null,
+        }
+      }));
+    });
+  }
+
+  useEffect(() => {
+    check_all_programs();
+  }
+  , []);
+
+  return (
+    <ThemeProvider defaultTheme="dark" storageKey="vite-ui-theme">
+      <div className="container">
+        <div className="topbar flex justify-between items-center mt-5">
+          <h1 className="text-xl font-bold">PytubePP Helper</h1>
+          <Button size="sm" onClick={check_all_programs}>Re-Check</Button>
+        </div>
+        <div className="programstats mt-5">
+          <p><b>Python:</b> {installedPrograms.python.installed ? 'installed' : 'not installed'} {installedPrograms.python.version ? `(${installedPrograms.python.version})` : ''}</p>
+          <p><b>FFmpeg:</b> {installedPrograms.ffmpeg.installed ? 'installed' : 'not installed'} {installedPrograms.ffmpeg.version ? `(${installedPrograms.ffmpeg.version})` : ''}</p>
+          <p><b>pytubefix:</b> {installedPrograms.pytubefix.installed ? 'installed' : 'not installed'} {installedPrograms.pytubefix.version ? `(${installedPrograms.pytubefix.version})` : ''}</p>
+          <p><b>pytubepp:</b> {installedPrograms.pytubepp.installed ? 'installed' : 'not installed'} {installedPrograms.pytubepp.version ? `(${installedPrograms.pytubepp.version})` : ''}</p>
+        </div>
+      </div>
+    </ThemeProvider>
+  );
+}
+
+export default App;

src/assets/react.svg

@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink" aria-hidden="true" role="img" class="iconify iconify--logos" width="35.93" height="32" preserveAspectRatio="xMidYMid meet" viewBox="0 0 256 228"><path fill="#00D8FF" d="M210.483 73.824a171.49 171.49 0 0 0-8.24-2.597c.465-1.9.893-3.777 1.273-5.621c6.238-30.281 2.16-54.676-11.769-62.708c-13.355-7.7-35.196.329-57.254 19.526a171.23 171.23 0 0 0-6.375 5.848a155.866 155.866 0 0 0-4.241-3.917C100.759 3.829 77.587-4.822 63.673 3.233C50.33 10.957 46.379 33.89 51.995 62.588a170.974 170.974 0 0 0 1.892 8.48c-3.28.932-6.445 1.924-9.474 2.98C17.309 83.498 0 98.307 0 113.668c0 15.865 18.582 31.778 46.812 41.427a145.52 145.52 0 0 0 6.921 2.165a167.467 167.467 0 0 0-2.01 9.138c-5.354 28.2-1.173 50.591 12.134 58.266c13.744 7.926 36.812-.22 59.273-19.855a145.567 145.567 0 0 0 5.342-4.923a168.064 168.064 0 0 0 6.92 6.314c21.758 18.722 43.246 26.282 56.54 18.586c13.731-7.949 18.194-32.003 12.4-61.268a145.016 145.016 0 0 0-1.535-6.842c1.62-.48 3.21-.974 4.76-1.488c29.348-9.723 48.443-25.443 48.443-41.52c0-15.417-17.868-30.326-45.517-39.844Zm-6.365 70.984c-1.4.463-2.836.91-4.3 1.345c-3.24-10.257-7.612-21.163-12.963-32.432c5.106-11 9.31-21.767 12.459-31.957c2.619.758 5.16 1.557 7.61 2.4c23.69 8.156 38.14 20.213 38.14 29.504c0 9.896-15.606 22.743-40.946 31.14Zm-10.514 20.834c2.562 12.94 2.927 24.64 1.23 33.787c-1.524 8.219-4.59 13.698-8.382 15.893c-8.067 4.67-25.32-1.4-43.927-17.412a156.726 156.726 0 0 1-6.437-5.87c7.214-7.889 14.423-17.06 21.459-27.246c12.376-1.098 24.068-2.894 34.671-5.345a134.17 134.17 0 0 1 1.386 6.193ZM87.276 214.515c-7.882 2.783-14.16 2.863-17.955.675c-8.075-4.657-11.432-22.636-6.853-46.752a156.923 156.923 0 0 1 1.869-8.499c10.486 2.32 22.093 3.988 34.498 4.994c7.084 9.967 14.501 19.128 21.976 27.15a134.668 134.668 0 0 1-4.877 4.492c-9.933 8.682-19.886 14.842-28.658 17.94ZM50.35 144.747c-12.483-4.267-22.792-9.812-29.858-15.863c-6.35-5.437-9.555-10.836-9.555-15.216c0-9.322 13.897-21.212 37.076-29.293c2.813-.98 5.757-1.905 8.812-2.773c3.204 10.42 7.406 21.315 12.477 32.332c-5.137 11.18-9.399 22.249-12.634 32.792a134.718 134.718 0 0 1-6.318-1.979Zm12.378-84.26c-4.811-24.587-1.616-43.134 6.425-47.789c8.564-4.958 27.502 2.111 47.463 19.835a144.318 144.318 0 0 1 3.841 3.545c-7.438 7.987-14.787 17.08-21.808 26.988c-12.04 1.116-23.565 2.908-34.161 5.309a160.342 160.342 0 0 1-1.76-7.887Zm110.427 27.268a347.8 347.8 0 0 0-7.785-12.803c8.168 1.033 15.994 2.404 23.343 4.08c-2.206 7.072-4.956 14.465-8.193 22.045a381.151 381.151 0 0 0-7.365-13.322Zm-45.032-43.861c5.044 5.465 10.096 11.566 15.065 18.186a322.04 322.04 0 0 0-30.257-.006c4.974-6.559 10.069-12.652 15.192-18.18ZM82.802 87.83a323.167 323.167 0 0 0-7.227 13.238c-3.184-7.553-5.909-14.98-8.134-22.152c7.304-1.634 15.093-2.97 23.209-3.984a321.524 321.524 0 0 0-7.848 12.897Zm8.081 65.352c-8.385-.936-16.291-2.203-23.593-3.793c2.26-7.3 5.045-14.885 8.298-22.6a321.187 321.187 0 0 0 7.257 13.246c2.594 4.48 5.28 8.868 8.038 13.147Zm37.542 31.03c-5.184-5.592-10.354-11.779-15.403-18.433c4.902.192 9.899.29 14.978.29c5.218 0 10.376-.117 15.453-.343c-4.985 6.774-10.018 12.97-15.028 18.486Zm52.198-57.817c3.422 7.8 6.306 15.345 8.596 22.52c-7.422 1.694-15.436 3.058-23.88 4.071a382.417 382.417 0 0 0 7.859-13.026a347.403 347.403 0 0 0 7.425-13.565Zm-16.898 8.101a358.557 358.557 0 0 1-12.281 19.815a329.4 329.4 0 0 1-23.444.823c-7.967 0-15.716-.248-23.178-.732a310.202 310.202 0 0 1-12.513-19.846h.001a307.41 307.41 0 0 1-10.923-20.627a310.278 310.278 0 0 1 10.89-20.637l-.001.001a307.318 307.318 0 0 1 12.413-19.761c7.613-.576 15.42-.876 23.31-.876H128c7.926 0 15.743.303 23.354.883a329.357 329.357 0 0 1 12.335 19.695a358.489 358.489 0 0 1 11.036 20.54a329.472 329.472 0 0 1-11 20.722Zm22.56-122.124c8.572 4.944 11.906 24.881 6.52 51.026c-.344 1.668-.73 3.367-1.15 5.09c-10.622-2.452-22.155-4.275-34.23-5.408c-7.034-10.017-14.323-19.124-21.64-27.008a160.789 160.789 0 0 1 5.888-5.4c18.9-16.447 36.564-22.941 44.612-18.3ZM128 90.808c12.625 0 22.86 10.235 22.86 22.86s-10.235 22.86-22.86 22.86s-22.86-10.235-22.86-22.86s10.235-22.86 22.86-22.86Z"></path></svg>

src/components/theme-provider.tsx

@@ -0,0 +1,73 @@
+import { createContext, useContext, useEffect, useState } from "react"
+
+type Theme = "dark" | "light" | "system"
+
+type ThemeProviderProps = {
+  children: React.ReactNode
+  defaultTheme?: Theme
+  storageKey?: string
+}
+
+type ThemeProviderState = {
+  theme: Theme
+  setTheme: (theme: Theme) => void
+}
+
+const initialState: ThemeProviderState = {
+  theme: "system",
+  setTheme: () => null,
+}
+
+const ThemeProviderContext = createContext<ThemeProviderState>(initialState)
+
+export function ThemeProvider({
+  children,
+  defaultTheme = "system",
+  storageKey = "vite-ui-theme",
+  ...props
+}: ThemeProviderProps) {
+  const [theme, setTheme] = useState<Theme>(
+    () => (localStorage.getItem(storageKey) as Theme) || defaultTheme
+  )
+
+  useEffect(() => {
+    const root = window.document.documentElement
+
+    root.classList.remove("light", "dark")
+
+    if (theme === "system") {
+      const systemTheme = window.matchMedia("(prefers-color-scheme: dark)")
+        .matches
+        ? "dark"
+        : "light"
+
+      root.classList.add(systemTheme)
+      return
+    }
+
+    root.classList.add(theme)
+  }, [theme])
+
+  const value = {
+    theme,
+    setTheme: (theme: Theme) => {
+      localStorage.setItem(storageKey, theme)
+      setTheme(theme)
+    },
+  }
+
+  return (
+    <ThemeProviderContext.Provider {...props} value={value}>
+      {children}
+    </ThemeProviderContext.Provider>
+  )
+}
+
+export const useTheme = () => {
+  const context = useContext(ThemeProviderContext)
+
+  if (context === undefined)
+    throw new Error("useTheme must be used within a ThemeProvider")
+
+  return context
+}

src/components/ui/button.tsx

@@ -0,0 +1,57 @@
+import * as React from "react"
+import { Slot } from "@radix-ui/react-slot"
+import { cva, type VariantProps } from "class-variance-authority"
+
+import { cn } from "@/lib/utils"
+
+const buttonVariants = cva(
+  "inline-flex items-center justify-center whitespace-nowrap rounded-md text-sm font-medium transition-colors focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:pointer-events-none disabled:opacity-50",
+  {
+    variants: {
+      variant: {
+        default:
+          "bg-primary text-primary-foreground shadow hover:bg-primary/90",
+        destructive:
+          "bg-destructive text-destructive-foreground shadow-sm hover:bg-destructive/90",
+        outline:
+          "border border-input bg-background shadow-sm hover:bg-accent hover:text-accent-foreground",
+        secondary:
+          "bg-secondary text-secondary-foreground shadow-sm hover:bg-secondary/80",
+        ghost: "hover:bg-accent hover:text-accent-foreground",
+        link: "text-primary underline-offset-4 hover:underline",
+      },
+      size: {
+        default: "h-9 px-4 py-2",
+        sm: "h-8 rounded-md px-3 text-xs",
+        lg: "h-10 rounded-md px-8",
+        icon: "h-9 w-9",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+      size: "default",
+    },
+  }
+)
+
+export interface ButtonProps
+  extends React.ButtonHTMLAttributes<HTMLButtonElement>,
+    VariantProps<typeof buttonVariants> {
+  asChild?: boolean
+}
+
+const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  ({ className, variant, size, asChild = false, ...props }, ref) => {
+    const Comp = asChild ? Slot : "button"
+    return (
+      <Comp
+        className={cn(buttonVariants({ variant, size, className }))}
+        ref={ref}
+        {...props}
+      />
+    )
+  }
+)
+Button.displayName = "Button"
+
+export { Button, buttonVariants }

src/index.css

@@ -0,0 +1,69 @@
+@tailwind base;
+@tailwind components;
+@tailwind utilities;
+
+@layer base {
+  :root {
+    --background: 0 0% 100%;
+    --foreground: 240 10% 3.9%;
+    --card: 0 0% 100%;
+    --card-foreground: 240 10% 3.9%;
+    --popover: 0 0% 100%;
+    --popover-foreground: 240 10% 3.9%;
+    --primary: 240 5.9% 10%;
+    --primary-foreground: 0 0% 98%;
+    --secondary: 240 4.8% 95.9%;
+    --secondary-foreground: 240 5.9% 10%;
+    --muted: 240 4.8% 95.9%;
+    --muted-foreground: 240 3.8% 46.1%;
+    --accent: 240 4.8% 95.9%;
+    --accent-foreground: 240 5.9% 10%;
+    --destructive: 0 84.2% 60.2%;
+    --destructive-foreground: 0 0% 98%;
+    --border: 240 5.9% 90%;
+    --input: 240 5.9% 90%;
+    --ring: 240 10% 3.9%;
+    --radius: 0.5rem;
+    --chart-1: 12 76% 61%;
+    --chart-2: 173 58% 39%;
+    --chart-3: 197 37% 24%;
+    --chart-4: 43 74% 66%;
+    --chart-5: 27 87% 67%;
+  }
+
+  .dark {
+    --background: 240 10% 3.9%;
+    --foreground: 0 0% 98%;
+    --card: 240 10% 3.9%;
+    --card-foreground: 0 0% 98%;
+    --popover: 240 10% 3.9%;
+    --popover-foreground: 0 0% 98%;
+    --primary: 0 0% 98%;
+    --primary-foreground: 240 5.9% 10%;
+    --secondary: 240 3.7% 15.9%;
+    --secondary-foreground: 0 0% 98%;
+    --muted: 240 3.7% 15.9%;
+    --muted-foreground: 240 5% 64.9%;
+    --accent: 240 3.7% 15.9%;
+    --accent-foreground: 0 0% 98%;
+    --destructive: 0 62.8% 30.6%;
+    --destructive-foreground: 0 0% 98%;
+    --border: 240 3.7% 15.9%;
+    --input: 240 3.7% 15.9%;
+    --ring: 240 4.9% 83.9%;
+    --chart-1: 220 70% 50%;
+    --chart-2: 160 60% 45%;
+    --chart-3: 30 80% 55%;
+    --chart-4: 280 65% 60%;
+    --chart-5: 340 75% 55%;
+  }
+}
+
+@layer base {
+  * {
+    @apply border-border;
+  }
+  body {
+    @apply bg-background text-foreground;
+  }
+}

src/lib/utils.ts

@@ -0,0 +1,97 @@
+import { type ClassValue, clsx } from "clsx"
+import { twMerge } from "tailwind-merge"
+import { Command } from '@tauri-apps/api/shell';
+import { Stream } from "@/types";
+import { invoke } from "@tauri-apps/api";
+
+export function extract_xml(input: string): string[] {
+  const regex = /<Stream: [^>]+>/g;
+  const matches = input.match(regex);
+  return matches ? matches : [];
+}
+
+
+function parseAttributes(attributesString: string): Partial<Stream> {
+  const attributes: Partial<Stream> = {};
+  const regex = /(\w+)="([^"]*)"/g;
+  let match;
+
+  while ((match = regex.exec(attributesString)) !== null) {
+      const key = match[1];
+      const value = match[2];
+      if (['itag', 'mime_type', 'res', 'fps', 'vcodec'].includes(key)) {
+          attributes[key as keyof Partial<Stream>] = value;
+      }
+  }
+
+  return attributes;
+}
+
+export function convert_xml_to_json(xmlStrings: string[]): Stream[] {
+  return xmlStrings
+      .map(xmlString => {
+          const attributesString = xmlString.replace('<Stream: ', '').replace('>', '');
+          return parseAttributes(attributesString);
+      })
+      .filter(stream => stream.res !== undefined) as Stream[];
+}
+
+export function cn(...inputs: ClassValue[]) {
+  return twMerge(clsx(inputs))
+}
+
+export async function is_installed(program: string, arg: string): Promise<{ installed: boolean, output: string | null }> {
+  try{
+    const output = await new Command('is-' + program + '-installed', [arg]).execute();
+    if (output.code === 0) {
+      return { installed: true, output: output.stdout };
+    } else {
+      return { installed: false, output: output.stdout };
+    }
+  } catch (error) {
+    console.error(error);
+    return { installed: false, output: null };
+  }
+}
+
+export function extract_version(output: string): string | null {
+  const versionPatterns = [
+      /ffmpeg version (\d+\.\d+)/,      // Pattern for ffmpeg
+      /Python (\d+\.\d+\.\d+)/,         // Pattern for Python
+      /pytubefix (\d+\.\d+\.\d+)/,      // Pattern for pytubefix
+      /pytubepp (\d+\.\d+\.\d+)/        // Pattern for pytubepp
+  ];
+  for (const pattern of versionPatterns) {
+      const match = output.match(pattern);
+      if (match) {
+          return match[1];
+      }
+  }
+  return null;
+}
+
+export async function sendStreamInfo(url: string) {
+  const fetchData = async () => {
+    try {
+      const output = await new Command('fetch-video-info', [url, '--list']).execute();
+      if (output.code === 0) {
+        console.log(output.stdout);
+        const sendStreamData = async () => {
+          try {
+            const streamsstr = JSON.stringify(convert_xml_to_json(extract_xml(output.stdout)))
+            await invoke('receive_frontend_response',  { response: streamsstr });
+          } catch (error) {
+            console.error(error);
+          }
+        };
+        sendStreamData();
+      } else {
+        console.log(output.stdout);
+      }
+    } catch (error) {
+      console.error(error);
+    }
+  };
+
+  fetchData();
+}

src/main.tsx

@@ -0,0 +1,9 @@
+import React from "react";
+import ReactDOM from "react-dom/client";
+import App from "./App";
+
+ReactDOM.createRoot(document.getElementById("root") as HTMLElement).render(
+  <React.StrictMode>
+    <App />
+  </React.StrictMode>,
+);

src/types.ts

@@ -0,0 +1,32 @@
+export interface InstalledPrograms {
+    python: {
+        installed: boolean;
+        version: string | null;
+    };
+    ffmpeg: {
+        installed: boolean;
+        version: string | null;
+    };
+    pytubefix: {
+        installed: boolean
+        version: string | null;
+    };
+    pytubepp: {
+        installed: boolean;
+        version: string | null;
+    };
+}
+
+export interface WebSocketMessage {
+    url: string;
+    command: string;
+    argument: string;
+}
+
+export interface Stream {
+    itag: string;
+    mime_type: string;
+    res: string;
+    fps: string;
+    vcodec: string;
+}

src/vite-env.d.ts

@@ -0,0 +1 @@
+/// <reference types="vite/client" />