| /** |
| * @license |
| * Copyright 2019 Google LLC |
| * SPDX-License-Identifier: Apache-2.0 |
| */ |
| import '../../../styles/shared-styles'; |
| import '../../shared/gr-dialog/gr-dialog'; |
| import '../../shared/gr-icon/gr-icon'; |
| import '../../../embed/diff/gr-diff/gr-diff'; |
| import {navigationToken} from '../../core/gr-navigation/gr-navigation'; |
| import { |
| NumericChangeId, |
| EDIT, |
| FixSuggestionInfo, |
| PatchSetNum, |
| BasePatchSetNum, |
| FilePathToDiffInfoMap, |
| } from '../../../types/common'; |
| import {DiffInfo, DiffPreferencesInfo} from '../../../types/diff'; |
| import {PROVIDED_FIX_ID} from '../../../utils/comment-util'; |
| import {OpenFixPreviewEvent} from '../../../types/events'; |
| import {getAppContext} from '../../../services/app-context'; |
| import {fireCloseFixPreview} from '../../../utils/event-util'; |
| import {DiffLayer, ParsedChangeInfo} from '../../../types/types'; |
| import {GrButton} from '../../shared/gr-button/gr-button'; |
| import {TokenHighlightLayer} from '../../../embed/diff/gr-diff-builder/token-highlight-layer'; |
| import {css, html, LitElement} from 'lit'; |
| import {customElement, property, query, state} from 'lit/decorators.js'; |
| import {sharedStyles} from '../../../styles/shared-styles'; |
| import {subscribe} from '../../lit/subscription-controller'; |
| import {assert} from '../../../utils/common-util'; |
| import {resolve} from '../../../models/dependency'; |
| import {createChangeUrl} from '../../../models/views/change'; |
| import {GrDialog} from '../../shared/gr-dialog/gr-dialog'; |
| import {userModelToken} from '../../../models/user/user-model'; |
| import {modalStyles} from '../../../styles/gr-modal-styles'; |
| import {GrSyntaxLayerWorker} from '../../../embed/diff/gr-syntax-layer/gr-syntax-layer-worker'; |
| import {highlightServiceToken} from '../../../services/highlight/highlight-service'; |
| import {anyLineTooLong} from '../../../embed/diff/gr-diff/gr-diff-utils'; |
| |
| interface FilePreview { |
| filepath: string; |
| preview: DiffInfo; |
| } |
| |
| @customElement('gr-apply-fix-dialog') |
| export class GrApplyFixDialog extends LitElement { |
| @query('#applyFixModal') |
| applyFixModal?: HTMLDialogElement; |
| |
| @query('#applyFixDialog') |
| applyFixDialog?: GrDialog; |
| |
| /** The currently observed dialog by `dialogOberserver`. */ |
| observedDialog?: GrDialog; |
| |
| /** The current observer observing the `observedDialog`. */ |
| dialogObserver?: ResizeObserver; |
| |
| @query('#nextFix') |
| nextFix?: GrButton; |
| |
| @property({type: Object}) |
| change?: ParsedChangeInfo; |
| |
| @property({type: Number}) |
| changeNum?: NumericChangeId; |
| |
| @state() |
| patchNum?: PatchSetNum; |
| |
| @state() |
| currentFix?: FixSuggestionInfo; |
| |
| @state() |
| currentPreviews: FilePreview[] = []; |
| |
| @state() |
| fixSuggestions?: FixSuggestionInfo[]; |
| |
| @state() |
| isApplyFixLoading = false; |
| |
| @state() |
| selectedFixIdx = 0; |
| |
| @state() |
| layers: DiffLayer[] = []; |
| |
| @state() |
| diffPrefs?: DiffPreferencesInfo; |
| |
| private readonly restApiService = getAppContext().restApiService; |
| |
| private readonly getUserModel = resolve(this, userModelToken); |
| |
| private readonly getNavigation = resolve(this, navigationToken); |
| |
| private readonly syntaxLayer = new GrSyntaxLayerWorker( |
| resolve(this, highlightServiceToken), |
| () => getAppContext().reportingService |
| ); |
| |
| constructor() { |
| super(); |
| subscribe( |
| this, |
| () => this.getUserModel().preferences$, |
| preferences => { |
| const layers: DiffLayer[] = [this.syntaxLayer]; |
| if (!preferences?.disable_token_highlighting) { |
| layers.push(new TokenHighlightLayer(this)); |
| } |
| this.layers = layers; |
| } |
| ); |
| subscribe( |
| this, |
| () => this.getUserModel().diffPreferences$, |
| diffPreferences => { |
| if (!diffPreferences) return; |
| this.diffPrefs = diffPreferences; |
| this.syntaxLayer.setEnabled(!!this.diffPrefs.syntax_highlighting); |
| } |
| ); |
| } |
| |
| static override styles = [ |
| sharedStyles, |
| modalStyles, |
| css` |
| .diffContainer { |
| padding: var(--spacing-l) 0; |
| border-bottom: 1px solid var(--border-color); |
| } |
| .file-name { |
| display: block; |
| padding: var(--spacing-s) var(--spacing-l); |
| background-color: var(--background-color-secondary); |
| border-bottom: 1px solid var(--border-color); |
| } |
| gr-button { |
| margin-left: var(--spacing-m); |
| } |
| .fix-picker { |
| display: flex; |
| align-items: center; |
| margin-right: var(--spacing-l); |
| } |
| `, |
| ]; |
| |
| override render() { |
| return html` |
| <dialog id="applyFixModal" tabindex="-1"> |
| <gr-dialog |
| id="applyFixDialog" |
| .confirmLabel=${this.isApplyFixLoading ? 'Saving...' : 'Apply Fix'} |
| .confirmTooltip=${this.computeTooltip()} |
| ?disabled=${this.computeDisableApplyFixButton()} |
| @confirm=${this.handleApplyFix} |
| @cancel=${this.onCancel} |
| > |
| ${this.renderHeader()} ${this.renderMain()} ${this.renderFooter()} |
| </gr-dialog> |
| </dialog> |
| `; |
| } |
| |
| override disconnectedCallback() { |
| super.disconnectedCallback(); |
| } |
| |
| private renderHeader() { |
| return html` |
| <div slot="header">${this.currentFix?.description ?? ''}</div> |
| `; |
| } |
| |
| private renderMain() { |
| const items = this.currentPreviews.map( |
| item => html` |
| <div class="file-name"> |
| <span>${item.filepath}</span> |
| </div> |
| <div class="diffContainer">${this.renderDiff(item)}</div> |
| ` |
| ); |
| return html`<div slot="main">${items}</div>`; |
| } |
| |
| private renderDiff(preview: FilePreview) { |
| const diff = preview.preview; |
| if (!anyLineTooLong(diff)) { |
| this.syntaxLayer.process(diff); |
| } |
| return html`<gr-diff |
| .prefs=${this.overridePartialDiffPrefs()} |
| .path=${preview.filepath} |
| .diff=${diff} |
| .layers=${this.layers} |
| ></gr-diff>`; |
| } |
| |
| private renderFooter() { |
| const id = this.selectedFixIdx; |
| const fixCount = this.fixSuggestions?.length ?? 0; |
| if (fixCount < 2) return; |
| return html` |
| <div slot="footer" class="fix-picker"> |
| <span>Suggested fix ${id + 1} of ${fixCount}</span> |
| <gr-button |
| id="prevFix" |
| @click=${this.onPrevFixClick} |
| ?disabled=${id === 0} |
| > |
| <gr-icon icon="chevron_left"></gr-icon> |
| </gr-button> |
| <gr-button |
| id="nextFix" |
| @click=${this.onNextFixClick} |
| ?disabled=${id === fixCount - 1} |
| > |
| <gr-icon icon="chevron_right"></gr-icon> |
| </gr-button> |
| </div> |
| `; |
| } |
| |
| /** |
| * Given event with fixSuggestions, fetch diffs associated with first |
| * suggested fix and open dialog. |
| */ |
| open(e: OpenFixPreviewEvent) { |
| this.patchNum = e.detail.patchNum; |
| this.fixSuggestions = e.detail.fixSuggestions; |
| assert(this.fixSuggestions.length > 0, 'no fix in the event'); |
| this.selectedFixIdx = 0; |
| this.applyFixModal?.showModal(); |
| return this.showSelectedFixSuggestion(this.fixSuggestions[0]); |
| } |
| |
| private async showSelectedFixSuggestion(fixSuggestion: FixSuggestionInfo) { |
| this.currentFix = fixSuggestion; |
| await this.fetchFixPreview(fixSuggestion); |
| } |
| |
| private async fetchFixPreview(fixSuggestion: FixSuggestionInfo) { |
| if (!this.changeNum || !this.patchNum) { |
| return Promise.reject( |
| new Error('Both patchNum and changeNum must be set') |
| ); |
| } |
| let res: FilePathToDiffInfoMap | undefined; |
| try { |
| if (fixSuggestion.fix_id === PROVIDED_FIX_ID) { |
| res = await this.restApiService.getFixPreview( |
| this.changeNum, |
| this.patchNum, |
| fixSuggestion.replacements |
| ); |
| } else { |
| res = await this.restApiService.getRobotCommentFixPreview( |
| this.changeNum, |
| this.patchNum, |
| fixSuggestion.fix_id |
| ); |
| } |
| if (res) { |
| this.currentPreviews = Object.keys(res).map(key => { |
| return {filepath: key, preview: res![key]}; |
| }); |
| } |
| } catch (e) { |
| this.close(false); |
| throw e; |
| } |
| return res; |
| } |
| |
| private overridePartialDiffPrefs() { |
| if (!this.diffPrefs) return undefined; |
| // generate a smaller gr-diff than fullscreen for dialog |
| return { |
| ...this.diffPrefs, |
| line_length: Math.min(this.diffPrefs.line_length, 100), |
| }; |
| } |
| |
| // visible for testing |
| onCancel(e: Event) { |
| if (e) e.stopPropagation(); |
| this.close(false); |
| } |
| |
| // visible for testing |
| onPrevFixClick(e: Event) { |
| if (e) e.stopPropagation(); |
| if (this.selectedFixIdx >= 1 && this.fixSuggestions) { |
| this.selectedFixIdx -= 1; |
| this.showSelectedFixSuggestion(this.fixSuggestions[this.selectedFixIdx]); |
| } |
| } |
| |
| // visible for testing |
| onNextFixClick(e: Event) { |
| if (e) e.stopPropagation(); |
| if ( |
| this.fixSuggestions && |
| this.selectedFixIdx < this.fixSuggestions.length |
| ) { |
| this.selectedFixIdx += 1; |
| this.showSelectedFixSuggestion(this.fixSuggestions[this.selectedFixIdx]); |
| } |
| } |
| |
| private close(fixApplied: boolean) { |
| this.currentFix = undefined; |
| this.currentPreviews = []; |
| this.isApplyFixLoading = false; |
| |
| fireCloseFixPreview(this, fixApplied); |
| this.applyFixModal?.close(); |
| } |
| |
| private computeTooltip() { |
| if (!this.change || !this.patchNum) return ''; |
| const latestPatchNum = |
| this.change.revisions[this.change.current_revision]._number; |
| return latestPatchNum !== this.patchNum |
| ? 'Fix can only be applied to the latest patchset' |
| : ''; |
| } |
| |
| private computeDisableApplyFixButton() { |
| if (!this.change || !this.patchNum) return true; |
| const latestPatchNum = |
| this.change.revisions[this.change.current_revision]._number; |
| return this.patchNum !== latestPatchNum || this.isApplyFixLoading; |
| } |
| |
| // visible for testing |
| async handleApplyFix(e: Event) { |
| if (e) e.stopPropagation(); |
| |
| const changeNum = this.changeNum; |
| const patchNum = this.patchNum; |
| const change = this.change; |
| if (!changeNum || !patchNum || !change || !this.currentFix) { |
| throw new Error('Not all required properties are set.'); |
| } |
| this.isApplyFixLoading = true; |
| let res; |
| if (this.fixSuggestions?.[0].fix_id === PROVIDED_FIX_ID) { |
| res = await this.restApiService.applyFixSuggestion( |
| changeNum, |
| patchNum, |
| this.fixSuggestions[0].replacements |
| ); |
| } else { |
| res = await this.restApiService.applyRobotFixSuggestion( |
| changeNum, |
| patchNum, |
| this.currentFix.fix_id |
| ); |
| } |
| if (res && res.ok) { |
| this.getNavigation().setUrl( |
| createChangeUrl({ |
| change, |
| patchNum: EDIT, |
| basePatchNum: patchNum as BasePatchSetNum, |
| }) |
| ); |
| this.close(true); |
| } |
| this.isApplyFixLoading = false; |
| } |
| } |
| |
| declare global { |
| interface HTMLElementTagNameMap { |
| 'gr-apply-fix-dialog': GrApplyFixDialog; |
| } |
| } |
