github.com/drone/go-convert@v0.0.0-20240307072510-6bd371c65e61/convert/gitlab/yaml/allow_failure.go (about)

     1  // Copyright 2022 Harness, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //      http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  package yaml
    16  
    17  import "errors"
    18  
    19  type AllowFailure struct {
    20  	Value     bool  `yaml:"-"`
    21  	ExitCodes []int `yaml:"exit_codes"`
    22  }
    23  
    24  // UnmarshalYAML implements the unmarshal interface.
    25  func (v *AllowFailure) UnmarshalYAML(unmarshal func(interface{}) error) error {
    26  	var out1 bool
    27  	var out2 = struct {
    28  		ExitCode int `yaml:"exit_codes"`
    29  	}{}
    30  	var out3 = struct {
    31  		ExitCodes []int `yaml:"exit_codes"`
    32  	}{}
    33  
    34  	if err := unmarshal(&out3); err == nil {
    35  		v.Value = true
    36  		v.ExitCodes = out3.ExitCodes
    37  		return nil
    38  	}
    39  
    40  	if err := unmarshal(&out2); err == nil {
    41  		v.Value = true
    42  		v.ExitCodes = []int{out2.ExitCode}
    43  		return nil
    44  	}
    45  
    46  	if err := unmarshal(&out1); err == nil {
    47  		v.Value = out1
    48  		return nil
    49  	}
    50  
    51  	return errors.New("failed to unmarshal allow_failure")
    52  }
    53  
    54  // MarshalYAML implements the Marshal interface.
    55  func (v *AllowFailure) MarshalYAML() (interface{}, error) {
    56  	type allowFailureMarshal AllowFailure
    57  	if !v.Value {
    58  		return nil, nil
    59  	}
    60  
    61  	if len(v.ExitCodes) == 0 {
    62  		return v.Value, nil
    63  	}
    64  
    65  	return allowFailureMarshal(*v), nil
    66  }