// Copyright 2021 The casbin Authors. 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.

package discuzx

type Class struct {
	Typeid       int
	Fid          int
	Name         string
	Displayorder int
}

func getClasses() []*Class {
	classes := []*Class{}
	err := adapter.Engine.Table("pre_forum_threadclass").Find(&classes)
	if err != nil {
		panic(err)
	}

	return classes
}

func getClass(id int) *Class {
	class := Class{Typeid: id}
	existed, err := adapter.Engine.Table("pre_forum_threadclass").Get(&class)
	if err != nil {
		panic(err)
	}

	if existed {
		return &class
	} else {
		return nil
	}
}

func getClassMap() map[int]*Class {
	classes := getClasses()

	m := map[int]*Class{}
	for _, class := range classes {
		m[class.Typeid] = class
	}
	return m
}
