JS中封装函数怎么弄？

这次月末出的考试题，学艺不精没有通过考试，事后请老手们帮解答一下！

var $=function(str){
	var element=null;
	if(str.length>0){
		switch (str[0]){
			case '.':
			element=document.getElementsByClassName(str.substr(1,str.length-1));
				break;
			case '#':
			element=document.getElementById(str.substr(1,str.length-1))
				break;
			default:
			element=document.getElementsByTagName(str);
				break;
		}
	}
	element.on=function(type,listener){
		element.addEventListener(type,listener);
	}
	return element;
}

 let $ = el => document.querySelectorAll(el)