@dhiwise/component-tagger
Version:
Automatically annotate JSX components and HTML elements with data attributes
5 lines (4 loc) • 9.07 kB
JavaScript
;var z=Object.create;var x=Object.defineProperty;var Q=Object.getOwnPropertyDescriptor;var Y=Object.getOwnPropertyNames;var Z=Object.getPrototypeOf,ee=Object.prototype.hasOwnProperty;var te=(e,t)=>{for(var r in t)x(e,r,{get:t[r],enumerable:!0})},$=(e,t,r,n)=>{if(t&&typeof t=="object"||typeof t=="function")for(let o of Y(t))!ee.call(e,o)&&o!==r&&x(e,o,{get:()=>t[o],enumerable:!(n=Q(t,o))||n.enumerable});return e};var G=(e,t,r)=>(r=e!=null?z(Z(e)):{},$(t||!e||!e.__esModule?x(r,"default",{value:e,enumerable:!0}):r,e)),re=e=>$(x({},"__esModule",{value:!0}),e);var le={};te(le,{createVitePlugin:()=>V,default:()=>ae});module.exports=re(le);var k=require("unplugin");var S=G(require("path")),D=require("@babel/parser"),H=require("@babel/core"),N=G(require("magic-string"));var I=G(require("picomatch"));var A=new Set(["Fragment","Suspense","StrictMode","Profiler","React.Fragment"]),B=["three","@react-three/fiber","@react-three/drei","@react-three/rapier","@react-three/a11y","@react-three/csg","@react-three/flex","@react-three/gpu-pathtracer","@react-three/postprocessing","@react-three/cannon","@react-three/xr","@react-three/test-renderer","three/addons","three/examples"],_=["object3D","audioListener","positionalAudio","mesh","batchedMesh","instancedMesh","scene","sprite","lOD","skinnedMesh","skeleton","bone","lineSegments","lineLoop","points","group","camera","perspectiveCamera","orthographicCamera","cubeCamera","arrayCamera","instancedBufferGeometry","bufferGeometry","boxBufferGeometry","circleBufferGeometry","coneBufferGeometry","cylinderBufferGeometry","dodecahedronBufferGeometry","extrudeBufferGeometry","icosahedronBufferGeometry","latheBufferGeometry","octahedronBufferGeometry","planeBufferGeometry","polyhedronBufferGeometry","ringBufferGeometry","shapeBufferGeometry","sphereBufferGeometry","tetrahedronBufferGeometry","torusBufferGeometry","torusKnotBufferGeometry","tubeBufferGeometry","wireframeGeometry","tetrahedronGeometry","octahedronGeometry","icosahedronGeometry","dodecahedronGeometry","polyhedronGeometry","tubeGeometry","torusKnotGeometry","torusGeometry","sphereGeometry","ringGeometry","planeGeometry","latheGeometry","shapeGeometry","extrudeGeometry","edgesGeometry","coneGeometry","cylinderGeometry","circleGeometry","boxGeometry","capsuleGeometry","material","shadowMaterial","spriteMaterial","rawShaderMaterial","shaderMaterial","pointsMaterial","meshPhysicalMaterial","meshStandardMaterial","meshPhongMaterial","meshToonMaterial","meshNormalMaterial","meshLambertMaterial","meshDepthMaterial","meshDistanceMaterial","meshBasicMaterial","meshMatcapMaterial","lineDashedMaterial","lineBasicMaterial","primitive","light","spotLightShadow","spotLight","pointLight","rectAreaLight","hemisphereLight","directionalLightShadow","directionalLight","ambientLight","lightShadow","ambientLightProbe","hemisphereLightProbe","lightProbe","spotLightHelper","skeletonHelper","pointLightHelper","hemisphereLightHelper","gridHelper","polarGridHelper","directionalLightHelper","cameraHelper","boxHelper","box3Helper","planeHelper","arrowHelper","axesHelper","texture","videoTexture","dataTexture","dataTexture3D","compressedTexture","cubeTexture","canvasTexture","depthTexture","raycaster","vector2","vector3","vector4","euler","matrix3","matrix4","quaternion","bufferAttribute","float16BufferAttribute","float32BufferAttribute","float64BufferAttribute","int8BufferAttribute","int16BufferAttribute","int32BufferAttribute","uint8BufferAttribute","uint16BufferAttribute","uint32BufferAttribute","instancedBufferAttribute","color","fog","fogExp2","shape","colorShiftMaterial"];var u={red:e=>`\x1B[31m${e}\x1B[0m`,green:e=>`\x1B[32m${e}\x1B[0m`,yellow:e=>`\x1B[33m${e}\x1B[0m`,blue:e=>`\x1B[34m${e}\x1B[0m`,cyan:e=>`\x1B[36m${e}\x1B[0m`};function O(e){let t=new Set;for(let r of e.program.body)if(r.type==="ImportDeclaration"){let n=r,o=n.source.value;if(B.some(i=>o===i||o.startsWith(i+"/")))for(let i of n.specifiers)i.type==="ImportSpecifier"||i.type==="ImportDefaultSpecifier"?t.add(i.local.name):i.type==="ImportNamespaceSpecifier"&&t.add(`${i.local.name}.*`)}return t}var P=new Map;function ne(e){let t=e.sort().join(",");if(!P.has(t)){let r=e.map(o=>`**/*${o}`),n=(0,I.default)(r);P.set(t,n)}return P.get(t)}function b(e,t){let r=t.extensions||[".jsx",".tsx",".js",".ts"];return ne(r)(e)?![".test.",".spec.","__tests__","__mocks__"].some(i=>e.includes(i)):!1}function v(e,t,r){return t.includeElements&&t.includeElements.length>0?t.includeElements.includes(e):t.shouldTag?t.shouldTag(e):!(t.excludeElements&&t.excludeElements.includes(e)||A.has(e)||r&&oe(e,r))}function oe(e,t){if(t.has(e)||_.includes(e))return!0;for(let r of t)if(r.endsWith(".*")){let n=r.slice(0,-2);if(e.startsWith(`${n}.`))return!0}return!1}function ie(e){if(e){if(e.type==="StringLiteral")return e.value;if(e.type==="JSXExpressionContainer"){let t=e.expression;if(t.type==="StringLiteral")return t.value;if(t.type==="TemplateLiteral"&&t.quasis.length===1)return t.quasis[0].value.raw;if(t.type==="BinaryExpression"&&t.operator==="+"){let r=t.left.type==="StringLiteral"?t.left.value:"",n=t.right.type==="StringLiteral"?t.right.value:"";return r+n}if(t.type==="Identifier")return`[var:${t.name}]`}}}function X(e,t,r){let n={},o=["className","id","src","alt","href","type","name","value"],a=new Set([...o,...t.extractAttributes||[]]);for(let s of e.attributes)if(s.type==="JSXAttribute"&&s.name.type==="JSXIdentifier"){let f=s.name.name;if(!a.has(f))continue;if(s.value){let m=ie(s.value);m!==void 0&&(n[f]=m)}else n[f]="true"}else s.type==="JSXSpreadAttribute"&&(n["[spread]"]="true");let i="";return r&&r.children&&(i=r.children.map(s=>s.type==="JSXText"?s.value.trim():s.type==="JSXExpressionContainer"&&s.expression.type==="StringLiteral"?s.expression.value:"").filter(Boolean).join(" ").trim()),i&&(n.textContent=i),n}function F(e){let t=e.name;return t.type==="JSXIdentifier"?t.name:t.type==="JSXMemberExpression"?se(t):"Unknown"}function se(e){return`${e.object.name}.${e.property.name}`}function g(e){return e.replace(/&/g,"&").replace(/"/g,""").replace(/'/g,"'").replace(/</g,"<").replace(/>/g,">")}function R(e,t,r){return`${e}:${t}:${r}`}function c(e,t){if(!t.verbose)return;let r=e;e.includes("Error")||e.includes("error")?r=u.red(e):e.includes("Tagged")||e.includes("Summary")?r=u.green(e):e.includes("Processing")||e.includes("started")?r=u.blue(e):r=u.yellow(e),console.log(`[component-tagger] ${r}`)}function w(e){return u.green(`Component Tagger Summary:
- Total files scanned: ${u.cyan(e.totalFiles)}
- Files processed: ${u.cyan(e.processedFiles)}
- Elements tagged: ${u.cyan(e.totalElements)}`)}async function j(e,t,r){if(!b(t,r)||t.includes("node_modules")&&!r.processNodeModules||r.excludeDirectories&&r.excludeDirectories.some(i=>t.includes(i)))return null;let n=process.cwd(),o=S.relative(n,t),a=S.basename(t);c(`Processing file: ${o}`,r);try{let s=(0,D.parse)(e,{sourceType:"module",plugins:["jsx","typescript"]}),f=O(s),m=new N.default(e),E=0,h=null;return(0,H.traverse)(s,{JSXElement:T=>{h=T.node},JSXOpeningElement:T=>{if(!h)return;let d=T.node,C=F(d);if(!v(C,r,f))return;let q=X(d,r,h),L={elementName:C};Object.entries(q).forEach(([M,p])=>{L[M]=p});let y=d.loc?.start?.line??0,J=d.loc?.start?.column??0,W=h?.loc?.end?.line??y,K=r.generateComponentId?r.generateComponentId(o,y,J):R(o,y,J),l=` ${r.attributePrefix}-id="${g(K)}"`;if(r.includeLegacyAttributes&&(l+=` ${r.attributePrefix}-path="${g(o)}"`,l+=` ${r.attributePrefix}-line="${y}"`,l+=` ${r.attributePrefix}-end-line="${W}"`,l+=` ${r.attributePrefix}-file="${g(a)}"`,l+=` ${r.attributePrefix}-name="${g(C)}"`),r.includeContentAttribute){let M=JSON.stringify(L),p=encodeURIComponent(M);r.maxContentLength&&p.length>r.maxContentLength&&(p=p.substring(0,r.maxContentLength)+"..."),l+=` ${r.attributePrefix}-content="${p}"`}m.appendLeft(d.name.end??0,l),c(l.toString(),r),E++}}),E>0?(c(`Tagged ${E} components in ${o}`,r),{code:m.toString(),map:r.sourceMaps?m.generateMap({hires:!0}):null}):null}catch(i){return console.error(`Error processing file ${o}:`,i),null}}var U=(0,k.createUnplugin)((e={})=>{let t={extensions:[".jsx",".tsx",".js",".ts"],verbose:!1,attributePrefix:"data-component",includeContentAttribute:!0,maxContentLength:1e3,includeLegacyAttributes:!0,sourceMaps:!0,excludeDirectories:[],processNodeModules:!1,...e},r={totalFiles:0,processedFiles:0,totalElements:0};return{name:"component-tagger",transformInclude(n){return b(n,t)},async transform(n,o){if(o.includes("node_modules")&&!t.processNodeModules||t.excludeDirectories&&t.excludeDirectories.some(i=>o.includes(i)))return null;r.totalFiles++;let a=await j(n,o,t);if(a){r.processedFiles++;let i=(a.code.match(new RegExp(`${t.attributePrefix}-id="`,"g"))||[]).length;r.totalElements+=i}else r.processedFiles++;return a},buildStart(){c("Component tagger plugin started",t),r.totalFiles=0,r.processedFiles=0,r.totalElements=0},buildEnd(){c(w(r),t)}}});function V(e={}){let t=U.vite(e);return{...Array.isArray(t)?t[0]:t,enforce:"pre"}}var ae=V;0&&(module.exports={createVitePlugin});