Add migration for totp table

This commit is contained in:
kolaente 2020-04-17 17:54:47 +02:00
parent 8e31996ee3
commit 5d2627d46a
Signed by: konrad
GPG Key ID: F40E70337AB24C9B
3 changed files with 48 additions and 1 deletions

View File

@ -41,5 +41,4 @@ func init() {
return tx.DropTables(list20200322214624{})
},
})
}

View File

@ -0,0 +1,47 @@
// Vikunja is a to-do list application to facilitate your life.
// Copyright 2018-2020 Vikunja and contributors. All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package migration
import (
"src.techknowlogick.com/xormigrate"
"xorm.io/xorm"
)
type totp20200417175201 struct {
ID int64 `xorm:"int(11) autoincr not null unique pk" json:"-"`
UserID int64 `xorm:"int(11) not null" json:"-"`
Secret string `xorm:"varchar(20) not null" json:"secret"`
Enabled bool `xorm:"null" json:"enabled"`
URL string `xorm:"text null" json:"url"`
}
func (t totp20200417175201) TableName() string {
return "totp"
}
func init() {
migrations = append(migrations, &xormigrate.Migration{
ID: "20200417175201",
Description: "Add totp table",
Migrate: func(tx *xorm.Engine) error {
return tx.Sync2(totp20200417175201{})
},
Rollback: func(tx *xorm.Engine) error {
return tx.DropTables(totp20200417175201{})
},
})
}

View File

@ -46,5 +46,6 @@ func InitDB() (err error) {
func GetTables() []interface{} {
return []interface{}{
&User{},
&TOTP{},
}
}