65 lines
2.1 KiB
VB.net
65 lines
2.1 KiB
VB.net
Imports System.Reflection
|
||
Imports Microsoft.VisualBasic.ComponentModel.DataSourceModel
|
||
Imports Microsoft.VisualBasic.Language
|
||
|
||
Namespace CommandLine.Reflection
|
||
|
||
''' <summary>
|
||
'''
|
||
''' </summary>
|
||
Public Class RunDllEntryPoint : Inherits [Namespace]
|
||
|
||
''' <summary>
|
||
''' rundll namespace::api
|
||
''' </summary>
|
||
''' <param name="Name"></param>
|
||
Sub New(Name As String)
|
||
Call MyBase.New(Name, "")
|
||
End Sub
|
||
|
||
''' <summary>
|
||
'''
|
||
''' </summary>
|
||
''' <param name="entrypoint$"></param>
|
||
''' <returns>
|
||
''' 假若没有api的名称的话，是默认使用一个名字为``Main``的主函数来运行的
|
||
''' </returns>
|
||
Public Shared Function GetPoint(entrypoint As String) As NamedValue(Of String)
|
||
Dim entry = entrypoint.GetTagValue("::", trim:=True)
|
||
|
||
If entry.Value.StringEmpty Then
|
||
entry.Value = "Main"
|
||
End If
|
||
|
||
Return entry
|
||
End Function
|
||
|
||
Public Shared Function GetDllMethod(assembly As Assembly, entryPoint$) As MethodInfo
|
||
Dim entry As NamedValue(Of String) = GetPoint(entryPoint)
|
||
Dim types As Type() = GetTypesHelper(assm:=assembly)
|
||
Dim dll As Type = LinqAPI.DefaultFirst(Of Type) _
|
||
_
|
||
() <= From type As Type
|
||
In types
|
||
Let load = type.GetCustomAttribute(Of RunDllEntryPoint)
|
||
Let name = load?.Namespace
|
||
Where Not load Is Nothing AndAlso name.TextEquals(entry.Name)
|
||
Select type
|
||
|
||
If dll Is Nothing Then
|
||
Return Nothing
|
||
Else
|
||
Dim matchName = Function(m As MethodInfo)
|
||
Return m.Name.TextEquals(entry.Value)
|
||
End Function
|
||
Dim method As MethodInfo = dll _
|
||
.GetMethods(PublicShared) _
|
||
.Where(predicate:=matchName) _
|
||
.FirstOrDefault
|
||
|
||
Return method
|
||
End If
|
||
End Function
|
||
End Class
|
||
|
||
End Namespace |