68 lines
2.1 KiB
TypeScript
68 lines
2.1 KiB
TypeScript
|
/*---------------------------------------------------------------------------------------------
|
||
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
||
|
|
*--------------------------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
import { API as GitAPI, RemoteSourceProvider, RemoteSource, Repository } from './typings/git';
|
||
|
|
import { getOctokit } from './auth';
|
||
|
|
import { Octokit } from '@octokit/rest';
|
||
|
|
import { publishRepository } from './publish';
|
||
|
|
|
||
|
|
function asRemoteSource(raw: any): RemoteSource {
|
||
|
|
return {
|
||
|
|
name: `$(github) ${raw.full_name}`,
|
||
|
|
description: raw.description || undefined,
|
||
|
|
url: raw.clone_url
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
export class GithubRemoteSourceProvider implements RemoteSourceProvider {
|
||
|
|
|
||
|
|
readonly name = 'GitHub';
|
||
|
|
readonly icon = 'github';
|
||
|
|
readonly supportsQuery = true;
|
||
|
|
|
||
|
|
private userReposCache: RemoteSource[] = [];
|
||
|
|
|
||
|
|
constructor(private gitAPI: GitAPI) { }
|
||
|
|
|
||
|
|
async getRemoteSources(query?: string): Promise<RemoteSource[]> {
|
||
|
|
const octokit = await getOctokit();
|
||
|
|
const [fromUser, fromQuery] = await Promise.all([
|
||
|
|
this.getUserRemoteSources(octokit, query),
|
||
|
|
this.getQueryRemoteSources(octokit, query)
|
||
|
|
]);
|
||
|
|
|
||
|
|
const userRepos = new Set(fromUser.map(r => r.name));
|
||
|
|
|
||
|
|
return [
|
||
|
|
...fromUser,
|
||
|
|
...fromQuery.filter(r => !userRepos.has(r.name))
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
private async getUserRemoteSources(octokit: Octokit, query?: string): Promise<RemoteSource[]> {
|
||
|
|
if (!query) {
|
||
|
|
const user = await octokit.users.getAuthenticated({});
|
||
|
|
const username = user.data.login;
|
||
|
|
const res = await octokit.repos.listForUser({ username, sort: 'updated', per_page: 100 });
|
||
|
|
this.userReposCache = res.data.map(asRemoteSource);
|
||
|
|
}
|
||
|
|
|
||
|
|
return this.userReposCache;
|
||
|
|
}
|
||
|
|
|
||
|
|
private async getQueryRemoteSources(octokit: Octokit, query?: string): Promise<RemoteSource[]> {
|
||
|
|
if (!query) {
|
||
|
|
return [];
|
||
|
|
}
|
||
|
|
|
||
|
|
const raw = await octokit.search.repos({ q: query, sort: 'updated' });
|
||
|
|
return raw.data.items.map(asRemoteSource);
|
||
|
|
}
|
||
|
|
|
||
|
|
publishRepository(repository: Repository): Promise<void> {
|
||
|
|
return publishRepository(this.gitAPI, repository);
|
||
|
|
}
|
||
|
|
}
|