From 70e84c584075b8d2964330cdf10d94fac0455256 Mon Sep 17 00:00:00 2001
From: William Harrell <wharrell1@protonmail.com>
Date: Wed, 2 Nov 2022 22:20:32 -0400
Subject: [PATCH] listing of files on connect

---
 client/src/components/files.vue           | 32 ++++++---------------
 client/src/neko/events.ts                 | 16 +++++++++++
 client/src/neko/index.ts                  |  9 ++++++
 client/src/neko/messages.ts               | 17 ++++++++++-
 client/src/neko/types.ts                  |  5 ++++
 client/src/store/files.ts                 | 35 +++++++++++++++++++++++
 client/src/store/index.ts                 |  3 +-
 server/internal/types/event/events.go     |  8 ++++++
 server/internal/types/message/messages.go | 11 +++++++
 server/internal/types/websocket.go        |  5 ++++
 server/internal/utils/files.go            | 30 +++++++++++++++++++
 server/internal/websocket/websocket.go    | 15 ++++++++++
 12 files changed, 160 insertions(+), 26 deletions(-)
 create mode 100644 client/src/store/files.ts
 create mode 100644 server/internal/utils/files.go

diff --git a/client/src/components/files.vue b/client/src/components/files.vue
index 7e81fb98..d663b31a 100644
--- a/client/src/components/files.vue
+++ b/client/src/components/files.vue
@@ -8,7 +8,8 @@
       <div v-for="item in files" :key="item.name" class="files-list-item">
         <i :class="fileIcon(item)" />
         <p>{{ item.name }}</p>
-        <i class="fas fa-download download" @click="() => download(item)" />
+        <i v-if="item.type !== 'dir'" class="fas fa-download download"
+        @click="() => download(item)" />
       </div>
     </div>
     <div class="files-transfer" @dragover.prevent @drop.prevent="onFileDrop">
@@ -111,7 +112,6 @@
 <script lang="ts">
 
   import { Component, Vue } from 'vue-property-decorator'
-  import { onMounted, onUnmounted } from 'vue'
 
   import Markdown from './markdown'
   import Content from './context.vue'
