From baa6f64a67b82b110e7de4f5c58becebe5598749 Mon Sep 17 00:00:00 2001
From: Wuvist <wuvist@gmail.com>
Date: Mon, 23 Mar 2020 20:40:56 +0800
Subject: [PATCH] Fix deps

---
 controller/cate.go | 8 ++++++--
 controller/home.go | 6 ++++--
 wire_gen.go        | 5 +++--
 3 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/controller/cate.go b/controller/cate.go
index a5ed183..8831369 100644
--- a/controller/cate.go
+++ b/controller/cate.go
@@ -18,8 +18,12 @@ type CateController struct {
 }
 
 // NewCateController return CateController bind with echo engine
-func NewCateController(e *echo.Echo) (*CateController, error) {
-	cate := &CateController{}
+func NewCateController(e *echo.Echo, bloggers models.BloggerQuery,
+	cates models.UserdefinecategoryQuery) (*CateController, error) {
+	cate := &CateController{
+		bloggers,
+		cates,
+	}
 	e.GET("/cate.go", cate.show)
 
 	return cate, nil
diff --git a/controller/home.go b/controller/home.go
index 818e13a..bde713d 100644
--- a/controller/home.go
+++ b/controller/home.go
@@ -16,8 +16,10 @@ type HomeController struct {
 }
 
 // NewHomeController return HomeController bind with echo engine
-func NewHomeController(e *echo.Echo) (*HomeController, error) {
-	ctrl := &HomeController{}
+func NewHomeController(e *echo.Echo, blogger models.BloggerQuery) (*HomeController, error) {
+	ctrl := &HomeController{
+		blogger,
+	}
 	e.GET("/", ctrl.show)
 
 	return ctrl, nil
diff --git a/wire_gen.go b/wire_gen.go
index f80d733..bb3fc88 100644
--- a/wire_gen.go
+++ b/wire_gen.go
@@ -24,11 +24,12 @@ func getWebApp() (*WebApp, error) {
 	if err != nil {
 		return nil, err
 	}
-	cateController, err := controller.NewCateController(echo)
+	userdefinecategoryQuery := getCategoryDB()
+	cateController, err := controller.NewCateController(echo, bloggerQuery, userdefinecategoryQuery)
 	if err != nil {
 		return nil, err
 	}
-	homeController, err := controller.NewHomeController(echo)
+	homeController, err := controller.NewHomeController(echo, bloggerQuery)
 	if err != nil {
 		return nil, err
 	}