import Protolude hiding (check, moduleName, trace)

import Data.Graph as G (graphFromEdges, reachable)

import Data.Map qualified as M

import Data.Set qualified as S

getReExported :: P.ExternsFile -> P.ExternsFile -> ModuleRefsMap -> RefsWithStatus

getReExported newExts oldExts depsDiffsMap =

where

checkRe (mn, ref)

| (mn, ref) `notElem` newReExports = Just (ref, Removed)

| Just True <- elem ref <$> M.lookup mn depsDiffsMap = Just (ref, Updated)

getChanged newExts oldExts depsDiffsMap =

(changedRefs, unchangedRefs)

where

-- class or (if the type class defined in another module) we have to

-- recompile modules that use types defined in this module affected by the

-- instance.

applyInstances (a, r, c, u) =

let checkType t (TypeRef t') = t' == t

checkType _ _ = False

go (TypeInstanceRef _ (clsMod, cls) types)

| clsRef <- TypeClassRef cls =

if clsMod == modName

then -- If the class is defined in this module we ensure that is marked as changed.

else case S.member clsRef <$> M.lookup clsMod depsDiffsMap of

Just True ->

-- If the type class is in another module and it has

-- changed we don't need to care about instance types

-- (because the instance change affects modules that use

-- the type class/its methods).

_ ->

-- Otherwise mark instance types as changed.

-- Check class instances in added, removed and changed.

in (a, r, c <> uc, uu)

-- Group/split exported refs of the module into (added, removed, changed,

-- Make the context for fixity's data constructor search: place all

-- known refs in the map.

fixityCtx = M.insert modName (getRefsSet declsSplit) depsDiffsMap

-- Determine which declarations where directly changed or removed by

, splitRefs (getTypeFixities newExts) (getTypeFixities oldExts) (pure . externsTypeFixityToRef)

]

changedRefs =

getCtorsSets :: P.ExternsFile -> P.ExternsFile -> Set Ref

getAffectedLocal modName diffsMap unchangedRefs =

affectedLocalRefs

where

hasChangedDeps (mn, ref) =

Just True == (S.member ref <$> M.lookup mn diffsMap)

(affectedByChanged, restLocalRefs) =

-- Use graph to go though local refs and their cyclic dependencies on each other.

-- The graph includes only local refs that depend on other local refs.

toNode (ref, deps) = (ref, ref, map snd $ filter ((== modName) . fst) (S.toList deps))

-- Make graph vertexes from the rest local refs with deps and affected refs

-- with no deps.

(graph, fromVtx, toVtx) = G.graphFromEdges vtxs

-- Graph is a list of refs with (refs) dependencies.

refsGraph = do

(_, t, _) <- vtxs

let v = fromMaybe (internalError "diffExterns: vertex not found") $ toVtx t

affectedLocalRefs =

S.fromList $

map fst $

diffExterns :: [ExternsDiff] -> P.ExternsFile -> P.ExternsFile -> ExternsDiff

-- Add data constructors to refs if all are implicitly imported using (..).

getCtor n (ConstructorRef tn _) = tn == n

getCtor _ _ = False

getRefs = (toRefs <>) . addCtors

go s (P.ImportDeclaration _ mn dt qual)

-- We return Nothing if we encounter removed refs in imports.

| Just diffs <- M.lookup mn diffsMap

, removed <- M.keysSet $ M.filter isRefRemoved diffs =

P.Explicit dRefs

| Just ref <- searchRef removed refs -> Left (mn, ref)

| otherwise ->

-- Search only refs encountered in the import.

where

refs = foldMap (getRefs mn) dRefs

P.Hiding dRefs

| Just ref <- searchRef removed refs -> Left (mn, ref)

| otherwise ->

-- Search only refs not encountered in the import.

where

refs = foldMap (getRefs mn) dRefs

-- Search all changed refs.

go s _ = Right s

toRefs = \case

isEmpty :: ExternsDiff -> Bool

isEmpty (ExternsDiff _ refs)

type Tuple4 m a = (m a, m a, m a, m a)

-- | Returns refs as a tuple of four (added, removed, changed, unchanged).

splitRefs new old toRef =

where

toMap = M.fromList . mapMaybe (((<$>) . flip (,)) <*> toRef)

newMap = toMap new

oldMap = toMap old

go ref decl (a, r, c, u) = case M.lookup ref newMap of

Just newDecl

-- | Traverses the type and finds all the refs within.

typeDeps :: P.Type a -> S.Set (ModuleName, Ref)

isPrimModule :: ModuleName -> Bool

isPrimModule = flip S.member (S.fromList primModules)