@@ -125,28 +125,12 @@
   })
   export default class extends Vue {
 
-    data() {
-      return {
-        cwd: '~/Downloads',
-        files: [
-          {
-            name: 'a.txt',
-            type: 'file'
-          },
-          {
-            name: 'b',
-            type: 'dir'
-          },
-          {
-            name: 'c.mkv',
-            type: 'file'
-          },
-          {
-            name: 'd.mp3',
-            type: 'file'
-          }
-        ]
-      }
+    get cwd() {
+      return this.$accessor.files.cwd
+    }
+
+    get files() {
+      return this.$accessor.files.files
     }
     
     refresh() {
diff --git a/client/src/neko/events.ts b/client/src/neko/events.ts
index 7b637a69..b6c6cbde 100644
--- a/client/src/neko/events.ts
+++ b/client/src/neko/events.ts
@@ -38,6 +38,13 @@ export const EVENT = {
     MESSAGE: 'chat/message',
     EMOTE: 'chat/emote',
   },
+  FILETRANSFER: {
+    ENABLE: 'filetransfer/enable',
+    DISABLE: 'filetransfer/disable',
+    UNPRIVENABLE: 'filetransfer/unprivenable',
+    UNPRIVDISABLE: 'filetransfer/unprivdisable',
+    LIST: 'filetransfer/list'
+  },
   SCREEN: {
     CONFIGURATIONS: 'screen/configurations',
     RESOLUTION: 'screen/resolution',
@@ -69,6 +76,7 @@ export type WebSocketEvents =
   | MemberEvents
   | SignalEvents
   | ChatEvents
+  | FileTransferEvents
   | ScreenEvents
   | BroadcastEvents
   | AdminEvents
@@ -91,6 +99,14 @@ export type SignalEvents =
   | typeof EVENT.SIGNAL.CANDIDATE
 
 export type ChatEvents = typeof EVENT.CHAT.MESSAGE | typeof EVENT.CHAT.EMOTE
+
+export type FileTransferEvents =
+  | typeof EVENT.FILETRANSFER.ENABLE
+  | typeof EVENT.FILETRANSFER.DISABLE
+  | typeof EVENT.FILETRANSFER.UNPRIVENABLE
+  | typeof EVENT.FILETRANSFER.UNPRIVDISABLE
+  | typeof EVENT.FILETRANSFER.LIST
+
 export type ScreenEvents = typeof EVENT.SCREEN.CONFIGURATIONS | typeof EVENT.SCREEN.RESOLUTION | typeof EVENT.SCREEN.SET
 
 export type BroadcastEvents =
diff --git a/client/src/neko/index.ts b/client/src/neko/index.ts
index 35f057fb..a066b3b9 100644
--- a/client/src/neko/index.ts
+++ b/client/src/neko/index.ts
@@ -24,6 +24,7 @@ import {
   AdminLockMessage,
   SystemInitPayload,
   AdminLockResource,
+  FileTransferListPayload,
 } from './messages'
 
 interface NekoEvents extends BaseEvents {}
@@ -351,6 +352,14 @@ export class NekoClient extends BaseClient implements EventEmitter<NekoEvents> {
     this.$accessor.chat.newEmote({ type: emote })
   }
 
+  /////////////////////////////
+  // Chat Events
+  /////////////////////////////
+  protected [EVENT.FILETRANSFER.LIST]({ cwd, files }: FileTransferListPayload) {
+    this.$accessor.files.setCwd(cwd)
+    this.$accessor.files.setFileList(files)
+  }
+
   /////////////////////////////
   // Screen Events
   /////////////////////////////
diff --git a/client/src/neko/messages.ts b/client/src/neko/messages.ts
index 65c3969b..feb60738 100644
--- a/client/src/neko/messages.ts
+++ b/client/src/neko/messages.ts
@@ -8,8 +8,14 @@ import {
   ChatEvents,
   ScreenEvents,
   AdminEvents,
+  FileTransferEvents,
 } from './events'
-import { Member, ScreenConfigurations, ScreenResolution } from './types'
+import {
+  FileListItem,
+  Member,
+  ScreenConfigurations,
+  ScreenResolution
+} from './types'
 
 export type WebSocketMessages =
   | WebSocketMessage
@@ -192,6 +198,15 @@ export interface EmojiSendPayload {
   emote: string
 }
 
+// file transfer
+export interface FileTransferMessage extends WebSocketMessage, FileTransferListPayload {
+  event: FileTransferEvents
+}
+export interface FileTransferListPayload {
+  cwd: string,
+  files: FileListItem[]
+}
+
 /*
   SCREEN PAYLOADS
 */
diff --git a/client/src/neko/types.ts b/client/src/neko/types.ts
index a4df6476..938fc8cd 100644
--- a/client/src/neko/types.ts
+++ b/client/src/neko/types.ts
@@ -22,3 +22,8 @@ export interface ScreenResolution {
   height: number
   rate: number
 }
+
+export interface FileListItem {
+  name: string,
+  type: 'file' | 'dir'
+}
diff --git a/client/src/store/files.ts b/client/src/store/files.ts
new file mode 100644
index 00000000..ce57a6ee
--- /dev/null
+++ b/client/src/store/files.ts
@@ -0,0 +1,35 @@
+import { actionTree, getterTree, mutationTree } from 'typed-vuex'
+import { FileListItem } from '~/neko/types'
+import { accessor } from '~/store'
+
+export const state = () => ({
+  cwd: '',
+  files: [] as FileListItem[]
+})
+
+export const getters = getterTree(state, {
+  //
+})
+
+export const mutations = mutationTree(state, {
+  _setCwd(state, cwd: string) {
+    state.cwd = cwd
+  },
+
+  _setFileList(state, files: FileListItem[]) {
+    state.files = files
+  }
+})
+
+export const actions = actionTree(
+  { state, getters, mutations },
+  {
+    setCwd(store, cwd: string) {
+      accessor.files._setCwd(cwd)
+    },
+
+    setFileList(store, files: FileListItem[]) {
+      accessor.files._setFileList(files)
+    }
+  }
+)
diff --git a/client/src/store/index.ts b/client/src/store/index.ts
index 3abb348b..11bafe2c 100644
--- a/client/src/store/index.ts
+++ b/client/src/store/index.ts
@@ -7,6 +7,7 @@ import { get, set } from '~/utils/localstorage'
 
 import * as video from './video'
 import * as chat from './chat'
+import * as files from './files'
 import * as remote from './remote'
 import * as user from './user'
 import * as settings from './settings'
@@ -97,7 +98,7 @@ export const storePattern = {
   state,
   mutations,
   actions,
-  modules: { video, chat, user, remote, settings, client, emoji },
+  modules: { video, chat, files, user, remote, settings, client, emoji },
 }
 
 Vue.use(Vuex)
diff --git a/server/internal/types/event/events.go b/server/internal/types/event/events.go
index fca67f8a..e6bf8b5d 100644
--- a/server/internal/types/event/events.go
+++ b/server/internal/types/event/events.go
@@ -34,6 +34,14 @@ const (
 	CHAT_EMOTE   = "chat/emote"
 )
 
+const (
+	FILETRANSFER_ENABLE        = "filetransfer/enable"
+	FILETRANSFER_DISABLE       = "filetransfer/disable"
+	FILETRANSFER_UNPRIVENABLE  = "filetransfer/unprivenable"
+	FILETRANSFER_UNPRIVDISABLE = "filetransfer/unprivdisable"
+	FILETRANSFER_LIST          = "filetransfer/list"
+)
+
 const (
 	SCREEN_CONFIGURATIONS = "screen/configurations"
 	SCREEN_RESOLUTION     = "screen/resolution"
diff --git a/server/internal/types/message/messages.go b/server/internal/types/message/messages.go
index 29542be5..2364360f 100644
--- a/server/internal/types/message/messages.go
+++ b/server/internal/types/message/messages.go
@@ -106,6 +106,17 @@ type EmoteSend struct {
 	Emote string `json:"emote"`
 }
 
+type FileTransferTarget struct {
+	Event string `json:"event"`
+	ID    string `json:"id"`
+}
+
+type FileList struct {
+	Event string               `json:"event"`
+	Cwd   string               `json:"cwd"`
+	Files []types.FileListItem `json:"files"`
+}
+
 type Admin struct {
 	Event string `json:"event"`
 	ID    string `json:"id"`
diff --git a/server/internal/types/websocket.go b/server/internal/types/websocket.go
index 374d4f9e..2fa14796 100644
--- a/server/internal/types/websocket.go
+++ b/server/internal/types/websocket.go
@@ -37,3 +37,8 @@ type WebSocketHandler interface {
 	CanTransferFiles(password string) (bool, error)
 	MakeFilePath(filename string) string
 }
+
+type FileListItem struct {
+	Filename string `json:"name"`
+	Type     string `json:"type"`
+}
diff --git a/server/internal/utils/files.go b/server/internal/utils/files.go
new file mode 100644
index 00000000..7ac3005f
--- /dev/null
+++ b/server/internal/utils/files.go
@@ -0,0 +1,30 @@
+package utils
+
+import (
+	"os"
+
+	"m1k1o/neko/internal/types"
+)
+
+func ListFiles(path string) (*[]types.FileListItem, error) {
+	items, err := os.ReadDir(path)
+	if err != nil {
+		return nil, err
+	}
+
+	out := make([]types.FileListItem, len(items))
+	for i, item := range items {
+		var itemType string = ""
+		if item.IsDir() {
+			itemType = "dir"
+		} else {
+			itemType = "file"
+		}
+		out[i] = types.FileListItem{
+			Filename: item.Name(),
+			Type:     itemType,
+		}
+	}
+
+	return &out, nil
+}
diff --git a/server/internal/websocket/websocket.go b/server/internal/websocket/websocket.go
index 00e8edae..ade10beb 100644
--- a/server/internal/websocket/websocket.go
+++ b/server/internal/websocket/websocket.go
@@ -133,6 +133,21 @@ func (ws *WebSocketHandler) Start() {
 			}
 		}
 
+		// send file list if necessary
+		if session.Admin() && ws.conf.FileTransfer || ws.conf.FileTransfer && ws.conf.UnprivFileTransfer {
+			files, err := utils.ListFiles(ws.conf.FileTransferPath)
+			if err == nil {
+				if err := session.Send(
+					message.FileList{
+						Event: event.FILETRANSFER_LIST,
+						Cwd:   ws.conf.FileTransferPath,
+						Files: *files,
+					}); err != nil {
+					ws.logger.Warn().Err(err).Msg("file list event has failed")
+				}
+			}
+		}
+
 		// remove outdated stats
 		if session.Admin() {
 			ws.lastAdminLeftAt = nil