Skip to content

Commit

Permalink
Add stack footer
Browse files Browse the repository at this point in the history
  • Loading branch information
mtsgrd committed Nov 6, 2024
1 parent 9e22c6f commit 840f327
Show file tree
Hide file tree
Showing 8 changed files with 118 additions and 6 deletions.
23 changes: 23 additions & 0 deletions apps/desktop/src/lib/branch/BranchLaneContextMenu.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,17 @@
import ContextMenu from '$lib/components/contextmenu/ContextMenu.svelte';
import ContextMenuItem from '$lib/components/contextmenu/ContextMenuItem.svelte';
import ContextMenuSection from '$lib/components/contextmenu/ContextMenuSection.svelte';
import { getForgePrService } from '$lib/forge/interface/forgePrService';
import { updatePrDescriptionTables } from '$lib/forge/shared/stackFooter';
import { User } from '$lib/stores/user';
import { BranchController } from '$lib/vbranches/branchController';
import { VirtualBranch } from '$lib/vbranches/types';
import { getContext, getContextStore } from '@gitbutler/shared/context';
import Button from '@gitbutler/ui/Button.svelte';
import Modal from '@gitbutler/ui/Modal.svelte';
import Toggle from '@gitbutler/ui/Toggle.svelte';
import Tooltip from '@gitbutler/ui/Tooltip.svelte';
import { isDefined } from '@gitbutler/ui/utils/typeguards';
interface Props {
prUrl?: string;
Expand All @@ -26,6 +30,8 @@
const branchStore = getContextStore(VirtualBranch);
const branchController = getContext(BranchController);
const prService = getForgePrService();
const user = getContextStore(User);
let deleteBranchModal: Modal;
let allowRebasing = $state<boolean>();
Expand All @@ -37,6 +43,8 @@
allowRebasing = branch.allowRebasing;
});
const allPrIds = $derived(branch.series.map((series) => series.pullRequestId).filter(isDefined));
async function toggleAllowRebasing() {
branchController.updateBranchAllowRebasing(branch.id, !allowRebasing);
}
Expand Down Expand Up @@ -113,6 +121,21 @@
}}
/>
</ContextMenuSection>
{#if $user && $user.role?.includes('admin')}
<ContextMenuSection label="admin only">
<ContextMenuItem
label="Update PR footers"
disabled={allPrIds.length === 0}
onclick={() => {
if ($prService && branch) {
const allPrIds = branch.series.map((series) => series.pullRequestId).filter(isDefined);
updatePrDescriptionTables($prService, allPrIds);
}
contextMenuEl?.close();
}}
/>
</ContextMenuSection>
{/if}
</ContextMenu>

<Modal
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,14 @@
<script lang="ts">
import type { Snippet } from 'svelte';
const { label, children }: { label?: string; children: Snippet } = $props();
</script>

<div class="context-menu-section">
<slot />
{#if label}
<div class="label text-12">{label}</div>
{/if}
{@render children()}
</div>

<style lang="postcss">
Expand All @@ -17,4 +23,10 @@
border-top: 1px solid var(--clr-border-2);
}
}
.label {
padding: 6px 8px;
color: var(--clr-scale-ntrl-50);
user-select: none;
}
</style>
10 changes: 10 additions & 0 deletions apps/desktop/src/lib/forge/github/githubPrService.ts
Original file line number Diff line number Diff line change
Expand Up @@ -96,4 +96,14 @@ export class GitHubPrService implements ForgePrService {
prMonitor(prNumber: number): GitHubPrMonitor {
return new GitHubPrMonitor(this, prNumber);
}

async update(prNumber: number, details: { description?: string }) {
const { description } = details;
await this.octokit.pulls.update({
owner: this.repo.owner,
repo: this.repo.name,
pull_number: prNumber,
body: description
});
}
}
3 changes: 2 additions & 1 deletion apps/desktop/src/lib/forge/interface/forgePrService.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ import type { Writable } from 'svelte/store';

export const [getForgePrService, createForgePrServiceStore] = buildContextStore<
ForgePrService | undefined
>('gitBranchService');
>('forgePrService');

