add discovery metadata

YoloDev · Jul 24, 2015 · 25206e2 · 25206e2
1 parent 816d405
commit 25206e2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/src/YoloDev.Dnx.NativeUtils.Generator/NativeGenerator.cs b/src/YoloDev.Dnx.NativeUtils.Generator/NativeGenerator.cs
@@ -19,6 +19,7 @@ class NativeGenerator
 
         readonly List<MemberDeclarationSyntax> _members = new List<MemberDeclarationSyntax>();
         readonly ClassDeclarationSyntax _class;
+        readonly AttributeSyntax _attribute;
 
         public NativeGenerator(NativeModel api, Compilation compilation)
         {
@@ -27,6 +28,11 @@ public NativeGenerator(NativeModel api, Compilation compilation)
             _class = SyntaxFactory.ClassDeclaration($"NativeMethods${api.Symbol.MetadataName}")
                 .AddBaseListTypes(SyntaxFactory.SimpleBaseType(SyntaxFactory.ParseTypeName(api.Symbol.ToDisplayString())))
                 .AddAttributeLists(SyntaxFactory.AttributeList().AddAttributes(CreateAttribute<CompilerGeneratedAttribute>()));
+            _attribute = SyntaxFactory.Attribute(SyntaxFactory.ParseName(compilation.GetTypeName<NativeApiAttribute>()))
+                .AddArgumentListArguments(
+                    SyntaxFactory.AttributeArgument(SyntaxFactory.TypeOfExpression(SyntaxFactory.ParseTypeName(api.Symbol.ToDisplayString()))),
+                    SyntaxFactory.AttributeArgument(SyntaxFactory.TypeOfExpression(SyntaxFactory.QualifiedName(_ns.Name, SyntaxFactory.IdentifierName(_class.Identifier))))
+                );
         }
 
         SyntaxTree Generate()
@@ -42,6 +48,7 @@ SyntaxTree Generate()
             }
 
             return SyntaxFactory.SyntaxTree(SyntaxFactory.CompilationUnit()
+                .AddAttributeLists(SyntaxFactory.AttributeList().WithTarget(SyntaxFactory.AttributeTargetSpecifier(SyntaxFactory.Token(SyntaxKind.AssemblyKeyword))).AddAttributes(_attribute))
                 .WithMembers(new SyntaxList<MemberDeclarationSyntax>().Add(_ns.AddMembers(_class.AddMembers(_members.ToArray())))).NormalizeWhitespace());
         }
 

diff --git a/src/YoloDev.Dnx.NativeUtils/NativeApiAttribute.cs b/src/YoloDev.Dnx.NativeUtils/NativeApiAttribute.cs
@@ -3,7 +3,17 @@
 
 namespace YoloDev.Dnx.NativeUtils
 {
-  [AttributeUsage(AttributeTargets.Class)]
+  [AttributeUsage(AttributeTargets.Assembly)]
   [EditorBrowsable(EditorBrowsableState.Never)]
-  public sealed class NativeApiAttribute : Attribute {}
+  public sealed class NativeApiAttribute : Attribute
+  {
+      readonly Type _target;
+      readonly Type _implementation;
+
+      public NativeApiAttribute(Type target, Type implementation)
+      {
+          _target = target;
+          _implementation = implementation;
+      }
+  }
 }