lib/packwerk/cache_deprecated_references.rb in packwerk-1.1.1 vs lib/packwerk/cache_deprecated_references.rb in packwerk-1.1.2

@@ -1,6 +1,6 @@
-# typed: true
+# typed: strict
 # frozen_string_literal: true
 
 require "sorbet-runtime"
 
 require "packwerk/deprecated_references"
@@ -13,13 +13,19 @@
     extend T::Sig
     extend T::Helpers
     include ReferenceLister
     abstract!
 
+    sig do
+      params(
+        root_path: String,
+        deprecated_references: T::Hash[Packwerk::Package, Packwerk::DeprecatedReferences]
+      ).void
+    end
     def initialize(root_path, deprecated_references = {})
       @root_path = root_path
-      @deprecated_references = T.let(deprecated_references, T::Hash[String, Packwerk::DeprecatedReferences])
+      @deprecated_references = T.let(deprecated_references, T::Hash[Packwerk::Package, Packwerk::DeprecatedReferences])
     end
 
     sig do
       params(reference: Packwerk::Reference, violation_type: ViolationType)
         .returns(T::Boolean)
@@ -31,16 +37,18 @@
       true
     end
 
     private
 
+    sig { params(package: Packwerk::Package).returns(Packwerk::DeprecatedReferences) }
     def deprecated_references_for(package)
       @deprecated_references[package] ||= Packwerk::DeprecatedReferences.new(
         package,
         deprecated_references_file_for(package),
       )
     end
 
+    sig { params(package: Packwerk::Package).returns(String) }
     def deprecated_references_file_for(package)
       File.join(@root_path, package.name, "deprecated_references.yml")
     end
   end
 end