iframe-proxy | Sunbelt Computer Software

211 lines (195 loc) · 5.73 KB
import { Gitlab } from '@gitbeaker/node';
  DiscussionNote,
  DiscussionSchema,
  IssueSchema,
  MergeRequestSchema,
  MilestoneSchema,
  NoteSchema,
  UserSchema,
} from '@gitbeaker/core/dist/types/types';
import { GitlabSettings } from './settings';
import axios from 'axios';
export type GitLabDiscussion = DiscussionSchema;
export type GitLabDiscussionNote = DiscussionNote;
export type GitLabIssue = IssueSchema;
export type GitLabNote = NoteSchema;
export type GitLabUser = Omit<UserSchema, 'created_at'>;
export type GitLabMilestone = MilestoneSchema;
export type GitLabMergeRequest = MergeRequestSchema;
export class GitlabHelper {
  // Wait for this issue to be resolved
  // https://github.com/jdalrymple/gitbeaker/issues/793
  gitlabApi: InstanceType<typeof Gitlab>;
  gitlabUrl?: string;
  gitlabToken: string;
  gitlabProjectId: number;
  archived?: boolean;
  sessionCookie: string;
  host: string;
  projectPath?: string;
  allBranches: any;
  constructor(
    gitlabApi: InstanceType<typeof Gitlab>,
    gitlabSettings: GitlabSettings
    this.gitlabApi = gitlabApi;
    this.gitlabUrl = gitlabSettings.url;
    this.gitlabToken = gitlabSettings.token;
    this.gitlabProjectId = gitlabSettings.projectId;
    this.host = gitlabSettings.url ? gitlabSettings.url : 'https://gitlab.com';
    this.host = this.host.endsWith('/')
      ? this.host.substring(0, this.host.length - 1)
      : this.host;
    this.archived = gitlabSettings.listArchivedProjects ?? true;
    this.sessionCookie = gitlabSettings.sessionCookie;
    this.allBranches = null;
   * List all projects that the GitLab user is associated with.
  async listProjects() {
      let projects;
      if (this.archived) {
        projects = await this.gitlabApi.Projects.all({ membership: true });
      } else {
        projects = await this.gitlabApi.Projects.all({ membership: true, archived: this.archived });
      // print each project with info
      for (let project of projects) {
        console.log(
          project.id.toString(),
          '\t',
          project.name,
          '\t--\t',
          project['description']
      // instructions for user
      console.log('\n\n');
      console.log(
        'Select which project ID should be transported to github. Edit the settings.js accordingly. (gitlab.projectID)'
      console.log('\n\n');
    } catch (err) {
      console.error('An Error occured while fetching all GitLab projects:');
      console.error(err);
   * Stores project path in a field
  async registerProjectPath(project_d: number) {
      const project = await this.gitlabApi.Projects.show(project_d);
      this.projectPath = project['path_with_namespace'];
    } catch (err) {
      console.error('An Error occured while fetching all GitLab projects:');
      console.error(err);
   * Gets all notes for a given issue.
  async getIssueNotes(issueIid: number): Promise<GitLabNote[]> {
      return await this.gitlabApi.IssueNotes.all(
        this.gitlabProjectId,
        issueIid,
    } catch (err) {
      console.error(`Could not fetch notes for GitLab issue #${issueIid}.`);
      return [];
   * Gets attachment using http get
  async getAttachment(relurl: string) {
      const attachmentUrl = this.host + '/-/project/' + this.gitlabProjectId + relurl;
      const data = (
        await axios.get(attachmentUrl, {
          responseType: 'arraybuffer',
          headers: {
            // HACK: work around GitLab's API lack of GET for attachments
            // See https://gitlab.com/gitlab-org/gitlab/-/issues/24155
            Cookie: `_gitlab_session=${this.sessionCookie}`,
      ).data;
      return Buffer.from(data, 'binary');
    } catch (err) {
      console.error(`Could not download attachment ${relurl}: ${err.response.statusText}`);
      console.error('Is your session cookie still valid?');
      return null;
   * Gets all branches.
  async getAllBranches() {
    if (!this.allBranches) {
      this.allBranches = await this.gitlabApi.Branches.all(
        this.gitlabProjectId
    return this.allBranches as any[];
  async getMergeRequestApprovals(mergeRequestIid: number): Promise<string[]> {
      let approvals = await this.gitlabApi.MergeRequestApprovals.configuration(
        this.gitlabProjectId,
          mergerequestIid: mergeRequestIid,
      if (approvals.approved_by) {
        return approvals.approved_by.map(user => user.user.username);
      console.log(`No approvals found for GitLab merge request !${mergeRequestIid}.`)
    } catch (err) {
      console.error(
        `Could not fetch approvals for GitLab merge request !${mergeRequestIid}: ${err}`
      throw err;
    return [];
   * Gets all notes for a given merge request.
  async getAllMergeRequestNotes(pullRequestIid: number): Promise<GitLabNote[]> {
      return this.gitlabApi.MergeRequestNotes.all(
        this.gitlabProjectId,
        pullRequestIid,
    } catch (err) {
      console.error(
        `Could not fetch notes for GitLab merge request #${pullRequestIid}.`
      return [];
   * Gets all notes for a given merge request.
  async getAllMergeRequestDiscussions(pullRequestIid: number): Promise<GitLabDiscussion[]> {
      return this.gitlabApi.MergeRequestDiscussions.all(
        this.gitlabProjectId,
        pullRequestIid,
    } catch (err) {
      console.error(
        `Could not fetch notes for GitLab merge request #${pullRequestIid}.`
      return [];
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

gitlabHelper.ts

Latest commit

History

gitlabHelper.ts

File metadata and controls
