import trimStart from 'lodash/trimStart'; import semaphore, { Semaphore } from 'semaphore'; import { trim } from 'lodash'; import { stripIndent } from 'common-tags'; import { CURSOR_COMPATIBILITY_SYMBOL, basename, Entry, AssetProxy, PersistOptions, Cursor, Implementation, DisplayURL, entriesByFolder, entriesByFiles, getMediaDisplayURL, getMediaAsBlob, User, Credentials, Config, ImplementationFile, unpublishedEntries, getPreviewStatus, UnpublishedEntryMediaFile, asyncLock, AsyncLock, runWithLock, getBlobSHA, blobToFileObj, contentKeyFromBranch, generateContentKey, localForage, allEntriesByFolder, filterByExtension, } from 'netlify-cms-lib-util'; import AuthenticationPage from './AuthenticationPage'; import API, { API_NAME } from './API'; const MAX_CONCURRENT_DOWNLOADS = 10; export default class GitLab implements Implementation { lock: AsyncLock; api: API | null; options: { proxied: boolean; API: API | null; initialWorkflowStatus: string; }; repo: string; branch: string; apiRoot: string; token: string | null; squashMerges: boolean; mediaFolder: string; previewContext: string; _mediaDisplayURLSem?: Semaphore; constructor(config: Config, options = {}) { this.options = { proxied: false, API: null, initialWorkflowStatus: '', ...options, }; if ( !this.options.proxied && (config.backend.repo === null || config.backend.repo === undefined) ) { throw new Error('The GitLab backend needs a "repo" in the backend configuration.'); } this.api = this.options.API || null; this.repo = config.backend.repo || ''; this.branch = config.backend.branch || 'master'; this.apiRoot = config.backend.api_root || 'https://gitlab.com/api/v4'; this.token = ''; this.squashMerges = config.backend.squash_merges || false; this.mediaFolder = config.media_folder; this.previewContext = config.backend.preview_context || ''; this.lock = asyncLock(); } isGitBackend() { return true; } async status() { const auth = (await this.api ?.user() .then(user => !!user) .catch(e => { console.warn('Failed getting GitLab user', e); return false; })) || false; return { auth }; } authComponent() { return AuthenticationPage; } restoreUser(user: User) { return this.authenticate(user); } async authenticate(state: Credentials) { this.token = state.token as string; this.api = new API({ token: this.token, branch: this.branch, repo: this.repo, apiRoot: this.apiRoot, squashMerges: this.squashMerges, initialWorkflowStatus: this.options.initialWorkflowStatus, }); const user = await this.api.user(); const isCollab = await this.api.hasWriteAccess().catch((error: Error) => { error.message = stripIndent` Repo "${this.repo}" not found. Please ensure the repo information is spelled correctly. If the repo is private, make sure you're logged into a GitLab account with access. `; throw error; }); // Unauthorized user if (!isCollab) { throw new Error('Your GitLab user account does not have access to this repo.'); } // Authorized user return { ...user, login: user.username, token: state.token as string }; } async logout() { this.token = null; return; } getToken() { return Promise.resolve(this.token); } filterFile( folder: string, file: { path: string; name: string }, extension: string, depth: number, ) { // gitlab paths include the root folder const fileFolder = trim(file.path.split(folder)[1] || '/', '/'); return filterByExtension(file, extension) && fileFolder.split('/').length <= depth; } async entriesByFolder(folder: string, extension: string, depth: number) { let cursor: Cursor; const listFiles = () => this.api!.listFiles(folder, depth > 1).then(({ files, cursor: c }) => { cursor = c.mergeMeta({ folder, extension, depth }); return files.filter(file => this.filterFile(folder, file, extension, depth)); }); const files = await entriesByFolder( listFiles, this.api!.readFile.bind(this.api!), this.api!.readFileMetadata.bind(this.api), API_NAME, ); // eslint-disable-next-line @typescript-eslint/ban-ts-ignore // @ts-ignore files[CURSOR_COMPATIBILITY_SYMBOL] = cursor; return files; } async listAllFiles(folder: string, extension: string, depth: number) { const files = await this.api!.listAllFiles(folder, depth > 1); const filtered = files.filter(file => this.filterFile(folder, file, extension, depth)); return filtered; } async allEntriesByFolder(folder: string, extension: string, depth: number) { const files = await allEntriesByFolder({ listAllFiles: () => this.listAllFiles(folder, extension, depth), readFile: this.api!.readFile.bind(this.api!), readFileMetadata: this.api!.readFileMetadata.bind(this.api), apiName: API_NAME, branch: this.branch, localForage, folder, extension, depth, getDefaultBranch: () => this.api!.getDefaultBranch().then(b => ({ name: b.name, sha: b.commit.id })), isShaExistsInBranch: this.api!.isShaExistsInBranch.bind(this.api!), getDifferences: (to, from) => this.api!.getDifferences(to, from), getFileId: path => this.api!.getFileId(path, this.branch), filterFile: file => this.filterFile(folder, file, extension, depth), }); return files; } entriesByFiles(files: ImplementationFile[]) { return entriesByFiles( files, this.api!.readFile.bind(this.api!), this.api!.readFileMetadata.bind(this.api), API_NAME, ); } // Fetches a single entry. getEntry(path: string) { return this.api!.readFile(path).then(data => ({ file: { path, id: null }, data: data as string, })); } getMedia(mediaFolder = this.mediaFolder) { return this.api!.listAllFiles(mediaFolder).then(files => files.map(({ id, name, path }) => { return { id, name, path, displayURL: { id, name, path } }; }), ); } getMediaDisplayURL(displayURL: DisplayURL) { this._mediaDisplayURLSem = this._mediaDisplayURLSem || semaphore(MAX_CONCURRENT_DOWNLOADS); return getMediaDisplayURL( displayURL, this.api!.readFile.bind(this.api!), this._mediaDisplayURLSem, ); } async getMediaFile(path: string) { const name = basename(path); const blob = await getMediaAsBlob(path, null, this.api!.readFile.bind(this.api!)); const fileObj = blobToFileObj(name, blob); const url = URL.createObjectURL(fileObj); const id = await getBlobSHA(blob); return { id, displayURL: url, path, name, size: fileObj.size, file: fileObj, url, }; } async persistEntry(entry: Entry, mediaFiles: AssetProxy[], options: PersistOptions) { // persistEntry is a transactional operation return runWithLock( this.lock, () => this.api!.persistFiles(entry, mediaFiles, options), 'Failed to acquire persist entry lock', ); } async persistMedia(mediaFile: AssetProxy, options: PersistOptions) { const fileObj = mediaFile.fileObj as File; const [id] = await Promise.all([ getBlobSHA(fileObj), this.api!.persistFiles(null, [mediaFile], options), ]); const { path } = mediaFile; const url = URL.createObjectURL(fileObj); return { displayURL: url, path: trimStart(path, '/'), name: fileObj!.name, size: fileObj!.size, file: fileObj, url, id, }; } deleteFile(path: string, commitMessage: string) { return this.api!.deleteFile(path, commitMessage); } traverseCursor(cursor: Cursor, action: string) { return this.api!.traverseCursor(cursor, action).then(async ({ entries, cursor: newCursor }) => { const [folder, depth, extension] = [ cursor.meta?.get('folder') as string, cursor.meta?.get('depth') as number, cursor.meta?.get('extension') as string, ]; if (folder && depth && extension) { entries = entries.filter(f => this.filterFile(folder, f, extension, depth)); newCursor = newCursor.mergeMeta({ folder, extension, depth }); } const entriesWithData = await entriesByFiles( entries, this.api!.readFile.bind(this.api!), this.api!.readFileMetadata.bind(this.api)!, API_NAME, ); return { entries: entriesWithData, cursor: newCursor, }; }); } loadMediaFile(branch: string, file: UnpublishedEntryMediaFile) { const readFile = ( path: string, id: string | null | undefined, { parseText }: { parseText: boolean }, ) => this.api!.readFile(path, id, { branch, parseText }); return getMediaAsBlob(file.path, null, readFile).then(blob => { const name = basename(file.path); const fileObj = blobToFileObj(name, blob); return { id: file.path, displayURL: URL.createObjectURL(fileObj), path: file.path, name, size: fileObj.size, file: fileObj, }; }); } async loadEntryMediaFiles(branch: string, files: UnpublishedEntryMediaFile[]) { const mediaFiles = await Promise.all(files.map(file => this.loadMediaFile(branch, file))); return mediaFiles; } async unpublishedEntries() { const listEntriesKeys = () => this.api!.listUnpublishedBranches().then(branches => branches.map(branch => contentKeyFromBranch(branch)), ); const readUnpublishedBranchFile = (contentKey: string) => this.api!.readUnpublishedBranchFile(contentKey); return unpublishedEntries(listEntriesKeys, readUnpublishedBranchFile, API_NAME); } async unpublishedEntry( collection: string, slug: string, { loadEntryMediaFiles = (branch: string, files: UnpublishedEntryMediaFile[]) => this.loadEntryMediaFiles(branch, files), } = {}, ) { const contentKey = generateContentKey(collection, slug); const data = await this.api!.readUnpublishedBranchFile(contentKey); const mediaFiles = await loadEntryMediaFiles( data.metaData.branch, data.metaData.objects.entry.mediaFiles, ); return { slug, file: { path: data.metaData.objects.entry.path, id: null }, data: data.fileData as string, metaData: data.metaData, mediaFiles, isModification: data.isModification, }; } async updateUnpublishedEntryStatus(collection: string, slug: string, newStatus: string) { // updateUnpublishedEntryStatus is a transactional operation return runWithLock( this.lock, () => this.api!.updateUnpublishedEntryStatus(collection, slug, newStatus), 'Failed to acquire update entry status lock', ); } async deleteUnpublishedEntry(collection: string, slug: string) { // deleteUnpublishedEntry is a transactional operation return runWithLock( this.lock, () => this.api!.deleteUnpublishedEntry(collection, slug), 'Failed to acquire delete entry lock', ); } async publishUnpublishedEntry(collection: string, slug: string) { // publishUnpublishedEntry is a transactional operation return runWithLock( this.lock, () => this.api!.publishUnpublishedEntry(collection, slug), 'Failed to acquire publish entry lock', ); } async getDeployPreview(collection: string, slug: string) { try { const statuses = await this.api!.getStatuses(collection, slug); const deployStatus = getPreviewStatus(statuses, this.previewContext); if (deployStatus) { const { target_url: url, state } = deployStatus; return { url, status: state }; } else { return null; } } catch (e) { return null; } } }