135 lines
5.0 KiB
135 lines
5.0 KiB
'use strict';
|
|
import { Git, GitStatusFileStatus, GitStatusFile, IGitStatus } from './../git';
|
|
|
|
interface IFileStatusEntry {
|
|
staged: boolean;
|
|
status: GitStatusFileStatus;
|
|
fileName: string;
|
|
originalFileName: string;
|
|
}
|
|
|
|
const aheadStatusV1Regex = /(?:ahead ([0-9]+))/;
|
|
const behindStatusV1Regex = /(?:behind ([0-9]+))/;
|
|
|
|
export class GitStatusParser {
|
|
|
|
static parse(data: string, repoPath: string, porcelainVersion: number): IGitStatus {
|
|
if (!data) return undefined;
|
|
|
|
const lines = data.split('\n').filter(_ => !!_);
|
|
if (!lines.length) return undefined;
|
|
|
|
const status = {
|
|
repoPath: Git.normalizePath(repoPath),
|
|
state: {
|
|
ahead: 0,
|
|
behind: 0
|
|
},
|
|
files: []
|
|
} as IGitStatus;
|
|
|
|
if (porcelainVersion >= 2) {
|
|
this._parseV2(lines, repoPath, status);
|
|
}
|
|
else {
|
|
this._parseV1(lines, repoPath, status);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
private static _parseV1(lines: string[], repoPath: string, status: IGitStatus) {
|
|
let position = -1;
|
|
while (++position < lines.length) {
|
|
const line = lines[position];
|
|
// Header
|
|
if (line.startsWith('##')) {
|
|
const lineParts = line.split(' ');
|
|
[status.branch, status.upstream] = lineParts[1].split('...');
|
|
if (lineParts.length > 2) {
|
|
const upstreamStatus = lineParts.slice(2).join(' ');
|
|
|
|
const aheadStatus = aheadStatusV1Regex.exec(upstreamStatus);
|
|
status.state.ahead = +aheadStatus[1] || 0;
|
|
|
|
const behindStatus = behindStatusV1Regex.exec(upstreamStatus);
|
|
status.state.behind = +behindStatus[1] || 0;
|
|
}
|
|
}
|
|
else {
|
|
let entry: IFileStatusEntry;
|
|
const rawStatus = line.substring(0, 2);
|
|
let fileName = line.substring(3);
|
|
if (rawStatus[0] === 'R') {
|
|
const [file1, file2] = fileName.replace(/\"/g, '').split('->');
|
|
entry = this._parseFileEntry(rawStatus, file2.trim(), file1.trim());
|
|
}
|
|
else {
|
|
entry = this._parseFileEntry(rawStatus, fileName);
|
|
}
|
|
status.files.push(new GitStatusFile(repoPath, entry.status, entry.fileName, entry.staged, entry.originalFileName));
|
|
}
|
|
}
|
|
}
|
|
|
|
private static _parseV2(lines: string[], repoPath: string, status: IGitStatus) {
|
|
let position = -1;
|
|
while (++position < lines.length) {
|
|
const line = lines[position];
|
|
// Headers
|
|
if (line.startsWith('#')) {
|
|
const lineParts = line.split(' ');
|
|
switch (lineParts[1]) {
|
|
case 'branch.oid':
|
|
status.sha = lineParts[2];
|
|
break;
|
|
case 'branch.head':
|
|
status.branch = lineParts[2];
|
|
break;
|
|
case 'branch.upstream':
|
|
status.upstream = lineParts[2];
|
|
break;
|
|
case 'branch.ab':
|
|
status.state.ahead = +lineParts[2].substring(1);
|
|
status.state.behind = +lineParts[3].substring(1);
|
|
break;
|
|
}
|
|
}
|
|
else {
|
|
let lineParts = line.split(' ');
|
|
let entry: IFileStatusEntry;
|
|
switch (lineParts[0][0]) {
|
|
case '1': // normal
|
|
entry = this._parseFileEntry(lineParts[1], lineParts.slice(8).join(' '));
|
|
break;
|
|
case '2': // rename
|
|
const file = lineParts.slice(9).join(' ').split('\t');
|
|
entry = this._parseFileEntry(lineParts[1], file[0], file[1]);
|
|
break;
|
|
case 'u': // unmerged
|
|
entry = this._parseFileEntry(lineParts[1], lineParts.slice(10).join(' '));
|
|
break;
|
|
case '?': // untracked
|
|
entry = this._parseFileEntry(' ?', lineParts.slice(1).join(' '));
|
|
break;
|
|
}
|
|
|
|
if (entry) {
|
|
status.files.push(new GitStatusFile(repoPath, entry.status, entry.fileName, entry.staged, entry.originalFileName));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
private static _parseFileEntry(rawStatus: string, fileName: string, originalFileName?: string): IFileStatusEntry {
|
|
const indexStatus = rawStatus[0] !== '.' ? rawStatus[0].trim() : undefined;
|
|
const workTreeStatus = rawStatus[1] !== '.' ? rawStatus[1].trim() : undefined;
|
|
|
|
return {
|
|
status: (indexStatus || workTreeStatus || '?') as GitStatusFileStatus,
|
|
fileName: fileName,
|
|
originalFileName: originalFileName,
|
|
staged: !!indexStatus
|
|
};
|
|
}
|
|
}
|