refactor: types

README.md

@@ -35,7 +35,7 @@ npx @defi-wonderland/natspec-smells --include solidity --exclude solidity/(test|
 
    ```javascript
    /**
-    * For full explanation of each supported config, make sure to check the Config type below
+    * List of supported options: https://github.com/defi-wonderland/natspec-smells?tab=readme-ov-file#options
     */
 
    /** @type {import('@defi-wonderland/natspec-smells').Config} */

package.json

@@ -10,7 +10,7 @@
   "license": "MIT",
   "author": "Wonderland",
   "main": "lib/main.js",
-  "types": "lib/main.js",
+  "types": "lib/types.d.ts",
   "bin": "./lib/main.js",
   "scripts": {
     "build": "tsc",

src/main.ts

@@ -4,12 +4,9 @@ import { hideBin } from 'yargs/helpers';
 import { globSync } from 'fast-glob';
 import { getProjectCompiledSources } from './utils';
 import { Processor } from './processor';
-import { Config } from './types/config';
+import { Config } from './types';
 import { Validator } from './validator';
 
-// export config for type reference
-export { Config } from './types/config';
-
 (async () => {
   const config: Config = getArguments();
 

src/processor.ts

@@ -1,7 +1,7 @@
 import fs from 'fs/promises';
 import { Validator } from './validator';
 import { SourceUnit, FunctionDefinition, ContractDefinition } from 'solc-typed-ast';
-import { NodeToProcess } from './types/solc-typed-ast.d';
+import { NodeToProcess } from './types';
 import { getLineNumberFromSrc, parseNodeNatspec } from './utils';
 
 interface IWarning {

src/types.ts

@@ -0,0 +1,51 @@
+import {
+  EnumDefinition,
+  ErrorDefinition,
+  EventDefinition,
+  FunctionDefinition,
+  ModifierDefinition,
+  StructDefinition,
+  VariableDeclaration,
+} from 'solc-typed-ast';
+
+export interface Config {
+  include: string; // Required: Glob pattern of files to process.
+  exclude: string[]; // Optional: Glob patterns of files to exclude.
+  root: string; // Optional: Project root directory.
+  enforceInheritdoc: boolean; // Optional: True if all external and public functions should have @inheritdoc.
+  constructorNatspec: boolean; // Optional: True if the constructor should have natspec.
+}
+
+export interface NatspecDefinition {
+  name?: string;
+  content: string;
+}
+
+export interface Natspec {
+  inheritdoc?: NatspecDefinition;
+  tags: NatspecDefinition[];
+  params: NatspecDefinition[];
+  returns: NatspecDefinition[];
+}
+
+export interface ASTNodeRawDocumentation {
+  id: number;
+  nodeType: string;
+  src: string;
+  text: string;
+}
+
+export interface ASTNodeRaw {
+  name: string;
+  kind: string;
+  documentation?: ASTNodeRawDocumentation;
+}
+
+export type NodeToProcess =
+  | FunctionDefinition
+  | EnumDefinition
+  | ErrorDefinition
+  | EventDefinition
+  | ModifierDefinition
+  | VariableDeclaration
+  | StructDefinition;

src/utils.ts

@@ -1,8 +1,7 @@
 import fs from 'fs/promises';
 import path from 'path';
 import { ASTKind, ASTReader, SourceUnit, compileSol } from 'solc-typed-ast';
-import { Natspec, NatspecDefinition } from './types/natspec.d';
-import { NodeToProcess } from './types/solc-typed-ast.d';
+import { Natspec, NatspecDefinition, NodeToProcess } from './types';
 
 export async function getSolidityFiles(dir: string): Promise<string[]> {
   let files = await fs.readdir(dir, { withFileTypes: true });

src/validator.ts

@@ -1,6 +1,4 @@
-import { Config } from './types/config';
-import { Natspec } from './types/natspec';
-import { NodeToProcess } from './types/solc-typed-ast';
+import { Config, Natspec, NodeToProcess } from './types';
 import {
   EnumDefinition,
   ErrorDefinition,

test/mocks.ts

@@ -1,6 +1,5 @@
-import { Natspec } from '../src/types/natspec.d';
 import { SourceUnit, ContractDefinition, FunctionDefinition } from 'solc-typed-ast';
-import { NodeToProcess } from '../src/types/solc-typed-ast.d';
+import { Natspec, NodeToProcess } from '../src/types';
 
 export function mockNatspec(mockNatspec: Partial<Natspec>): Natspec {
   const natspec: Natspec = {

test/processor.test.ts

@@ -1,5 +1,5 @@
-import { Config } from '../src/types/config';
 import { ContractDefinition, FunctionDefinition, UserDefinedType, UsingForDirective } from 'solc-typed-ast';
+import { Config } from '../src/types';
 import { getFileCompiledSource } from './utils';
 import { Processor } from '../src/processor';
 import { Validator } from '../src/validator';

test/validator.test.ts

@@ -1,8 +1,7 @@
+import { ContractDefinition } from 'solc-typed-ast';
 import { Validator } from '../src/validator';
 import { getFileCompiledSource } from './utils';
-import { NodeToProcess } from '../src/types/solc-typed-ast.d';
-import { ContractDefinition } from 'solc-typed-ast';
-import { Config } from '../src/types/config';
+import { Config, NodeToProcess } from '../src/types';
 
 describe('Validator', () => {
   let contract: ContractDefinition;