github.com/qiniu/go-sdk/v7@v7.27.0/storagev2/apis/api_set_bucket_refer_anti_leech.go (about)

     1  // THIS FILE IS GENERATED BY api-generator, DO NOT EDIT DIRECTLY!
     2  
     3  package apis
     4  
     5  import (
     6  	"context"
     7  	"net/http"
     8  	"net/url"
     9  	"strconv"
    10  	"strings"
    11  	"time"
    12  
    13  	auth "github.com/qiniu/go-sdk/v7/auth"
    14  	uplog "github.com/qiniu/go-sdk/v7/internal/uplog"
    15  	setbucketreferantileech "github.com/qiniu/go-sdk/v7/storagev2/apis/set_bucket_refer_anti_leech"
    16  	errors "github.com/qiniu/go-sdk/v7/storagev2/errors"
    17  	httpclient "github.com/qiniu/go-sdk/v7/storagev2/http_client"
    18  	region "github.com/qiniu/go-sdk/v7/storagev2/region"
    19  	uptoken "github.com/qiniu/go-sdk/v7/storagev2/uptoken"
    20  )
    21  
    22  type innerSetBucketReferAntiLeechRequest setbucketreferantileech.Request
    23  
    24  func (query *innerSetBucketReferAntiLeechRequest) getBucketName(ctx context.Context) (string, error) {
    25  	return query.Bucket, nil
    26  }
    27  
    28  func (query *innerSetBucketReferAntiLeechRequest) buildQuery() (url.Values, error) {
    29  	allQuery := make(url.Values)
    30  	if query.Bucket != "" {
    31  		allQuery.Set("bucket", query.Bucket)
    32  	} else {
    33  		return nil, errors.MissingRequiredFieldError{Name: "Bucket"}
    34  	}
    35  	allQuery.Set("mode", strconv.FormatInt(query.Mode, 10))
    36  	allQuery.Set("pattern", query.Pattern)
    37  	allQuery.Set("norefer", strconv.FormatInt(query.AllowEmptyReferer, 10))
    38  	allQuery.Set("source_enabled", strconv.FormatInt(query.SourceEnabled, 10))
    39  	return allQuery, nil
    40  }
    41  
    42  type (
    43  	SetBucketReferAntiLeechRequest  = setbucketreferantileech.Request
    44  	SetBucketReferAntiLeechResponse = setbucketreferantileech.Response
    45  )
    46  
    47  // 设置存储空间的防盗链模式
    48  func (storage *Storage) SetBucketReferAntiLeech(ctx context.Context, request *SetBucketReferAntiLeechRequest, options *Options) (*SetBucketReferAntiLeechResponse, error) {
    49  	if options == nil {
    50  		options = &Options{}
    51  	}
    52  	innerRequest := (*innerSetBucketReferAntiLeechRequest)(request)
    53  	serviceNames := []region.ServiceName{region.ServiceBucket}
    54  	if innerRequest.Credentials == nil && storage.client.GetCredentials() == nil {
    55  		return nil, errors.MissingRequiredFieldError{Name: "Credentials"}
    56  	}
    57  	pathSegments := make([]string, 0, 1)
    58  	pathSegments = append(pathSegments, "referAntiLeech")
    59  	path := "/" + strings.Join(pathSegments, "/")
    60  	var rawQuery string
    61  	if query, err := innerRequest.buildQuery(); err != nil {
    62  		return nil, err
    63  	} else {
    64  		rawQuery += query.Encode()
    65  	}
    66  	headers := http.Header{}
    67  	bucketName := options.OverwrittenBucketName
    68  	if bucketName == "" {
    69  		var err error
    70  		if bucketName, err = innerRequest.getBucketName(ctx); err != nil {
    71  			return nil, err
    72  		}
    73  	}
    74  	uplogInterceptor, err := uplog.NewRequestUplog("setBucketReferAntiLeech", bucketName, "", func() (string, error) {
    75  		credentials := innerRequest.Credentials
    76  		if credentials == nil {
    77  			credentials = storage.client.GetCredentials()
    78  		}
    79  		putPolicy, err := uptoken.NewPutPolicy(bucketName, time.Now().Add(time.Hour))
    80  		if err != nil {
    81  			return "", err
    82  		}
    83  		return uptoken.NewSigner(putPolicy, credentials).GetUpToken(ctx)
    84  	})
    85  	if err != nil {
    86  		return nil, err
    87  	}
    88  	req := httpclient.Request{Method: "POST", ServiceNames: serviceNames, Path: path, RawQuery: rawQuery, Endpoints: options.OverwrittenEndpoints, Region: options.OverwrittenRegion, Interceptors: []httpclient.Interceptor{uplogInterceptor}, Header: headers, AuthType: auth.TokenQiniu, Credentials: innerRequest.Credentials, OnRequestProgress: options.OnRequestProgress}
    89  	if options.OverwrittenEndpoints == nil && options.OverwrittenRegion == nil && storage.client.GetRegions() == nil {
    90  		bucketHosts := httpclient.DefaultBucketHosts()
    91  		if options.OverwrittenBucketHosts != nil {
    92  			req.Endpoints = options.OverwrittenBucketHosts
    93  		} else {
    94  			req.Endpoints = bucketHosts
    95  		}
    96  	}
    97  	resp, err := storage.client.Do(ctx, &req)
    98  	if err != nil {
    99  		return nil, err
   100  	}
   101  	return &SetBucketReferAntiLeechResponse{}, resp.Body.Close()
   102  }