mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-10-19 07:55:07 +00:00
181 lines
4.5 KiB
Go
181 lines
4.5 KiB
Go
|
//
|
||
|
// Copyright 2021, Sander van Harmelen
|
||
|
//
|
||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
// you may not use this file except in compliance with the License.
|
||
|
// You may obtain a copy of the License at
|
||
|
//
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
//
|
||
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
// See the License for the specific language governing permissions and
|
||
|
// limitations under the License.
|
||
|
//
|
||
|
|
||
|
package gitlab
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"fmt"
|
||
|
"io"
|
||
|
"mime/multipart"
|
||
|
"net/http"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"strconv"
|
||
|
)
|
||
|
|
||
|
// GroupImportExportService handles communication with the group import export
|
||
|
// related methods of the GitLab API.
|
||
|
//
|
||
|
// GitLab API docs: https://docs.gitlab.com/ce/api/group_import_export.html
|
||
|
type GroupImportExportService struct {
|
||
|
client *Client
|
||
|
}
|
||
|
|
||
|
// ScheduleExport starts a new group export.
|
||
|
//
|
||
|
// GitLab API docs:
|
||
|
// https://docs.gitlab.com/ce/api/group_import_export.html#schedule-new-export
|
||
|
func (s *GroupImportExportService) ScheduleExport(gid interface{}, options ...RequestOptionFunc) (*Response, error) {
|
||
|
group, err := parseID(gid)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
u := fmt.Sprintf("groups/%s/export", pathEscape(group))
|
||
|
|
||
|
req, err := s.client.NewRequest(http.MethodPost, u, nil, options)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return s.client.Do(req, nil)
|
||
|
}
|
||
|
|
||
|
// ExportDownload downloads the finished export.
|
||
|
//
|
||
|
// GitLab API docs:
|
||
|
// https://docs.gitlab.com/ce/api/group_import_export.html#export-download
|
||
|
func (s *GroupImportExportService) ExportDownload(gid interface{}, options ...RequestOptionFunc) (*bytes.Reader, *Response, error) {
|
||
|
group, err := parseID(gid)
|
||
|
if err != nil {
|
||
|
return nil, nil, err
|
||
|
}
|
||
|
u := fmt.Sprintf("groups/%s/export/download", pathEscape(group))
|
||
|
|
||
|
req, err := s.client.NewRequest(http.MethodGet, u, nil, options)
|
||
|
if err != nil {
|
||
|
return nil, nil, err
|
||
|
}
|
||
|
|
||
|
exportDownload := new(bytes.Buffer)
|
||
|
resp, err := s.client.Do(req, exportDownload)
|
||
|
if err != nil {
|
||
|
return nil, resp, err
|
||
|
}
|
||
|
|
||
|
return bytes.NewReader(exportDownload.Bytes()), resp, err
|
||
|
}
|
||
|
|
||
|
// GroupImportFileOptions represents the available ImportFile() options.
|
||
|
//
|
||
|
// GitLab API docs:
|
||
|
// https://docs.gitlab.com/ce/api/group_import_export.html#import-a-file
|
||
|
type GroupImportFileOptions struct {
|
||
|
Name *string `url:"name,omitempty" json:"name,omitempty"`
|
||
|
Path *string `url:"path,omitempty" json:"path,omitempty"`
|
||
|
File *string `url:"file,omitempty" json:"file,omitempty"`
|
||
|
ParentID *int `url:"parent_id,omitempty" json:"parent_id,omitempty"`
|
||
|
}
|
||
|
|
||
|
// ImportFile imports a file.
|
||
|
//
|
||
|
// GitLab API docs:
|
||
|
// https://docs.gitlab.com/ce/api/group_import_export.html#import-a-file
|
||
|
func (s *GroupImportExportService) ImportFile(opt *GroupImportFileOptions, options ...RequestOptionFunc) (*Response, error) {
|
||
|
// First check if we got all required options.
|
||
|
if opt.Name == nil || *opt.Name == "" {
|
||
|
return nil, fmt.Errorf("Missing required option: Name")
|
||
|
}
|
||
|
if opt.Path == nil || *opt.Path == "" {
|
||
|
return nil, fmt.Errorf("Missing required option: Path")
|
||
|
}
|
||
|
if opt.File == nil || *opt.File == "" {
|
||
|
return nil, fmt.Errorf("Missing required option: File")
|
||
|
}
|
||
|
|
||
|
f, err := os.Open(*opt.File)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
defer f.Close()
|
||
|
|
||
|
b := &bytes.Buffer{}
|
||
|
w := multipart.NewWriter(b)
|
||
|
|
||
|
_, filename := filepath.Split(*opt.File)
|
||
|
fw, err := w.CreateFormFile("file", filename)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
_, err = io.Copy(fw, f)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// Populate the additional fields.
|
||
|
fw, err = w.CreateFormField("name")
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
_, err = fw.Write([]byte(*opt.Name))
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
fw, err = w.CreateFormField("path")
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
_, err = fw.Write([]byte(*opt.Path))
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
if opt.ParentID != nil {
|
||
|
fw, err = w.CreateFormField("parent_id")
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
_, err = fw.Write([]byte(strconv.Itoa(*opt.ParentID)))
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if err = w.Close(); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
req, err := s.client.NewRequest(http.MethodPost, "groups/import", nil, options)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// Set the buffer as the request body.
|
||
|
if err = req.SetBody(b); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
// Overwrite the default content type.
|
||
|
req.Header.Set("Content-Type", w.FormDataContentType())
|
||
|
|
||
|
return s.client.Do(req, nil)
|
||
|
}
|