From caddb2091e97e5a16e5e59f72b585497211b2464 Mon Sep 17 00:00:00 2001
From: Harvey Zack <laozhaochaguan@gmail.com>
Date: Wed, 3 Jul 2024 15:56:49 +0800
Subject: [PATCH] feat: Add support for custom file names in screenshot
 functionality

---
 CHANGELOG_CN.md                                | 1 +
 packages/artplayer/src/player/screenshotMix.js | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG_CN.md b/CHANGELOG_CN.md
index 849498730..48aa2c702 100644
--- a/CHANGELOG_CN.md
+++ b/CHANGELOG_CN.md
@@ -7,6 +7,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 ## [Unreleased]
 
 -   新增 `Artplayer.STYLE` 属性，返回播放器样式文本
+-   `art.screenshot('your-name')` 截图功能支持自定义文件名字
 
 ## [5.1.6] - 2024-6-15
 
diff --git a/packages/artplayer/src/player/screenshotMix.js b/packages/artplayer/src/player/screenshotMix.js
index 484c3e8d0..5469ae370 100644
--- a/packages/artplayer/src/player/screenshotMix.js
+++ b/packages/artplayer/src/player/screenshotMix.js
@@ -41,9 +41,10 @@ export default function screenshotMix(art) {
     });
 
     def(art, 'screenshot', {
-        value: async () => {
+        value: async (name) => {
             const dataUri = await art.getDataURL();
-            download(dataUri, `artplayer_${secondToTime($video.currentTime)}.png`);
+            const fileName = name || `artplayer_${secondToTime($video.currentTime)}`;
+            download(dataUri, `${fileName}.png`);
             art.emit('screenshot', dataUri);
             return dataUri;
         },