mirror of
https://github.com/mattermost/mattermost.git
synced 2026-03-19 00:52:48 -04:00
121 lines
2.6 KiB
Go
121 lines
2.6 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"archive/zip"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
const (
|
|
DefaultPageSize = 200
|
|
)
|
|
|
|
func checkInteractiveTerminal() error {
|
|
fileInfo, err := os.Stdout.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if (fileInfo.Mode() & os.ModeCharDevice) == 0 {
|
|
return errors.New("this is not an interactive shell")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func zipDir(zipPath, dir string) error {
|
|
zipFile, err := os.Create(zipPath)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot create file %q: %w", zipPath, err)
|
|
}
|
|
defer zipFile.Close()
|
|
|
|
zipWriter := zip.NewWriter(zipFile)
|
|
defer zipWriter.Close()
|
|
|
|
if err := addToZip(zipWriter, dir, "."); err != nil {
|
|
return fmt.Errorf("could not add %q to zip: %w", dir, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func addToZip(zipWriter *zip.Writer, basedir, path string) error {
|
|
dirPath := filepath.Join(basedir, path)
|
|
dirEntries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot read directory %q: %w", dirPath, err)
|
|
}
|
|
|
|
for _, dirEntry := range dirEntries {
|
|
fileInfo, err := dirEntry.Info()
|
|
if err != nil {
|
|
return fmt.Errorf("cannot get file info for directory entry %q: %w", dirEntry.Name(), err)
|
|
}
|
|
|
|
filePath := filepath.Join(path, fileInfo.Name())
|
|
if fileInfo.IsDir() {
|
|
filePath += "/"
|
|
}
|
|
header, err := zip.FileInfoHeader(fileInfo)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot create zip file info header for %q path: %w", filePath, err)
|
|
}
|
|
header.Name = filePath
|
|
header.Method = zip.Deflate
|
|
|
|
w, err := zipWriter.CreateHeader(header)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot create header for path %q: %w", filePath, err)
|
|
}
|
|
|
|
if fileInfo.IsDir() {
|
|
if err = addToZip(zipWriter, basedir, filePath); err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
|
|
file, err := os.Open(filepath.Join(dirPath, fileInfo.Name()))
|
|
if err != nil {
|
|
return fmt.Errorf("cannot open file %q: %w", filePath, err)
|
|
}
|
|
|
|
_, err = io.Copy(w, file)
|
|
file.Close()
|
|
if err != nil {
|
|
return fmt.Errorf("cannot zip file contents for file %q: %w", filePath, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func getPages[T any](fn func(page, numPerPage int, etag string) ([]T, *model.Response, error), perPage int) ([]T, error) {
|
|
var (
|
|
results []T
|
|
etag string
|
|
)
|
|
|
|
for i := 0; ; i++ {
|
|
result, resp, err := fn(i, perPage, etag)
|
|
if err != nil {
|
|
return results, err
|
|
}
|
|
if len(result) == 0 {
|
|
break
|
|
}
|
|
|
|
results = append(results, result...)
|
|
etag = resp.Etag
|
|
}
|
|
return results, nil
|
|
}
|