qkpr 1.0.3 → 1.0.4

package/dist/index.mjs

@@ -1237,7 +1237,7 @@ inquirer.registerPrompt("search-checkbox", searchCheckbox);
 * 通用的分支选择函数，支持单选和多选
 */
 async function promptBranchSelection(branches, options) {
-	const { title, message, mode, filterPinned = false } = options;
+	const { title, message, mode, filterPinned = false, defaultSelected = [] } = options;
 	console.log(cyan(`\n${title}`));
 	console.log(dim(""));
 	if (branches.length === 0) {
@@ -1262,7 +1262,8 @@ async function promptBranchSelection(branches, options) {
 			choices.push({
 				name: `📌 ${branch.name.padEnd(45)} ${dim(`(${branch.lastCommitTimeFormatted})`)}`,
 				value: branch.name,
-				short: branch.name
+				short: branch.name,
+				checked: defaultSelected.includes(branch.name)
 			});
 		});
 		choices.push(new inquirer.Separator(" "));
@@ -1273,7 +1274,8 @@ async function promptBranchSelection(branches, options) {
 			choices.push({
 				name: `   ${branch.name.padEnd(45)} ${dim(`(${branch.lastCommitTimeFormatted})`)}`,
 				value: branch.name,
-				short: branch.name
+				short: branch.name,
+				checked: defaultSelected.includes(branch.name)
 			});
 		});
 		choices.push(new inquirer.Separator(" "));
@@ -1300,7 +1302,7 @@ async function promptBranchSelection(branches, options) {
 			type: "search-checkbox",
 			name: "selectedBranches",
 			message,
-			choices
+			choices: choices.filter((c) => c.value)
 		}]);
 		return selectedBranches || [];
 	}
@@ -1368,25 +1370,24 @@ async function handlePinCommand(branchName) {
 			return;
 		}
 		const pinnedBranches = getPinnedBranches();
-		const availableBranches = branches.filter((b) => !pinnedBranches.includes(b));
-		if (availableBranches.length === 0) {
-			console.log(yellow("⚠️  All branches are already pinned"));
-			return;
-		}
-		const selectedBranches = await promptBranchSelection(availableBranches, {
-			title: "📌  Pin Branches",
-			message: "Select branches to pin (type to search, Space to select, Enter to confirm):",
+		const selectedBranches = await promptBranchSelection(branches, {
+			title: "📌  Manage Pinned Branches",
+			message: "Select branches to pin (type to search, Space to toggle, Enter to confirm):",
 			mode: "multiple",
-			filterPinned: true
+			defaultSelected: pinnedBranches
 		});
-		if (selectedBranches.length === 0) {
-			console.log(yellow("⚠️  No branches selected"));
-			return;
-		}
-		selectedBranches.forEach((branch) => {
+		const toAdd = selectedBranches.filter((b) => !pinnedBranches.includes(b));
+		const toRemove = pinnedBranches.filter((b) => !selectedBranches.includes(b));
+		toAdd.forEach((branch) => {
 			addPinnedBranch(branch);
 		});
-		console.log(green(`✅  Pinned ${selectedBranches.length} branch(es)`));
+		toRemove.forEach((branch) => {
+			removePinnedBranch(branch);
+		});
+		if (toAdd.length > 0 || toRemove.length > 0) {
+			if (toAdd.length > 0) console.log(green(`✅  Pinned ${toAdd.length} branch(es)`));
+			if (toRemove.length > 0) console.log(green(`✅  Unpinned ${toRemove.length} branch(es)`));
+		} else console.log(dim("No changes made"));
 	}
 	const updatedPinnedBranches = getPinnedBranches();
 	console.log(cyan("\n📌  Current pinned branches:"));

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "qkpr",
   "type": "module",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Create a Pull Request with interactive branch selection",
   "author": "KazooTTT <work@kazoottt.top>",
   "license": "MIT",