js 拾取器封装

var $=function(r){
    function cut(str){
        return str.slice(1);
    }
    try{
        var regId=/#(.*)/;
        var regClass=/\.(.*)/;
        var regHtmlTag=/<(.*)>/;
        if(typeof r == "function"){
            //alert("function");
            return window.onload=r;
        }else if(typeof r == "string" && regId.test(r)){
            //alert("id");
            return document.getElementById(cut(r));
        }else if(typeof r == "string" && regClass.test(r)){
            //alert("class");
            var classArr=[];
            var classAll=document.getElementsByTagName("*");
            for(var i=0; i<classAll.length; i++){
                if(classAll[i].className==cut(r)){
                    classArr.push(classAll[i]);
                }
            }
            return classArr;
        }else if(typeof r == "string" && regHtmlTag.test(r)){
            //alert("tag");
            var tagArr=[];
            var tagAll=document.getElementsByTagName("*");
            for(var i=0; i<tagAll.length; i++){
                if("<"+tagAll[i].tagName+">"==r.toUpperCase()){
                    tagArr.push(tagAll[i]);
                }
            }
            return tagArr
        }else{
            //alert("object");
            return r;
        }
    }catch(e){
        alert(e.message)
    }finally{
    }
}