Files
mostlymatter/server/cmd/mmctl/commands/utils.go
Agniva De Sarker efaa6264cc MM-53032: Fix module path after repo rename (#23689)
It was a good decision in hindsight to keep the public module as 0.x
because this would have been a breaking change again.

https://mattermost.atlassian.net/browse/MM-53032
```release-note
Changed the Go module path from github.com/mattermost/mattermost-server/server/v8 to github.com/mattermost/mattermost/server/v8.

For the public facing module, it's path is also changed from github.com/mattermost/mattermost-server/server/public to github.com/mattermost/mattermost/server/public
```
2023-06-11 10:54:35 +05:30

114 строки
2.4 KiB
Go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
package commands
import (
"archive/zip"
"fmt"
"io"
"io/ioutil"
"os"
"path/filepath"
"github.com/mattermost/mattermost/server/public/model"
"github.com/pkg/errors"
)
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)
fileInfos, err := ioutil.ReadDir(dirPath)
if err != nil {
return fmt.Errorf("cannot read directory %q: %w", dirPath, err)
}
for _, fileInfo := range fileInfos {
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
}