123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243
import util from '../shared/util.ts'import type { Module } from '../types.d.ts'import type { Aleph } from './aleph.ts'import { getAlephPkgUri } from './helper.ts'
export type DependencyGraph = {  specifier: string  external?: boolean  isPreload?: boolean  isDynamic?: boolean  isLoop?: boolean  isShared?: boolean  size?: number  sourceSize?: number  deps: DependencyGraph[]}
export class Analyzer {  #aleph: Aleph  #entries: DependencyGraph[]
  constructor(app: Aleph) {    this.#aleph = app    this.#entries = [      Analyzer.blankDependencyGraph('virtual:/vendor.js', true),      Analyzer.blankDependencyGraph('virtual:/common.js', true),    ]  }
  get entries() {    return [...this.#entries]  }
  reset() {    const { framework } = this.#aleph.config    const bootstrapModuleUrl = `${getAlephPkgUri()}/framework/${framework}/bootstrap.ts`
    this.#entries = [      Analyzer.blankDependencyGraph('virtual:/vendor.js', true),      Analyzer.blankDependencyGraph('virtual:/common.js', true),    ]
    // main.js    this.#entries.push(this.createDependencyGraph({      specifier: 'virtual:/main.js',      deps: [        { specifier: bootstrapModuleUrl }      ],    }, true))
    // app.js    const appMoudle = this.#aleph.getModule('app')    if (appMoudle) {      this.#entries.push(this.createDependencyGraph(appMoudle, true))    }  }
  addEntry(module: Module) {    if (this.#entries.findIndex(c => module.specifier === c.specifier) === -1) {      let isShared = false      for (const eg of this.#entries) {        Analyzer.walkDependencyGraph(eg, graph => {          const eq = graph.specifier === module.specifier          if (eq && !graph.external && !graph.isDynamic && !graph.isLoop && !graph.isShared) {            this.#entries.push({ ...graph, isPreload: true })            graph.isShared = true            graph.deps = []            isShared = true            return false // break walking          }        })        if (isShared) {          return        }      }      this.#entries.push(this.createDependencyGraph(module, false))    }  }
  private createDependencyGraph(    module: Pick<Module, 'specifier' | 'deps' | 'external'>,    isPreload?: boolean,    __tracing = new Set<string>()  ): DependencyGraph {    const { specifier, external } = module    const deps: Module["deps"] = []    const graph: DependencyGraph = {      specifier,      external,      deps: []    }
    if (__tracing.size == 0 && isPreload) {      graph.isPreload = true    }
    if (external) {      return graph    }
    if (__tracing.has(specifier)) {      graph.isLoop = true      return graph    }    __tracing.add(specifier)
    module.deps.forEach(dep => {      if (        this.#aleph.getModule(dep.specifier) !== null &&        deps.findIndex(({ specifier, isDynamic }) => dep.specifier === specifier && dep.isDynamic === isDynamic) === -1      ) {        deps.push(dep)      }    })    graph.deps = deps.filter(({ specifier }) => this.#aleph.getModule(specifier) !== null)      .map(dep => {        const depMod = this.#aleph.getModule(dep.specifier)!        const isRemote = util.isLikelyHttpURL(dep.specifier)        const sharedGraph = this.#entries[isRemote ? 0 : 1]
        // external dependency        if (depMod.external) {          return {            ...Analyzer.blankDependencyGraph(dep.specifier),            external: true,            isDynamic: dep.isDynamic,          }        }
        // check the dep whether it is global shared        let isShared = false        if (!isRemote) {          for (const sg of this.entries) {            if (sg.isPreload && sg.specifier === dep.specifier) {              isShared = true              break            }          }        }        if (!isShared) {          for (const graph of sharedGraph.deps) {            isShared = graph.specifier === dep.specifier            if (isShared) {              break            }          }        }        if (!isShared) {          Analyzer.walkDependencyGraph(sharedGraph, graph => {            const eq = graph.specifier === dep.specifier            if (eq && !graph.external && !graph.isDynamic && !graph.isLoop && !graph.isShared) {              sharedGraph.deps.push({ ...graph })              graph.isShared = true              graph.isLoop = true              graph.deps = []              isShared = true              return false // break walking            }          })        }        if (!isShared && !dep.isDynamic) {          if (isPreload) {            const g = this.createDependencyGraph(depMod, isPreload, __tracing)            sharedGraph.deps.push(g)            isShared = true          } else {            for (const eg of this.#entries) {              if (!eg.isPreload) {                if (eg.specifier === dep.specifier) {                  eg.isPreload = true                  isShared = true                } else {                  Analyzer.walkDependencyGraph(eg, graph => {                    const eq = graph.specifier === dep.specifier                    if (eq && !graph.external && !graph.isDynamic && !graph.isLoop && !graph.isShared) {                      sharedGraph.deps.push({ ...graph })                      graph.isShared = true                      graph.deps = []                      isShared = true                      return false // break walking                    }                  })                }                if (isShared) {                  break                }              }            }          }        }        if (isShared) {          return {            ...Analyzer.blankDependencyGraph(dep.specifier),            isShared,            isDynamic: dep.isDynamic,          }        }
        // dynamic dep        if (dep.isDynamic) {          if (this.#entries.findIndex(c => dep.specifier === c.specifier) === -1) {            this.#entries.push(this.createDependencyGraph(depMod))          }          return {            ...Analyzer.blankDependencyGraph(dep.specifier),            isDynamic: true,          }        }
        return this.createDependencyGraph(depMod, isPreload, __tracing)      })
    return graph  }
  static blankDependencyGraph(specifier: string, isPreload?: boolean): DependencyGraph {    return {      specifier,      isPreload,      deps: []    }  }
  static walkDependencyGraph(    graph: DependencyGraph,    callback: (graph: DependencyGraph) => void | false,    __tracing: Set<string> = new Set()  ): void | false {    if (__tracing.has(graph.specifier)) {      return    }    __tracing.add(graph.specifier)    for (const dg of graph.deps) {      if (callback(dg) === false) {        return false      }      if ((Analyzer.walkDependencyGraph(dg, callback, __tracing)) === false) {        return false      }    }  }}