From dfeff840f306696771f8a9cb54f5bf94c514dc2f Mon Sep 17 00:00:00 2001
From: jvlflame <jli141928@gmail.com>
Date: Mon, 31 Aug 2020 00:30:23 -0700
Subject: [PATCH] Update function name Get-JVUrlLocation

---
 ...-UrlLocation.ps1 => Get-JVUrlLocation.ps1} | 49 +++++++++----------
 src/Javinizer/Public/Get-JavlibraryUrl.ps1    |  2 +-
 2 files changed, 24 insertions(+), 27 deletions(-)
 rename src/Javinizer/Private/{Test-UrlLocation.ps1 => Get-JVUrlLocation.ps1} (60%)

diff --git a/src/Javinizer/Private/Test-UrlLocation.ps1 b/src/Javinizer/Private/Get-JVUrlLocation.ps1
similarity index 60%
rename from src/Javinizer/Private/Test-UrlLocation.ps1
rename to src/Javinizer/Private/Get-JVUrlLocation.ps1
index e504b2b8..12b92402 100644
--- a/src/Javinizer/Private/Test-UrlLocation.ps1
+++ b/src/Javinizer/Private/Get-JVUrlLocation.ps1
@@ -1,81 +1,78 @@
-function Test-UrlLocation {
+function Get-JVUrlLocation {
     [CmdletBinding()]
     param(
-        [Parameter(Mandatory = $true, Position = 0)]
+        [Parameter(Mandatory = $true, Position = 0, ValueFromPipeline = $true)]
         [array]$Url
     )
 
-    begin {
-        Write-Debug "[$(Get-TimeStamp)][$($MyInvocation.MyCommand.Name)] Function started"
-        $testUrlObject = @()
-    }
-
     process {
+        $testUrlObject = @()
         foreach ($link in $Url) {
             if ($link -match 'r18.com') {
                 if ($link -match 'lg=zh') {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'r18zh'
+                        Source = 'r18zh'
                     }
                 } else {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'r18'
+                        Source = 'r18'
                     }
                 }
             } elseif ($link -match 'javlibrary.com') {
                 if ($link -match '/ja/') {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javlibraryja'
+                        Source = 'javlibraryja'
                     }
                 } elseif ($link -match '/cn/') {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javlibraryzh'
+                        Source = 'javlibraryzh'
                     }
                 } elseif ($link -match '/tw/') {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javlibraryzh'
+                        Source = 'javlibraryzh'
                     }
                 } else {
                     $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javlibrary'
+                        Source = 'javlibrary'
                     }
                 }
             } elseif ($link -match 'dmm.co.jp') {
-                $testUrlObject = [pscustomobject]@{
+                $testUrlObject += [pscustomobject]@{
                     Url    = $link
-                    Result = 'dmm'
+                    Source = 'dmm'
                 }
             } elseif ($link -match 'javbus') {
                 if ($link -match '/ja') {
-                    $testUrlObject = [pscustomobject]@{
+                    $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javbusja'
+                        Source = 'javbusja'
+                    }
+                } elseif ($link -match '/zh') {
+                    $testUrlObject += [pscustomobject]@{
+                        Url    = $link
+                        Source = 'javbuszh'
                     }
                 } else {
-                    $testUrlObject = [pscustomobject]@{
+                    $testUrlObject += [pscustomobject]@{
                         Url    = $link
-                        Result = 'javbus'
+                        Source = 'javbus'
                     }
                 }
             } elseif ($link -match 'jav321.com') {
-                $testUrlObject = [pscustomboject]@{
+                $testUrlObject += [pscustomboject]@{
                     Url    = $link
-                    Result = 'jav321'
+                    Source = 'jav321'
                 }
             } else {
-                Write-Warning "[$(Get-TimeStamp)][$($MyInvocation.MyCommand.Name)] Url: [$Url] not matched"
+                Write-JLog -Level Warning -Message "[$($MyInvocation.MyCommand.Name)] [Url - $Url] not matched"
             }
         }
         Write-Output $testUrlObject
     }
-
-    end {
-        Write-Debug "[$(Get-TimeStamp)][$($MyInvocation.MyCommand.Name)] Function ended"
-    }
 }
diff --git a/src/Javinizer/Public/Get-JavlibraryUrl.ps1 b/src/Javinizer/Public/Get-JavlibraryUrl.ps1
index f4a4afea..d2fb5ffd 100644
--- a/src/Javinizer/Public/Get-JavlibraryUrl.ps1
+++ b/src/Javinizer/Public/Get-JavlibraryUrl.ps1
@@ -62,7 +62,7 @@ function Get-JavlibraryUrl {
                     Write-JLog -Level Debug -Message "[$Id] [$($MyInvocation.MyCommand.Name)] Result [$count] is [$resultId]"
 
                     if ($resultId -eq $Id) {
-                        $javlibraryUrl = (Test-UrlLocation -Url $webRequest.BaseResponse.RequestMessage.RequestUri.AbsoluteUri).Url
+                        $javlibraryUrl = (Get-JVUrlLocation -Url $webRequest.BaseResponse.RequestMessage.RequestUri.AbsoluteUri).Url
                         break
                     }