Example #1
0
func gen(dir string, names []string) {
	logger := Logger.WithField("state", "gen_file")
	logger.WithFields(logrus.Fields{
		"type": "info",
		"info": "gen file start",
	}).Info("gen file start")
	_, err := os.Stat(filepath.Join(dir, "problem.toml"))
	if err != nil {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  err.Error(),
		}).Fatal("\"problem.toml\" does not exist")
	}
	p, err := libra.Load(dir, logger)
	var a []libra.SrcInfo
	//answer add
	for _, x := range p.Answer {
		a = append(a, libra.SrcInfo{Name: x.Name, Type: libra.ANS})
	}
	//checker add
	a = append(a, libra.SrcInfo{Name: p.Checker, Type: libra.CHK})
	//generator add
	for _, x := range p.Generator {
		a = append(a, libra.SrcInfo{Name: x.Name, Type: libra.GEN})
	}
	//solution add
	a = append(a, libra.SrcInfo{Name: p.Solution, Type: libra.SOL})
	//validator add
	a = append(a, libra.SrcInfo{Name: p.Validator, Type: libra.VAL})

	//trygen start
	for _, x := range a {
		if contain(x.Name, names) {
			tryGen(dir, x.Name, logger)
		}
	}
	if err != nil {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  err.Error(),
		}).Fatal("Loading failed.")
	}

	logger.WithFields(logrus.Fields{
		"type": "info",
		"info": "gen file end"}).Infof("add file end")

}
Example #2
0
func rem(dir, name string) {
	logger := Logger.WithField("state", "rem_file")
	logger.WithFields(logrus.Fields{
		"type": "info",
		"info": "remove file start",
	}).Info("rem file start")
	if name == "" {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  "empty string",
		}).Fatal("empty string is not allowed")
	}
	_, err := os.Stat(filepath.Join(dir, "problem.toml"))
	if err != nil {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  err.Error(),
		}).Fatal("\"problem.toml\" does not exist")
	}
	p, err := libra.Load(dir, logger)
	if err != nil {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  err.Error(),
		}).Fatal("Loading failed.")
	}
	ok := false
	//answer
	for i := 0; i < len(p.Answer); {
		if p.Answer[i].Name == name {
			p.Answer = append(p.Answer[:i], p.Answer[i+1:]...)
			ok = true
		} else {
			i++
		}
	}
	//checker
	if p.Checker == name {
		p.Checker = ""
		ok = true
	}
	//generator
	for i := 0; i < len(p.Generator); {
		if p.Generator[i].Name == name {
			p.Generator = append(p.Generator[:i], p.Generator[i+1:]...)
			ok = true
		} else {
			i++
		}
	}
	//solution
	if p.Solution == name {
		p.Solution = ""
		ok = true
	}
	//validator
	if p.Validator == name {
		p.Validator = ""
		ok = true
	}
	if !ok {
		logger.WithFields(logrus.Fields{
			"type":  "info",
			"error": libra.IE,
		}).Warnf("file not found: %s", name)
	}
	err = libra.Save(dir, p, logger)
	if err != nil {
		logger.WithFields(logrus.Fields{
			"type":  "error",
			"error": libra.IE,
			"info":  err.Error(),
		}).Fatal("Save failed.")
	}
	logger.WithFields(logrus.Fields{
		"type": "info",
		"info": "rem file end"}).Infof("remove file end")

}