From b037c041f4ed313de69caa0b73d555ae6ebfa882 Mon Sep 17 00:00:00 2001
From: marcobaobao <marco.piovanello@gmail.com>
Date: Sat, 24 Sep 2022 12:53:36 +0200
Subject: [PATCH] first proposal download path selection

---
 frontend/src/Home.tsx               | 74 +++++++++++++++++++++--------
 server/src/core/Process.ts          |  6 +++
 server/src/core/downloader.ts       |  8 +++-
 server/src/interfaces/IPayload.d.ts |  1 +
 server/src/main.ts                  | 13 ++++-
 server/src/utils/directoryUtils.ts  | 62 ++++++++++++++++++++++++
 server/src/utils/logger.ts          |  4 +-
 7 files changed, 144 insertions(+), 24 deletions(-)
 create mode 100644 server/src/utils/directoryUtils.ts

diff --git a/frontend/src/Home.tsx b/frontend/src/Home.tsx
index ee6e3ab..a64d6b1 100644
--- a/frontend/src/Home.tsx
+++ b/frontend/src/Home.tsx
@@ -5,10 +5,13 @@ import {
     ButtonGroup,
     CircularProgress,
     Container,
+    FormControl,
     Grid,
     IconButton,
     InputAdornment,
+    MenuItem,
     Paper,
+    Select,
     Snackbar,
     styled,
     TextField,
@@ -44,6 +47,9 @@ export default function Home({ socket }: Props) {
     const [pickedAudioFormat, setPickedAudioFormat] = useState('');
     const [pickedBestFormat, setPickedBestFormat] = useState('');
 
+    const [downloadPath, setDownloadPath] = useState<number>(0);
+    const [availableDownloadPaths, setAvailableDownloadPaths] = useState<string[]>([]);
+
     const [url, setUrl] = useState('');
     const [workingUrl, setWorkingUrl] = useState('');
     const [showBackdrop, setShowBackdrop] = useState(false);
@@ -106,6 +112,14 @@ export default function Home({ socket }: Props) {
         })
     }, [])
 
+    useEffect(() => {
+        fetch(`${window.location.protocol}//${settings.serverAddr}:${settings.serverPort}/tree`)
+            .then(res => res.json())
+            .then(data => {
+                setAvailableDownloadPaths(data.flat)
+            })
+    }, [])
+
     /* -------------------- component functions -------------------- */
 
     /**
@@ -119,6 +133,7 @@ export default function Home({ socket }: Props) {
 
         socket.emit('send-url', {
             url: immediate || url || workingUrl,
+            path: availableDownloadPaths[downloadPath],
             params: settings.cliArgs.toString() + toFormatArgs(codes),
         })
         setUrl('')
@@ -211,25 +226,44 @@ export default function Home({ socket }: Props) {
                             flexDirection: 'column',
                         }}
                     >
-                        <TextField
-                            id="urlInput"
-                            label={settings.i18n.t('urlInput')}
-                            variant="outlined"
-                            onChange={handleUrlChange}
-                            disabled={!status.connected || (settings.formatSelection && downloadFormats != null)}
-                            InputProps={{
-                                endAdornment: (
-                                    <InputAdornment position="end">
-                                        <label htmlFor="icon-button-file">
-                                            <Input id="icon-button-file" type="file" accept=".txt" onChange={parseUrlListFile} />
-                                            <IconButton color="primary" aria-label="upload file" component="span">
-                                                <FileUpload />
-                                            </IconButton>
-                                        </label>
-                                    </InputAdornment>
-                                ),
-                            }}
-                        />
+                        <Grid container spacing={1}>
+                            <Grid item xs={10}>
+                                <TextField
+                                    fullWidth
+                                    id="urlInput"
+                                    label={settings.i18n.t('urlInput')}
+                                    variant="outlined"
+                                    onChange={handleUrlChange}
+                                    disabled={!status.connected || (settings.formatSelection && downloadFormats != null)}
+                                    InputProps={{
+                                        endAdornment: (
+                                            <InputAdornment position="end">
+                                                <label htmlFor="icon-button-file">
+                                                    <Input id="icon-button-file" type="file" accept=".txt" onChange={parseUrlListFile} />
+                                                    <IconButton color="primary" aria-label="upload file" component="span">
+                                                        <FileUpload />
+                                                    </IconButton>
+                                                </label>
+                                            </InputAdornment>
+                                        ),
+                                    }}
+                                />
+                            </Grid>
+                            <Grid item xs={2}>
+                                <FormControl fullWidth>
+                                    <Select
+                                        defaultValue={0}
+                                        value={availableDownloadPaths[downloadPath]}
+                                        onChange={(e) => setDownloadPath(e.target.value)}
+                                    >
+
+                                        {availableDownloadPaths.map((val: string, idx: number) => (
+                                            <MenuItem key={idx} value={idx}>{val}</MenuItem>
+                                        ))}
+                                    </Select>
+                                </FormControl>
+                            </Grid>
+                        </Grid>
                         <Grid container spacing={1} pt={2}>
                             <Grid item>
                                 <Button
@@ -365,7 +399,7 @@ export default function Home({ socket }: Props) {
             <Grid container spacing={{ xs: 2, md: 2 }} columns={{ xs: 4, sm: 8, md: 12 }} pt={2}>
                 { /*Super big brain flatMap moment*/
                     Array
