cleanup code

packages/state/vite/autoKey.ts

@@ -1,5 +1,5 @@
 import { Plugin } from 'vite';
-import { astAddAutoNaming } from './internal/astAutoNaming';
+import { astAddAutoNaming } from './babel/astAutoNaming';
 import { parseModule } from 'magicast';
 
 interface StatebuilderAutonamingOptions {

packages/state/vite/internal/replaceStateProviderDirective.ts → packages/state/vite/babel/replaceStateProviderDirective.ts

@@ -1,6 +1,11 @@
 import * as t from '@babel/types';
 import type * as babel from '@babel/core';
 
+/**
+ * @experimental
+ *
+ * Search for components that use `use stateprovider` directive, then wrap them into StateProvider.
+ */
 export function babelReplaceStateProviderDirective(): babel.PluginObj<any> {
   return {
     name: 'statebuilder:stateprovider-directive',

packages/state/vite/index.ts

@@ -26,7 +26,7 @@ export interface StateBuilderPluginOptions {
     /**
      * Transform components that make use of 'use stateprovider'.
      */
-    transformStateProviderDirective: boolean;
+    transformStateProviderDirective?: boolean;
   };
 }
 

packages/state/vite/stateProviderDirective.ts

@@ -1,7 +1,7 @@
 import * as babel from '@babel/core';
 import { basename } from 'node:path';
 import { Plugin } from 'vite';
-import { babelReplaceStateProviderDirective } from './internal/replaceStateProviderDirective';
+import { babelReplaceStateProviderDirective } from './babel/replaceStateProviderDirective';
 
 export function stateProviderDirective(): Plugin {
   return {