harbor/src/ui/main.go

152 lines
4.5 KiB
Go
Raw Normal View History

2017-04-13 12:54:58 +02:00
// Copyright (c) 2017 VMware, Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
2016-04-11 10:43:13 +02:00
package main
import (
"fmt"
"os"
"reflect"
2016-04-11 10:43:13 +02:00
2016-10-19 08:32:00 +02:00
"github.com/vmware/harbor/src/common/utils"
"github.com/vmware/harbor/src/common/utils/log"
2016-09-13 11:41:32 +02:00
"github.com/astaxie/beego"
_ "github.com/astaxie/beego/session/redis"
2016-04-11 10:43:13 +02:00
"github.com/vmware/harbor/src/common/dao"
"github.com/vmware/harbor/src/common/models"
2017-07-06 18:38:38 +02:00
"github.com/vmware/harbor/src/common/notifier"
"github.com/vmware/harbor/src/common/scheduler"
2017-11-29 08:14:13 +01:00
"github.com/vmware/harbor/src/replication/core"
_ "github.com/vmware/harbor/src/replication/event"
2016-10-19 08:32:00 +02:00
"github.com/vmware/harbor/src/ui/api"
_ "github.com/vmware/harbor/src/ui/auth/db"
_ "github.com/vmware/harbor/src/ui/auth/ldap"
_ "github.com/vmware/harbor/src/ui/auth/uaa"
"github.com/vmware/harbor/src/ui/config"
2017-04-26 09:28:13 +02:00
"github.com/vmware/harbor/src/ui/filter"
2017-05-02 13:14:47 +02:00
"github.com/vmware/harbor/src/ui/proxy"
2017-02-26 12:53:13 +01:00
"github.com/vmware/harbor/src/ui/service/token"
2016-04-11 10:43:13 +02:00
)
const (
adminUserID = 1
)
func updateInitPassword(userID int, password string) error {
queryUser := models.User{UserID: userID}
user, err := dao.GetUser(queryUser)
if err != nil {
return fmt.Errorf("Failed to get user, userID: %d %v", userID, err)
}
if user == nil {
2016-11-14 07:46:20 +01:00
return fmt.Errorf("user id: %d does not exist", userID)
2016-04-11 10:43:13 +02:00
}
if user.Salt == "" {
2016-09-13 11:41:32 +02:00
salt := utils.GenerateRandomString()
2016-04-11 10:43:13 +02:00
user.Salt = salt
user.Password = password
err = dao.ChangeUserPassword(*user)
if err != nil {
return fmt.Errorf("Failed to update user encrypted password, userID: %d, err: %v", userID, err)
}
log.Infof("User id: %d updated its encypted password successfully.", userID)
} else {
log.Infof("User id: %d already has its encrypted password.", userID)
}
return nil
}
func main() {
beego.BConfig.WebConfig.Session.SessionOn = true
2016-07-27 14:12:53 +02:00
//TODO
redisURL := os.Getenv("_REDIS_URL")
if len(redisURL) > 0 {
beego.BConfig.WebConfig.Session.SessionProvider = "redis"
beego.BConfig.WebConfig.Session.SessionProviderConfig = redisURL
}
beego.AddTemplateExt("htm")
2016-09-13 11:41:32 +02:00
2016-12-30 11:04:01 +01:00
log.Info("initializing configurations...")
if err := config.Init(); err != nil {
log.Fatalf("failed to initialize configurations: %v", err)
}
log.Info("configurations initialization completed")
2017-02-26 12:53:13 +01:00
token.InitCreators()
2016-12-30 11:04:01 +01:00
database, err := config.Database()
if err != nil {
log.Fatalf("failed to get database configuration: %v", err)
}
if err := dao.InitDatabase(database); err != nil {
log.Fatalf("failed to initialize database: %v", err)
}
if config.WithClair() {
clairDB, err := config.ClairDB()
if err != nil {
log.Fatalf("failed to load clair database information: %v", err)
}
if err := dao.InitClairDB(clairDB); err != nil {
log.Fatalf("failed to initialize clair database: %v", err)
}
}
2016-12-30 11:04:01 +01:00
password, err := config.InitialAdminPassword()
if err != nil {
log.Fatalf("failed to get admin's initia password: %v", err)
}
if err := updateInitPassword(adminUserID, password); err != nil {
2016-04-11 10:43:13 +02:00
log.Error(err)
}
2017-04-26 09:28:13 +02:00
2017-07-06 18:38:38 +02:00
//Enable the policy scheduler here.
scheduler.DefaultScheduler.Start()
//Subscribe the policy change topic.
notifier.Subscribe(notifier.ScanAllPolicyTopic, &notifier.ScanPolicyNotificationHandler{})
//Get policy configuration.
scanAllPolicy := config.ScanAllPolicy()
if scanAllPolicy.Type == notifier.PolicyTypeDaily {
dailyTime := 0
if t, ok := scanAllPolicy.Parm["daily_time"]; ok {
if reflect.TypeOf(t).Kind() == reflect.Int {
dailyTime = t.(int)
}
2017-07-06 18:38:38 +02:00
}
//Send notification to handle first policy change.
notifier.Publish(notifier.ScanAllPolicyTopic, notifier.ScanPolicyNotification{Type: scanAllPolicy.Type, DailyTime: (int64)(dailyTime)})
2017-07-06 18:38:38 +02:00
}
if err := core.Init(); err != nil {
log.Errorf("failed to initialize the replication controller: %v", err)
2017-11-29 08:14:13 +01:00
}
2017-06-18 07:51:42 +02:00
filter.Init()
2017-04-26 09:28:13 +02:00
beego.InsertFilter("/*", beego.BeforeRouter, filter.SecurityFilter)
beego.InsertFilter("/api/*", beego.BeforeRouter, filter.MediaTypeFilter("application/json"))
2017-04-26 09:28:13 +02:00
2016-04-11 10:43:13 +02:00
initRouters()
2017-05-25 07:33:51 +02:00
if err := api.SyncRegistry(config.GlobalProjectMgr); err != nil {
log.Error(err)
}
2017-05-02 13:14:47 +02:00
log.Info("Init proxy")
proxy.Init()
//go proxy.StartProxy()
2016-04-11 10:43:13 +02:00
beego.Run()
}