Support 'lang' option.

ISKU · Apr 4, 2018 · b0ca87b · b0ca87b
1 parent 2f7bf9c
commit b0ca87b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 5 deletions.
diff --git a/app.js b/app.js
@@ -125,13 +125,18 @@ function downloadSource() {
 							console.log('* Failed to download source number #' + sourceNumber + ', problem number #' + problemNumber + '\n');
 							next(index - 1);
 						} else if (option.private(info, userInfo)) {
-							saveSource(sourceTree, sourceName, info, function() {
-								gitAll(info, function() {
-									next(index - 1);
+							if (option.lang(language, userInfo)) {
+								saveSource(sourceTree, sourceName, info, function() {
+									gitAll(info, function() {
+										next(index - 1);
+									});
 								});
-							});
+							} else {
+								console.log('* Problem #' + problemNumber + ' is not a ' + userInfo.lang + ' language.');
+								next(index - 1);
+							}
 						} else {
-							console.log('* Problem #' + info.problemNumber + ' is private.');
+							console.log('* Problem #' + problemNumber + ' is private.');
 							next(index - 1);
 						}
 					});

diff --git a/option.js b/option.js
@@ -58,6 +58,14 @@ exports.sourceName = function(problemNumber, language, userInfo) {
 	return problemNumber + getExt(language);
 }
 
+exports.lang = function(language, userInfo) {
+	if (userInfo.hasOwnProperty('lang'))
+		if (userInfo.lang.toUpperCase() == language.toUpperCase())
+			return true;
+
+	return false;
+}
+
 function replaceNumber(value, problemNumber) {
 	return value.replace(/\[NO\]/gi, problemNumber);
 }