Merge pull request #71 from hateofhades/privateGitlab

Private Repos and Self-Hosted Gitlab
wyliodrinstudio · Jun 29, 2021 · 2cf2be2 · 2cf2be2
2 parents 32c7888 + a48f3b1
commit 2cf2be2
Show file tree

Hide file tree

Showing 4 changed files with 61 additions and 15 deletions.
diff --git a/source/plugins/libs/github/index.js b/source/plugins/libs/github/index.js
@@ -48,6 +48,12 @@ let github = {
 	async downloadFile (filePath, owner, repo, ref, responseType = 'json') {
 		let response = await Axios.get(`https://raw.githubusercontent.com/${owner}/${repo}/${ref}/${filePath}`,  {responseType: responseType,});
 		return response.data;
+	},
+	authenticate(token) {
+		return token;
+	},
+	changeURL(newURL) {
+		return newURL;
 	}
 };
 

diff --git a/source/plugins/libs/gitlab/index.js b/source/plugins/libs/gitlab/index.js
@@ -1,12 +1,18 @@
-import Axios from 'axios';
+import axios from 'axios';
 
 let gitlab = {
+	token: null,
+	urlAPI: 'https://gitlab.com',
 	async getDirListOfFiles (path, fileHierarchy, owner, repo, ref) {
-		let gitURL = `https://gitlab.com/api/v4/projects/${owner}%2F${repo}/repository/tree?path=${path}`;
+		let gitURL = `${this.urlAPI}/api/v4/projects/${owner}%2F${repo}/repository/tree?path=${path}`;
 		if (ref) {
 			gitURL += `&ref=${ref}`;
 		}
-		let response = await Axios.get(gitURL);
+		if(this.token) {
+			gitURL += `&private_token=${this.token}`;
+		}
+
+		let response = await axios.get(gitURL);
 
 		for(let item of response.data) {
 			if (item.type === 'blob') {
@@ -21,11 +27,15 @@ let gitlab = {
 		}
 	},
 	async getContentOfDir(path, owner, repo, ref) {
-		let gitURL = `https://gitlab.com/api/v4/projects/${owner}%2F${repo}/repository/tree?path=${path}`;
+		let gitURL = `${this.urlAPI}/api/v4/projects/${owner}%2F${repo}/repository/tree?path=${path}`;
 		if (ref) {
 			gitURL += `&ref=${ref}`;
 		}
-		let response = await Axios.get(gitURL);
+		if(this.token) {
+			gitURL += `&private_token=${this.token}`;
+		}
+
+		let response = await axios.get(gitURL);
 
 		let contents = {dirs: [], files: []};
 
@@ -46,8 +56,28 @@ let gitlab = {
 		return fileHierarchy;
 	},
 	async downloadFile (filePath, owner, repo, ref, responseType = 'json') {
-		let response = await Axios.get(`https://gitlab.com/${owner}/${repo}/-/raw/${ref}/${filePath}`,  {responseType: responseType,});
-		return response.data;
+		filePath = filePath.replace(/\//g, '%2F');
+
+		let gitURL = `${this.urlAPI}/api/v4/projects/${owner}%2F${repo}/repository/files/${filePath}?ref=${ref}`;
+		if(this.token) {
+			gitURL += `&private_token=${this.token}`;
+		}
+		let response = await axios.get(gitURL);	
+
+		response = Buffer.from(response.data.content, response.data.encoding);
+
+		if(responseType == 'json')
+			return JSON.parse(response.toString());
+		else return response.toString();
+	},
+	authenticate(token) {
+		this.token = token;
+	},
+	changeURL(newURL) {
+		if(newURL.endsWith('/'))
+			newURL = newURL.slice(0, -1);
+
+		this.urlAPI = newURL;
 	}
 };
 

diff --git a/source/plugins/tutorials/index.js b/source/plugins/tutorials/index.js
@@ -5,15 +5,20 @@ let studio = null;
 export function setup(options, imports, register) 
 {
 	studio = imports;
-	let platformData = 'github';
+	let platformData = {
+		platform: 'github',
+		branch: 'main',
+		token: null,
+		gitlabURL: null
+	};
 
 	let tutorials = {
 		/**
 		 * Show a list for tutorials from a repository
 		 * 
 		 * @param {String} repository - username/repository
 		 */
-		showTutorials (repository) {
+		showTutorials (repository) {		
 			let owner = repository.split('/')[0];
 			repository = repository.split('/')[1];
 			studio.workspace.showDialog (Tutorials, {

diff --git a/source/plugins/tutorials/views/Tutorials.vue b/source/plugins/tutorials/views/Tutorials.vue
@@ -56,7 +56,7 @@
 <script>
 export default {
 	name: 'Tutorials',
-	props: ['repository', 'owner', 'platformData'],
+	props: ['owner', 'repository', 'platformData'],
 	data ()
 	{
 		return  {
@@ -67,14 +67,19 @@ export default {
 		};
 	},
 	async created () {
-		if(this.platformData == 'github') this.platform = this.studio.github;
+		if(this.platformData.platform == 'github') this.platform = this.studio.github;
 		else this.platform = this.studio.gitlab;
 
-		let response = await this.platform.getContentOfDir('', this.owner, this.repository, 'main');
+		if(this.platformData.token)
+			this.platform.authenticate(this.platformData.token);
+		if(this.platformData.gitlabURL)
+			this.platform.changeURL(this.platformData.gitlabURL);
+
+		let response = await this.platform.getContentOfDir('', this.owner, this.repository, this.platformData.branch);
 		
 		let tutorials = [];
 		for (let dir of response.dirs) {
-			let tutorial = await this.platform.downloadFile(`${dir}/.project/tutorial.json`, this.owner, this.repository, 'main');
+			let tutorial = await this.platform.downloadFile(`${dir}/.project/tutorial.json`, this.owner, this.repository, this.platformData.branch);
 
 			tutorials.push(tutorial);
 			tutorial['path'] = dir;
@@ -114,7 +119,7 @@ export default {
 				let createProject = await this.studio.projects.createEmptyProject(nameProject, tutorial.language);
 				if (createProject) {
 					let dirInfos = {};
-					await this.platform.getDirListOfFiles(tutorial.path, dirInfos, this.owner, this.repository, 'main');
+					await this.platform.getDirListOfFiles(tutorial.path, dirInfos, this.owner, this.repository, this.platformData.branch);
 					let numberOfFiles = 0;
 					for (let key in dirInfos) {
 						numberOfFiles += dirInfos[key].length;
@@ -130,7 +135,7 @@ export default {
 						for (let file of dirInfos[key]) {
 						
 							let filePath = file.replace(tutorial.path, '');
-							let fileData = await this.platform.downloadFile(file, this.owner, this.repository, 'main', 'arraybuffer');
+							let fileData = await this.platform.downloadFile(file, this.owner, this.repository, this.platformData.branch, 'arraybuffer');
 
 							await this.studio.projects.newFile(createProject, filePath, Buffer.from (fileData));
 							downloadedFiles++;