docker 镜像推送脚本

脚本

https://github.xyh.moe:8888/xian.yuheng/docker-push

#! /bin/bash

# 禁止 ^C 显示
if [ -t 0 ]; then
    stty -echoctl
    #stty echoctl
fi
################################################################################
# 帮助文档
help_str=$(cat << EOF
用法: $0 [选项]
    -h, --help               帮助文档
    -f, --file               指定镜像列表文件
    -r, --registry           自定义推送仓库地址
    -a, --arch               指定架构 "linux/amd64" "linux/arm64"
    --delete                 删除已推送本地镜像
    CTRL + C                 跳过当前镜像
    CTRL + \                 退出脚本程序
EOF
)
#################################################################################
# 函数
function check_requirements() {
    local missing=()

    for cmd in skopeo jq docker; do
        if ! command -v "$cmd" &>/dev/null; then
            missing+=("$cmd")
        fi
    done

    if [[ ${#missing[@]} -gt 0 ]]; then
        echo "以下必要命令未安装，请先安装后再运行脚本："
        for cmd in "${missing[@]}"; do
            echo "  - $cmd"
        done
        exit 1
    fi
}

function generate_horizontal_line() {
    # 创建一条和终端等宽的 分割线 [horizontal line]
    if [ -t 1 ]; then
        printf "%0.s=" $(seq $(tput cols)) && printf "\n"
    else
        printf "%0.s=" $(seq 100) && printf "\n"
    fi
}

skip_signal=false
function handle_skip()
{   # 跳过信号
    skip_signal=true
}
trap handle_skip SIGINT     # Ctrl+C -> 跳过

quit_signal=false
function handle_quit()
{   # 退出信号
    echo -e "❌ 收到退出信号，正在终止脚本..."
    quit_signal=true
}
trap handle_quit SIGQUIT    # Ctrl+\ -> 退出

function check_signals()
{   # 检查跳过和终止信号
    local image="$1"
    if [[ "$quit_signal" == true ]]; then
        exit 0
    fi

    if [[ "$skip_signal" == true ]]; then
        timed_echo "\e[33m[跳过镜像] $image\e[0m"
        skip_signal=false
        return 1   # 返回 1 表示跳过当前镜像
    fi

    return 0       # 返回 0 表示继续处理
}

function timed_echo()
{
    local timestamp
    timestamp=$(date '+%Y-%m-%d %H:%M:%S')
    echo -e "[$timestamp] $*"
}

function check_remote_image() {
    # $1 源镜像
    # $2 标镜像
    
    # return 0 一致
    # return 1 不同 或 标镜像不存在
    # return 2 源镜像不存在
    # return 3 拉取次数受限

    $dockerhub_limited && [[ "$1" == docker.io/* ]] && return 3
    
    local ajson bjson acreated bcreated

    export HTTP_PROXY="http://tianjin.xyh.moe:42200"
    export HTTPS_PROXY="http://tianjin.xyh.moe:42200"
    export NO_PROXY="localhost,127.0.0.0/8,10.0.0.0/8,172.16.0.0/12,192.168.0.0/16,.xyh.moe"
    
    ajson=$(skopeo inspect docker://$1 2>&1)
    [[ "$ajson" == *"manifest unknown"* ]] && return 2
    [[ "$ajson" == *"toomanyrequests"* ]] && dockerhub_limited=true && return 3
    bjson=$(skopeo inspect docker://$2 2>/dev/null)
    [[ -z "$bjson" ]] && return 1
    
    acreated=$(echo "$ajson" | jq -r .Created)
    bcreated=$(echo "$bjson" | jq -r .Created)

    # echo $acreated
    # echo $bcreated

    if [[ "$acreated" == "$bcreated" ]]; then
        return 0
    else
        return 1
    fi
}

function read_image_file()
{   # 处理 输入文件
    # 排除 空白字符行
    # 排除 # 开头的行
    # 排除 > 开头的行
    # 排除 ` 开头的行

    local input_file="$1"
    local image_list=()
    while IFS= read -r line; do
        [[ "$line" =~ ^\s*$ ]] && continue
        [[ "$line" =~ ^[#\>] ]] && continue
        [[ "$line" =~ ^\` ]] && continue
        image_list+=("$line")
    done < "$input_file"
    echo "${image_list[@]}"
}

function format_image_name()
{   # 格式化镜像名
    local image="$1"
    local fallback_registry="$2"

    local tag name

    # 判断是否带tag，是否带:，且冒号后是否不带 /。
    if [[ "$image" == *:* && "${image##*:}" != */* ]]; then
        tag="${image##*:}"
        name="${image%:*}"
    else
        tag="latest"
        name="$image"
    fi

    IFS='/' read -ra parts <<< "$name"

    local registry repo_parts image_name repo

    if [[ "${#parts[@]}" -ge 2 && ( "${parts[0]}" == *.* || "${parts[0]}" == *:* || "${parts[0]}" == "localhost" ) ]]; then
        registry="$fallback_registry"
        image_name="${parts[${#parts[@]}-1]}"
        repo_parts=("${parts[@]:1:${#parts[@]}-2}")
    else
        registry="$fallback_registry"
        image_name="${parts[${#parts[@]}-1]}"
        repo_parts=("${parts[@]:0:${#parts[@]}-1}")
        [[ "${#repo_parts[@]}" -eq 0 ]] && repo_parts=("library")
    fi

    repo="$(IFS=/; echo "${repo_parts[*]}")"

    if [[ -n "$repo" ]]; then
        echo "${registry}/${repo}/${image_name}:${tag}"
    else
        echo "${registry}/library/${image_name}:${tag}"
    fi
}
################################################################################
# 依赖检测
check_requirements
# 检测是否有 docker 运行权限
docker version    1>/dev/null 2>/dev/null || { echo -e "\e[31m[错误] 无 Docker 运行权限。\e[0m"; exit 1; }
# 指定架构
architecture=""
# DockerHub 拉取受限默认值
dockerhub_limited=false
################################################################################
# 参数判断
ARGS=` \
    getopt \
    -o hf:r:t:a: \
    -l help,file:,registry:,arch,force-tag:,delete \
    -n "$0" \
    -- "$@" \
`
[ $? != 0 ] && echo -e "$help_str" && exit 1;
eval set -- "${ARGS}"

while true; do
    case "$1" in
        -h | --help)             echo -e "$help_str"; exit  0;;
        -f | --file)                       file="$2"; shift 2;;
        -r | --registry)        custom_registry="$2"; shift 2;;
        -a | --arch)               architecture="$2"; shift 2;;
        --delete)                  remove_local=true; shift  ;;
        --) shift; break;;
        *) echo "未知错误"; exit 1 ;;
    esac
