go-onlyoffice/cmd/oo/dav.go

package main

import (
	"fmt"
	"os"
	"time"

	onlyoffice "github.com/eslider/go-onlyoffice"
	"github.com/spf13/cobra"
)

func init() {
	rootCmd.AddCommand(davCmd())
}

// davCmd exposes the Documents module through the backend-agnostic FileStore
// (DAV backend). The underlying Dav calls are the oo-webdav proven path:
// MoveDavItems sends resolveType=Skip + holdResult=true, which the legacy
// fileops/move call without those params silently ignores (200 without move).
func davCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "dav",
		Short: "Documents module by folder/file id (oo-webdav proven path)",
	}
	cmd.AddCommand(davLsCmd())
	cmd.AddCommand(davMoveCmd())
	cmd.AddCommand(davCopyCmd())
	cmd.AddCommand(davMkdirCmd())
	cmd.AddCommand(davRemoveCmd())
	cmd.AddCommand(davRenameFileCmd())
	cmd.AddCommand(davRenameFolderCmd())
	cmd.AddCommand(davDownloadCmd())
	cmd.AddCommand(davFileOpsCmd())
	return cmd
}

func davLsCmd() *cobra.Command {
	cmd := &cobra.Command{
		Use:   "ls FOLDER_ID",
		Short: "List a Documents folder (@root for virtual sections)",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ctx := cmd.Context()
			if args[0] == "@root" {
				sections, err := c.ListDavSections(ctx)
				if err != nil {
					return err
				}
				rows := make([]map[string]any, 0, len(sections))
				for _, s := range sections {
					rows = append(rows, map[string]any{
						"id":           s.ID,
						"title":        s.Title,
						"filesCount":   s.FilesCount,
						"foldersCount": s.FoldersCount,
					})
				}
				printTable([]string{"id", "title", "filesCount", "foldersCount"}, rows)
				return nil
			}
			entries, err := c.FileStore(onlyoffice.ProviderDAV).List(ctx, args[0])
			if err != nil {
				return err
			}
			folders := make([]onlyoffice.Entry, 0, len(entries))
			files := make([]onlyoffice.Entry, 0, len(entries))
			for _, e := range entries {
				if e.Kind == onlyoffice.Folder {
					folders = append(folders, e)
				} else {
					files = append(files, e)
				}
			}
			frows := make([]map[string]any, 0, len(folders))
			for _, f := range folders {
				frows = append(frows, map[string]any{
					"id":           f.ID,
					"title":        f.Title,
					"filesCount":   f.FilesCount,
					"foldersCount": f.FoldersCount,
				})
			}
			rows := make([]map[string]any, 0, len(files))
			for _, f := range files {
				rows = append(rows, map[string]any{
					"id":      f.ID,
					"title":   f.Title,
					"size":    f.Size,
					"updated": entryUpdated(f),
				})
			}
			if outputFormat == "json" {
				printObject(map[string]any{"folders": frows, "files": rows})
				return nil
			}
			if len(frows) > 0 {
				fmt.Println("folders:")
				printTable([]string{"id", "title", "filesCount", "foldersCount"}, frows)
			}
			fmt.Println("files:")
			printTable([]string{"id", "title", "size", "updated"}, rows)
			return nil
		},
	}
	return cmd
}

// entryUpdated prefers the backend-native timestamp string so table/JSON output
// round-trips what the API returned.
func entryUpdated(e onlyoffice.Entry) string {
	if e.Updated != "" {
		return e.Updated
	}
	if e.Modified.IsZero() {
		return ""
	}
	return e.Modified.Format(time.RFC3339)
}

func davMoveCmd() *cobra.Command {
	var folderIDs []string
	cmd := &cobra.Command{
		Use:   "move DEST_FOLDER_ID FILE_ID [FILE_ID...]",
		Short: "Move file(s) into a Documents folder (resolveType=Skip, holdResult)",
		Args:  cobra.MinimumNArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ids := append(append([]string{}, folderIDs...), args[1:]...)
			if err := c.FileStore(onlyoffice.ProviderDAV).Move(cmd.Context(), ids, args[0]); err != nil {
				return err
			}
			printObject(map[string]any{"moved_files": args[1:], "moved_folders": folderIDs, "dest": args[0]})
			return nil
		},
	}
	cmd.Flags().StringSliceVar(&folderIDs, "folders", nil, "folder ids to move along with the files")
	return cmd
}