export interface ForgePrService {
loading: Writable<boolean>;
Expand All @@ -20,4 +20,5 @@ export interface ForgePrService {
merge(method: MergeMethod, prNumber: number): Promise<void>;
reopen(prNumber: number): Promise<void>;
prMonitor(prNumber: number): ForgePrMonitor;
update(prNumber: number, details: { description?: string }): Promise<void>;
}
48 changes: 48 additions & 0 deletions apps/desktop/src/lib/forge/shared/stackFooter.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
import { type DetailedPullRequest } from '$lib/forge/interface/types';
import type { ForgePrService } from '../interface/forgePrService';

export const GITBUTLER_FOOTER_BOUNDARY = '<!-- GitButler Footer Boundary -->';

export async function updatePrDescriptionTables(prService: ForgePrService, prIds: number[]) {
if (prService && prIds.length > 1) {
const prs = await Promise.all(prIds.map(async (prId) => await prService.get(prId)));
await Promise.all(
prIds.map(async (prId) => {
const pr = prs.find((p) => p.number === prId) as DetailedPullRequest;
const currentDescription = pr.body ? stripFooter(pr.body.trim()) : '';
await prService.update(pr.id, {
description: currentDescription + '\n' + generateFooter(prId, prs)
});
})
);
}
}

/**
* Generates a footer for use in pull request descriptions when part of a stack.
*/
export function generateFooter(id: number, all: DetailedPullRequest[]) {
const stackIndex = all.findIndex((pr) => pr.id === id);
let footer = '';
footer += GITBUTLER_FOOTER_BOUNDARY + '\n\n';
footer += `| # | PR |\n`;
footer += '| --- | --- |\n';
all.forEach((pr, i) => {
const current = i === stackIndex;
const rankNumber = all.length - i;
const rankStr = current ? bold(rankNumber) : rankNumber;
const prNumber = `#${pr.number}`;
const prStr = current ? bold(prNumber) : prNumber;
footer += `| ${rankStr} | ${prStr} |\n`;
});
return footer;
}

function stripFooter(description: string) {
console.log(description.split(GITBUTLER_FOOTER_BOUNDARY));
return description.split(GITBUTLER_FOOTER_BOUNDARY)[0];
}

function bold(text: string | number) {
return `**${text}**`;
}
15 changes: 14 additions & 1 deletion apps/desktop/src/lib/pr/PrDetailsModal.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@
import { mapErrorToToast } from '$lib/forge/github/errorMap';
import { getForge } from '$lib/forge/interface/forge';
import { getForgePrService } from '$lib/forge/interface/forgePrService';
import { type DetailedPullRequest, type PullRequest } from '$lib/forge/interface/types';
import { updatePrDescriptionTables } from '$lib/forge/shared/stackFooter';
import { showError, showToast } from '$lib/notifications/toasts';
import { isFailure } from '$lib/result';
import ScrollableContainer from '$lib/scroll/ScrollableContainer.svelte';
Expand All @@ -39,8 +41,8 @@
import Textarea from '@gitbutler/ui/Textarea.svelte';
import Textbox from '@gitbutler/ui/Textbox.svelte';
import ToggleButton from '@gitbutler/ui/ToggleButton.svelte';
import { isDefined } from '@gitbutler/ui/utils/typeguards';
import { tick } from 'svelte';
import type { DetailedPullRequest, PullRequest } from '$lib/forge/interface/types';
interface BaseProps {
type: 'display' | 'preview' | 'preview-series';
Expand Down Expand Up @@ -165,6 +167,9 @@
error('Pull request service not available');
return;
}
if (props.type !== 'preview-series') {
return;
}
isLoading = true;
try {
Expand Down Expand Up @@ -203,6 +208,9 @@
return;
}
// All ids that exists prior to creating a new one.
const priorIds = branch.series.map((series) => series.pullRequestId).filter(isDefined);
const pr = await $prService.createPr({
title: params.title,
body: params.body,
Expand All @@ -217,6 +225,11 @@
pr.number
);
}
// If we now have two or more pull requests we add a stack table to the description.
if (priorIds.length > 0) {
updatePrDescriptionTables($prService, priorIds.concat([pr.number]));
}
} catch (err: any) {
console.error(err);
const toast = mapErrorToToast(err);
Expand Down
8 changes: 5 additions & 3 deletions apps/desktop/src/lib/stack/CurrentSeries.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,10 @@
import { createForgeChecksMonitorStore } from '$lib/forge/interface/forgeChecksMonitor';
import { getForgeListingService } from '$lib/forge/interface/forgeListingService';
import { createForgePrMonitorStore } from '$lib/forge/interface/forgePrMonitor';
import { createForgePrServiceStore } from '$lib/forge/interface/forgePrService';
import {
createForgePrServiceStore,

Check failure on line 8 in apps/desktop/src/lib/stack/CurrentSeries.svelte

View workflow job for this annotation

GitHub Actions / lint-node

'createForgePrServiceStore' is defined but never used. Allowed unused vars must match /^_/u
getForgePrService
} from '$lib/forge/interface/forgePrService';
import type { PatchSeries } from '$lib/vbranches/types';
import type { Snippet } from 'svelte';
Expand All @@ -17,8 +20,7 @@
// Setup PR Store and Monitor on a per-series basis
const forge = getForge();
const prService = createForgePrServiceStore(undefined);
$effect(() => prService.set($forge?.prService()));
const prService = getForgePrService();
// Pretty cumbersome way of getting the PR number, would be great if we can
// make it more concise somehow.
Expand Down
3 changes: 3 additions & 0 deletions apps/desktop/src/routes/[projectId]/+layout.svelte
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
import { octokitFromAccessToken } from '$lib/forge/github/octokit';
import { createForgeStore } from '$lib/forge/interface/forge';
import { createForgeListingServiceStore } from '$lib/forge/interface/forgeListingService';
import { createForgePrServiceStore } from '$lib/forge/interface/forgePrService';
import History from '$lib/history/History.svelte';
import { HistoryService } from '$lib/history/history';
import { SyncedSnapshotService } from '$lib/history/syncedSnapshotService';
Expand Down Expand Up @@ -104,6 +105,7 @@
const listServiceStore = createForgeListingServiceStore(undefined);
const forgeStore = createForgeStore(undefined);
const prService = createForgePrServiceStore(undefined);
$effect.pre(() => {
const combinedBranchListingService = new CombinedBranchListingService(
Expand Down Expand Up @@ -143,6 +145,7 @@
const ghListService = forge?.listService();
listServiceStore.set(ghListService);
forgeStore.set(forge);
prService.set(forge ? forge.prService() : undefined);
});
// Once on load and every time the project id changes
Expand Down

0 comments on commit 840f327

Please sign in to comment.