Skip to content

Commit 7c72ad3

Browse files
committed
feat: new route
1 parent 81f720b commit 7c72ad3

File tree

2 files changed

+36
-2
lines changed

2 files changed

+36
-2
lines changed

Diff for: cmd/http.go

+2
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,9 @@ func setUpRoutes(app *fiber.App, conn *start.Conn) {
3737
app.Get("count", conn.Count)
3838
app.Get("enrich", conn.Enrich)
3939
app.Get("linkedin", conn.Linkedin)
40+
app.Get("logs", conn.Logs)
4041
app.Get("search", conn.Search)
4142
app.Get("status", conn.Status)
43+
app.Get("usage", conn.Usage)
4244
app.Get("verify", conn.Verify)
4345
}

Diff for: pkg/start/routes.go

+34-2
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,12 @@ func (d *Conn) Home(c *fiber.Ctx) error {
1919
func (init *Conn) Author(c *fiber.Ctx) error {
2020
url := strings.ToLower(c.Query("url"))
2121
if !_url.IsValidURL(url) {
22-
log.Info(ErrArgumentsURL.Error())
22+
log.Error(ErrArgumentsURL.Error())
2323
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentsURL.Error()})
2424
}
2525
result, err := init.Tomba.AuthorFinder(url)
2626
if err != nil {
27-
log.Info(ErrErrInvalidLogin.Error())
27+
log.Error(ErrErrInvalidLogin.Error())
2828
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
2929
}
3030
if result.Data.Email == "" {
@@ -39,10 +39,12 @@ func (init *Conn) Author(c *fiber.Ctx) error {
3939
func (init *Conn) Count(c *fiber.Ctx) error {
4040
domain := strings.ToLower(c.Query("domain"))
4141
if !_domain.IsValidDomain(domain) {
42+
log.Error(ErrArgumentsDomain.Error())
4243
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentsDomain.Error()})
4344
}
4445
result, err := init.Tomba.Count(domain)
4546
if err != nil {
47+
log.Error(ErrErrInvalidLogin.Error())
4648
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
4749
}
4850
if result.Data.Total == 0 {
@@ -56,10 +58,12 @@ func (init *Conn) Enrich(c *fiber.Ctx) error {
5658
email := strings.ToLower(c.Query("email"))
5759

5860
if !_email.IsValidEmail(email) {
61+
log.Error(ErrArgumentEmail.Error())
5962
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentEmail.Error()})
6063
}
6164
result, err := init.Tomba.Enrichment(email)
6265
if err != nil {
66+
log.Error(ErrErrInvalidLogin.Error())
6367
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
6468
}
6569
if result.Data.Email == "" {
@@ -73,10 +77,12 @@ func (init *Conn) Linkedin(c *fiber.Ctx) error {
7377
url := strings.ToLower(c.Query("url"))
7478

7579
if !_url.IsValidURL(url) {
80+
log.Error(ErrArgumentsURL.Error())
7681
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentsURL.Error()})
7782
}
7883
result, err := init.Tomba.LinkedinFinder(url)
7984
if err != nil {
85+
log.Error(ErrErrInvalidLogin.Error())
8086
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
8187
}
8288
if result.Data.Email == "" {
@@ -85,15 +91,27 @@ func (init *Conn) Linkedin(c *fiber.Ctx) error {
8591
return c.Status(200).JSON(result)
8692
}
8793

94+
// Logs query logs
95+
func (init *Conn) Logs(c *fiber.Ctx) error {
96+
result, err := init.Tomba.Logs()
97+
if err != nil {
98+
log.Error(ErrErrInvalidLogin.Error())
99+
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
100+
}
101+
return c.Status(200).JSON(result)
102+
}
103+
88104
// Search query domain search
89105
func (init *Conn) Search(c *fiber.Ctx) error {
90106
domain := strings.ToLower(c.Query("domain"))
91107

92108
if !_domain.IsValidDomain(domain) {
109+
log.Error(ErrArgumentsDomain.Error())
93110
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentsDomain.Error()})
94111
}
95112
result, err := init.Tomba.DomainSearch(domain, "10", "0")
96113
if err != nil {
114+
log.Error(ErrErrInvalidLogin.Error())
97115
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
98116
}
99117
if result.Meta.Total == 0 {
@@ -108,10 +126,22 @@ func (init *Conn) Status(c *fiber.Ctx) error {
108126
domain := strings.ToLower(c.Query("domain"))
109127

110128
if !_domain.IsValidDomain(domain) {
129+
log.Error(ErrArgumentsDomain.Error())
111130
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentsDomain.Error()})
112131
}
113132
result, err := init.Tomba.Status(domain)
114133
if err != nil {
134+
log.Error(ErrErrInvalidLogin.Error())
135+
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
136+
}
137+
return c.Status(200).JSON(result)
138+
}
139+
140+
// Usage query usage
141+
func (init *Conn) Usage(c *fiber.Ctx) error {
142+
result, err := init.Tomba.Usage()
143+
if err != nil {
144+
log.Error(ErrErrInvalidLogin.Error())
115145
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
116146
}
117147
return c.Status(200).JSON(result)
@@ -122,11 +152,13 @@ func (init *Conn) Verify(c *fiber.Ctx) error {
122152
email := strings.ToLower(c.Query("email"))
123153

124154
if !_email.IsValidEmail(email) {
155+
log.Error(ErrArgumentEmail.Error())
125156
return c.Status(400).JSON(fiber.Map{"status": "error", "message": ErrArgumentEmail.Error()})
126157
}
127158

128159
result, err := init.Tomba.EmailVerifier(email)
129160
if err != nil {
161+
log.Error(ErrErrInvalidLogin.Error())
130162
return c.Status(401).JSON(fiber.Map{"status": "error", "message": ErrErrInvalidLogin.Error()})
131163
}
132164
if result.Data.Email.Email != "" {

0 commit comments

Comments
 (0)