done

# 推送源
push_registry="${custom_registry:-docker.io}"
################################################################################
if [[ -n "$file" ]]; then
    images=($(read_image_file "$file"))
else
    images=($(docker images --format '{{.Repository}}:{{.Tag}}'))
fi

generate_horizontal_line
printf "%-24s: %s\n" 推送仓库地址 "$push_registry"
printf "%-24s: %s\n" 镜像列表文件 "${file:-[使用本地镜像]}"
generate_horizontal_line

# 输出 镜像 列表
index=1
for image in "${images[@]}"; do
    printf "%4d. %s\n" "$index" "$image"
    ((index++))
done
generate_horizontal_line
################################################################################
# 推送镜像
for image in "${images[@]}";do
    check_signals "$image" || continue
    
    formated_image_name=$(format_image_name "$image" "$push_registry")
    
    # 获取 tag
    # tag="${image##*:}"
    # force_tag=false
    
    timed_echo "\e[32m[检测镜像] $image\e[0m"
    
    check_remote_image "$image" "$formated_image_name"
    
    case $? in
        0)
            # 镜像一致
            timed_echo "\e[34m[镜像一致] $image\e[0m"
            continue
            ;;
        1)
            # 标镜像不存在 或 镜像不同
            timed_echo "\e[34m[开始拉取] $image\e[0m"
            if [[ -n "$architecture" ]]; then
                docker pull --platform "$architecture" "$image" 1>/dev/null 2>/dev/null
            else
                docker pull                            "$image" 1>/dev/null 2>/dev/null
            fi
            
            if [[ $? == 0 ]]; then
                timed_echo "\e[34m[拉取成功] $image\e[0m"
                check_signals "$image" || continue
                docker tag "$image" "$formated_image_name"
                docker push "$formated_image_name" 1>/dev/null
                if [[ $? == 0 ]]; then
                    cleaning_list+=("$image")
                    timed_echo "\e[34m[推送成功] $formated_image_name\e[0m"
                else
                    timed_echo "\e[31m[推送失败] $formated_image_name\e[0m"
                fi
                docker rmi "$formated_image_name" 1>/dev/null 2>/dev/null
            else
                check_signals "$image" || continue
                timed_echo "\e[31m[拉取失败] $image\e[0m"
            fi
            ;;
        2)
            # 源镜像不存在
            timed_echo "\e[31m[无源镜像] $image\e[0m"
            continue
            ;;
        3)
            # DockerHub 拉取受限
            timed_echo "\e[33m[拉取受限] $image\e[0m"
            continue
            ;;
        *)
            # 其它
            timed_echo "\e[31m[其它故障] $image\e[0m"
            continue
            ;;
    esac
done
################################################################################
# 清理镜像
if [[ "$remove_local" == true ]]; then
    #generate_horizontal_line
    #echo "开始删除本地镜像..."
    for image in "${cleaning_list[@]}"; do
        docker rmi "$image" 1>/dev/null 2>/dev/null
        [ $? -eq 0 ] && timed_echo "\e[32m[清理镜像] $image\e[0m"
    done
    #echo "本地镜像删除完成"
    #generate_horizontal_line
fi
################################################################################

exit 0

镜像文件格式

# List

## Nginx

> <https://nginx.org/>

```text
docker.io/library/nginx:1.25-alpine
docker.io/library/nginx:1.25.0-alpine
docker.io/library/nginx:1.25.1-alpine
docker.io/library/nginx:1.25.2-alpine
```