// Copyright (c) 2020 tickstep.
//
// 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 panupload

import (
	"crypto/sha1"
	"encoding/hex"
	"os"
	"path"
	"strings"
	"time"

	"github.com/tickstep/aliyunpan/internal/config"
	"github.com/tickstep/aliyunpan/internal/utils"
	"github.com/tickstep/library-go/converter"
	"github.com/tickstep/library-go/logger"
)

const (
	// MaxUploadBlockSize 最大上传的文件分片大小
	MaxUploadBlockSize = 2 * converter.GB
	// MinUploadBlockSize 最小的上传的文件分片大小
	MinUploadBlockSize = 4 * converter.MB
	// MaxRapidUploadSize 秒传文件支持的最大文件大小
	MaxRapidUploadSize = 20 * converter.GB

	// UploadingFileName 上传文件上传状态的文件名
	UploadingFileName = "aliyunpan_uploading.json"
	// UploadingBackupFileName 上传文件上传状态的副本
	UploadingBackupFileName = "aliyunpan_uploading.json.bak"
)

var (
	cmdUploadVerbose = logger.New("FILE_UPLOAD", config.EnvVerbose)
)

func getBlockSize(fileSize int64) int64 {
	blockNum := fileSize / MinUploadBlockSize
	if blockNum > 999 {
		return fileSize/999 + 1
	}
	return MinUploadBlockSize
}

// IsUrlExpired 上传链接是否已过期。过期返回True
func IsUrlExpired(urlStr string) bool {
	expiredTimeSec, err := utils.GetExpiredTimeSecFromOSSUrl(urlStr)
	if err != nil {
		// 解析错误，默认过期
		return true
	}
	if (expiredTimeSec - time.Now().Unix()) <= 300 { // 小于5分钟
		// expired
		return true
	}
	// 返回未过期
	return false
}

func IsVideoFile(fileName string) bool {
	if fileName == "" {
		return false
	}
	extName := strings.ToLower(path.Ext(fileName))
	if strings.Index(extName, ".") == 0 {
		extName = strings.TrimPrefix(extName, ".")
	}
	extList := config.Config.GetVideoExtensionList()
	for _, ext := range extList {
		if ext == extName {
			return true
		}
	}
	return false
}

// CalcFilePreHash 计算文件 PreHash
func CalcFilePreHash(filePath string) string {
	localFile, _ := os.OpenFile(filePath, os.O_RDONLY, 0)
	defer localFile.Close()
	bytes := make([]byte, 1024)
	localFile.ReadAt(bytes, 0)
	sha1w := sha1.New()
	sha1w.Write(bytes)
	shaBytes := sha1w.Sum(nil)
	hashCode := hex.EncodeToString(shaBytes)
	return strings.ToUpper(hashCode)
}