func davCopyCmd() *cobra.Command {
	var folderIDs []string
	cmd := &cobra.Command{
		Use:   "copy DEST_FOLDER_ID FILE_ID [FILE_ID...]",
		Short: "Copy file(s) into a Documents folder (conflictResolveType=Skip)",
		Args:  cobra.MinimumNArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ids := append(append([]string{}, folderIDs...), args[1:]...)
			if err := c.FileStore(onlyoffice.ProviderDAV).Copy(cmd.Context(), ids, args[0]); err != nil {
				return err
			}
			printObject(map[string]any{"copied_files": args[1:], "copied_folders": folderIDs, "dest": args[0]})
			return nil
		},
	}
	cmd.Flags().StringSliceVar(&folderIDs, "folders", nil, "folder ids to copy along with the files")
	return cmd
}

func davMkdirCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "mkdir PARENT_FOLDER_ID TITLE",
		Short: "Create a subfolder in Documents",
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			f, err := c.FileStore(onlyoffice.ProviderDAV).CreateFolder(cmd.Context(), args[0], args[1])
			if err != nil {
				return err
			}
			printObject(map[string]any{"id": f.ID, "title": f.Title, "parent": args[0]})
			return nil
		},
	}
}

func davRemoveCmd() *cobra.Command {
	var folderIDs []string
	cmd := &cobra.Command{
		Use:     "rm [FILE_ID...]",
		Aliases: []string{"delete"},
		Short:   "Permanently delete file(s) and/or folder(s) from Documents",
		Args:    cobra.ArbitraryArgs,
		RunE: func(cmd *cobra.Command, args []string) error {
			if len(args) == 0 && len(folderIDs) == 0 {
				return fmt.Errorf("dav rm: give at least one FILE_ID or --folders")
			}
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ids := append(append([]string{}, args...), folderIDs...)
			if err := c.FileStore(onlyoffice.ProviderDAV).Delete(cmd.Context(), ids); err != nil {
				return err
			}
			printObject(map[string]any{"deleted_files": args, "deleted_folders": folderIDs})
			return nil
		},
	}
	cmd.Flags().StringSliceVar(&folderIDs, "folders", nil, "folder ids to delete")
	return cmd
}

func davRenameFileCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "rename-file FILE_ID NEW_TITLE",
		Short: "Rename a Documents file (include extension in NEW_TITLE)",
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			if err := c.FileStore(onlyoffice.ProviderDAV).Rename(cmd.Context(), args[0], args[1]); err != nil {
				return err
			}
			printObject(map[string]any{"id": args[0], "title": args[1]})
			return nil
		},
	}
}

func davRenameFolderCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "rename-folder FOLDER_ID NEW_TITLE",
		Short: "Rename a Documents folder",
		Args:  cobra.ExactArgs(2),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			if err := c.FileStore(onlyoffice.ProviderDAV).Rename(cmd.Context(), args[0], args[1]); err != nil {
				return err
			}
			printObject(map[string]any{"id": args[0], "title": args[1]})
			return nil
		},
	}
}

func davDownloadCmd() *cobra.Command {
	var to string
	cmd := &cobra.Command{
		Use:   "download FILE_ID",
		Short: "Download Documents file bytes (default path: ./<title>)",
		Args:  cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ctx := cmd.Context()
			store := c.FileStore(onlyoffice.ProviderDAV)
			e, err := store.Stat(ctx, args[0])
			if err != nil {
				return err
			}
			path := to
			if path == "" {
				path = onlyoffice.SafeLocalFileName(e.Title)
			}
			out, err := os.Create(path)
			if err != nil {
				return err
			}
			defer out.Close()
			n, err := store.Download(ctx, args[0], out)
			if err != nil {
				_ = os.Remove(path)
				return err
			}
			printObject(map[string]any{"path": path, "bytes": n})
			return nil
		},
	}
	cmd.Flags().StringVar(&to, "to", "", "output path (default: ./<server title>)")
	return cmd
}

func davFileOpsCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "fileops",
		Short: "List active file operations (move/copy status polling)",
		Args:  cobra.NoArgs,
		RunE: func(cmd *cobra.Command, args []string) error {
			c, err := newOO(cmd)
			if err != nil {
				return err
			}
			ops, err := c.ListFileOps(cmd.Context())
			if err != nil {
				return err
			}
			rows := make([]map[string]any, 0, len(ops))
			for _, op := range ops {
				rows = append(rows, map[string]any{
					"id":        fmt.Sprint(op["id"]),
					"operation": fmt.Sprint(op["operation"]),
					"progress":  fmt.Sprint(op["progress"]),
					"finished":  fmt.Sprint(op["finished"]),
					"error":     fmt.Sprint(op["error"]),
				})
			}
			printTable([]string{"id", "operation", "progress", "finished", "error"}, rows)
			return nil
		},
	}
}