-                        .from(messageMap)
+                        .from<any>(messageMap)
                         .filter(flattened => [...flattened][0])
                         .filter(flattened => [...flattened][1].toString() !== serverStates.PROG_DONE)
                         .flatMap(message => (
diff --git a/server/src/core/Process.ts b/server/src/core/Process.ts
index a0ee2ba..18c1fe5 100644
--- a/server/src/core/Process.ts
+++ b/server/src/core/Process.ts
@@ -41,6 +41,12 @@ class Process {
     public async start(callback?: Function): Promise<this> {
         const sanitizedParams = this.params.filter((param: string) => availableParams.includes(param));
 
+        if (this.settings?.download_path) {
+            if (this.settings.download_path.charAt(this.settings.download_path.length - 1) !== '/') {
+                this.settings.download_path = `${this.settings.download_path}/`
+            }
+        }
+
         const ytldp = spawn(this.exePath,
             ['-o', `${this.settings?.download_path || 'downloads/'}%(title)s.%(ext)s`]
                 .concat(sanitizedParams)
diff --git a/server/src/core/downloader.ts b/server/src/core/downloader.ts
index d39d621..cfbc5ab 100644
--- a/server/src/core/downloader.ts
+++ b/server/src/core/downloader.ts
@@ -61,6 +61,9 @@ export async function download(socket: Socket, payload: IPayload) {
         payload.params.split(' ') :
         payload.params;
 
+    const scopedSettings = { ...settings }
+    scopedSettings.download_path = payload.path;
+
     let p = new Process(url, params, settings);
 
     p.start().then(downloader => {
@@ -111,7 +114,10 @@ function streamProcess(process: Process, socket: Socket) {
             map(stdout => formatter(String(stdout), process.getPid()))
         )
         .subscribe({
-            next: (stdout) => socket.emit('progress', stdout),
+            next: (stdout) => {
+                socket.emit('progress', stdout)
+                log.info(`proc-${stdout.pid}`, `${stdout.progress}\t${stdout.dlSpeed}`)
+            },
             complete: () => {
                 process.kill().then(() => {
                     emitAbort();
diff --git a/server/src/interfaces/IPayload.d.ts b/server/src/interfaces/IPayload.d.ts
index f3cf324..e18ba11 100644
--- a/server/src/interfaces/IPayload.d.ts
+++ b/server/src/interfaces/IPayload.d.ts
@@ -5,6 +5,7 @@
 export interface IPayload {
     url: string
     params: Array<string> | string,
+    path: string,
     title?: string,
     thumbnail?: string,
     size?: string,
diff --git a/server/src/main.ts b/server/src/main.ts
index f9cfbf7..794fa1f 100644
--- a/server/src/main.ts
+++ b/server/src/main.ts
@@ -2,7 +2,13 @@ import { splash } from './utils/logger';
 import { join } from 'path';
 import { Server } from 'socket.io';
 import { ytdlpUpdater } from './utils/updater';
-import { download, abortDownload, retrieveDownload, abortAllDownloads, getFormatsAndMetadata } from './core/downloader';
+import {
+    download,
+    abortDownload,
+    retrieveDownload,
+    abortAllDownloads,
+    getFormatsAndMetadata
+} from './core/downloader';
 import { getFreeDiskSpace } from './utils/procUtils';
 import { listDownloaded } from './core/downloadArchive';
 import { createServer } from 'http';
@@ -13,6 +19,7 @@ import * as serve from 'koa-static';
 import * as cors from '@koa/cors';
 import Logger from './utils/BetterLogger';
 import { ISettings } from './interfaces/ISettings';
+import { directoryTree } from './utils/directoryUtils';
 
 const app = new Koa();
 const server = createServer(app.callback());
@@ -56,6 +63,10 @@ router.get('/archive', (ctx, next) => {
 router.get('/stream/:filepath', (ctx, next) => {
     streamer(ctx, next)
 })
+router.get('/tree', (ctx, next) => {
+    ctx.body = directoryTree()
+    next()
+})
 
 // WebSocket listeners
 io.on('connection', socket => {
diff --git a/server/src/utils/directoryUtils.ts b/server/src/utils/directoryUtils.ts
new file mode 100644
index 0000000..3982f5c
--- /dev/null
+++ b/server/src/utils/directoryUtils.ts
@@ -0,0 +1,62 @@
+import { readdirSync, statSync } from "fs";
+import { ISettings } from "../interfaces/ISettings";
+
+let settings: ISettings;
+
+class Node {
+    public path: string
+    public children: Node[]
+
+    constructor(path: string) {
+        this.path = path
+        this.children = []
+    }
+}
+
+function buildTreeDFS(rootPath: string, directoryOnly: boolean) {
+    const root = new Node(rootPath)
+    const stack: Node[] = []
+    const flattened: string[] = []
+
+    stack.push(root)
+    flattened.push(rootPath)
+
+    while (stack.length) {
+        const current = stack.pop()
+        if (current) {
+            const children = readdirSync(current.path)
+            for (const it of children) {
+                const childPath = `${current.path}/${it}`
+                const childNode = new Node(childPath)
+
+                if (directoryOnly) {
+                    if (statSync(childPath).isDirectory()) {
+                        current.children.push(childNode)
+                        stack.push(childNode)
+                        flattened.push(childNode.path)
+                    }
+                } else {
+                    current.children.push(childNode)
+                    if (statSync(childPath).isDirectory()) {
+                        stack.push(childNode)
+                        flattened.push(childNode.path)
+                    }
+                }
+            }
+        }
+    }
+
+    return {
+        tree: root,
+        flat: flattened
+    }
+}
+
+try {
+    settings = require('../../settings.json');
+} catch (e) { }
+
+export function directoryTree() {
+    const tree = buildTreeDFS(settings.download_path || 'downloads', true)
+    return tree
+}
\ No newline at end of file
diff --git a/server/src/utils/logger.ts b/server/src/utils/logger.ts
index 646da5e..cfe322e 100644
--- a/server/src/utils/logger.ts
+++ b/server/src/utils/logger.ts
@@ -17,8 +17,8 @@ export const splash = () => {
     const reset = "\u001b[0m"
     console.log(`${fg}        __         ____                  __   __  ______`)
     console.log("  __ __/ /________/ / /__    _    _____ / /  / / / /  _/")
-    console.log(" / // / __/___/ _  / / _ \  | |/|/ / -_) _ \/ /_/ // /  ")
-    console.log(" \_, /\__/    \_,_/_/ .__/  |__,__/\__/_.__/\____/___/  ")
+    console.log(" / // / __/___/ _  / / _ \\  | |/|/ / -_) _ \\/ /_/ // /  ")
+    console.log(" \\_, /\\__/    \\_,_/_/ .__/  |__,__/\\__/_.__/\\____/___/  ")
     console.log(`/___/              /_/                                \n${reset}`)
     console.log(" yt-dlp-webUI - A web-ui for yt-dlp, simply enough")
     console.log("---------------------------------------------------\n")