Overleaf/services/web/app/coffee/Features/Labels/LabelsHandler.coffee

ProjectEntityHandler = require "../Project/ProjectEntityHandler"
DocumentUpdaterHandler = require('../DocumentUpdater/DocumentUpdaterHandler')


module.exports = LabelsHandler =

	labelCaptureRegex: () ->
		/\\label\{([^\}\n\\]{0,80})\}/g

	packageCaptureRegex: () ->
		/^\\usepackage(?:\[((?:.|\n)*?)])?\s*?{((?:.|\n)*?)}/gm

	getAllMetaForProject: (projectId, callback=(err, projectMeta)->) ->
		DocumentUpdaterHandler.flushProjectToMongo projectId, (err) ->
			if err?
				return callback err
			ProjectEntityHandler.getAllDocs projectId, (err, docs) ->
				if err?
					return callback err
				projectMeta = LabelsHandler.extractMetaFromProjectDocs docs
				callback null, projectMeta

	getMetaForDoc: (projectId, docId, callback=(err, docMeta)->) ->
		DocumentUpdaterHandler.flushDocToMongo projectId, docId, (err) ->
			if err?
				return callback err
			ProjectEntityHandler.getDoc projectId, docId, (err, lines) ->
				if err?
					return callback err
				docMeta = LabelsHandler.extractMetaFromDoc lines
				callback null, docMeta

	extractMetaFromDoc: (lines) ->
		docMeta = {labels: [], packages: []}
		label_re = LabelsHandler.labelCaptureRegex()
		package_re = LabelsHandler.packageCaptureRegex()
		for line in lines
			while labelMatch = label_re.exec line
				if labelMatch[1]
					docMeta.labels.push labelMatch[1]
			while packageMatch = package_re.exec line
				if packageMatch[2]
					for pkg in packageMatch[2].split ','
						docMeta.packages.push pkg.trim()
		return docMeta

	extractMetaFromProjectDocs: (projectDocs) ->
		projectMeta = {}
		for _path, doc of projectDocs
			projectMeta[doc._id] = LabelsHandler.extractMetaFromDoc doc.lines
		return projectMeta