From e811a9d563f9a6cab00f0b57529fc38716234987 Mon Sep 17 00:00:00 2001
From: pikachu0310 <pikachu13711@gmail.com>
Date: Mon, 30 Sep 2024 20:22:16 +0900
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix=20some=20logic?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 grpc/federation/cel/url.go | 52 +++++++++++++++++++++++---------------
 1 file changed, 31 insertions(+), 21 deletions(-)

diff --git a/grpc/federation/cel/url.go b/grpc/federation/cel/url.go
index 05c39cf1..de351295 100644
--- a/grpc/federation/cel/url.go
+++ b/grpc/federation/cel/url.go
@@ -28,24 +28,28 @@ func (x *URL) GoURL() (url.URL, error) {
 	}
 
 	return url.URL{
-		Scheme:     x.GetScheme(),
-		Opaque:     x.GetOpaque(),
-		User:       user,
-		Host:       x.GetHost(),
-		Path:       x.GetPath(),
-		RawPath:    x.GetRawPath(),
-		ForceQuery: x.GetForceQuery(),
-		RawQuery:   x.GetRawQuery(),
-		Fragment:   x.GetFragment(),
+		Scheme:      x.GetScheme(),
+		Opaque:      x.GetOpaque(),
+		User:        user,
+		Host:        x.GetHost(),
+		Path:        x.GetPath(),
+		RawPath:     x.GetRawPath(),
+		OmitHost:    x.GetOmitHost(),
+		ForceQuery:  x.GetForceQuery(),
+		RawQuery:    x.GetRawQuery(),
+		Fragment:    x.GetFragment(),
+		RawFragment: x.GetRawFragment(),
 	}, nil
 }
 
 func (x *Userinfo) GoUserinfo() (*url.Userinfo, error) {
+	if x == nil {
+		return nil, nil
+	}
 	if x.GetPasswordSet() {
 		return url.UserPassword(x.GetUsername(), x.GetPassword()), nil
-	} else {
-		return url.User(x.GetUsername()), nil
 	}
+	return url.User(x.GetUsername()), nil
 }
 
 var _ cel.SingletonLibrary = new(URLLibrary)
@@ -267,17 +271,27 @@ func (lib *URLLibrary) CompileOptions() []cel.EnvOption {
 				},
 			),
 		),
-		// forceQuery is changed to return String
+		BindMemberFunction(
+			"omitHost",
+			MemberOverloadFunc(createURLID("omitHost_url_bool"), URLType, []*cel.Type{}, cel.BoolType,
+				func(_ context.Context, self ref.Val, args ...ref.Val) ref.Val {
+					v, err := self.Value().(*URL).GoURL()
+					if err != nil {
+						return types.NewErr(err.Error())
+					}
+					return types.Bool(v.OmitHost)
+				},
+			),
+		),
 		BindMemberFunction(
 			"forceQuery",
-			MemberOverloadFunc(createURLID("forceQuery_url_string"), URLType, []*cel.Type{}, cel.StringType,
+			MemberOverloadFunc(createURLID("forceQuery_url_string"), URLType, []*cel.Type{}, cel.BoolType,
 				func(_ context.Context, self ref.Val, args ...ref.Val) ref.Val {
 					v, err := self.Value().(*URL).GoURL()
 					if err != nil {
 						return types.NewErr(err.Error())
 					}
-					v.ForceQuery = true
-					return types.String(v.String())
+					return types.Bool(v.ForceQuery)
 				},
 			),
 		),
@@ -565,12 +579,8 @@ func (lib *URLLibrary) CompileOptions() []cel.EnvOption {
 					if err != nil {
 						return types.NewErr(err.Error())
 					}
-					password, hasPassword := v.Password()
-					if hasPassword {
-						return types.String(password)
-					} else {
-						return types.String("")
-					}
+					password, _ := v.Password()
+					return types.String(password)
 				},
 			),
 		),