@@ -16,30 +16,11 @@ class HeaderNewsChannels: UITableViewCell {
1616 weak var vc: NewsController?
1717 weak var vcProfile: ProfileViewController?
1818
19− var type: HeaderType = .PROFILE("Posts", "Channel")
20− {
21− didSet{
22− switch type {
23− case .PROFILE(let f, let s):
24− postsChannelsSegment = UISegmentedControl(items: [f,s])
25− postsChannelsSegment?.selectedSegmentIndex = 0
26− case .BASIC_INFO(let b, let i):
27− postsChannelsSegment = UISegmentedControl(items: [b,i])
28− postsChannelsSegment?.selectedSegmentIndex = 1
29− case .NEWS(let n, let bi):
30− postsChannelsSegment = UISegmentedControl(items: [n, bi])
31− postsChannelsSegment?.selectedSegmentIndex = 0
32− default:
33− break
34− }
35− setUpCell()
36− }
37− }
3819
3920 override init(style: UITableViewCell.CellStyle, reuseIdentifier: String?) {
4021 super.init(style: style, reuseIdentifier: reuseIdentifier)
4122 backgroundColor = .white
42− //setUpCell()
23+ setUpCell()
4324 }
4425
4526 required init?(coder aDecoder: NSCoder) {
@@ -48,43 +29,23 @@ class HeaderNewsChannels: UITableViewCell {
4829
4930 func setUpCell(){
5031
51− guard let postsChannelsSegment = postsChannelsSegment else {
52− return
53− }
54−
5532 addSubview(postsChannelsSegment)
56−
33+ postsChannelsSegment.selectedSegmentIndex = 0
5734 postsChannelsSegment.addTarget(self, action: #selector(changedValue), for: .valueChanged)
5835 postsChannelsSegment.edgesToSuperview(insets: .left(16) + .right(16) + .top(8) + .bottom(8))
5936 }
6037
6138 // Initialize
62− var postsChannelsSegment: UISegmentedControl?
39+ var postsChannelsSegment: UISegmentedControl = UISegmentedControl(items: ["Posts", "Channels"])
6340
6441 @objc func changedValue(_ sender: UISegmentedControl) {
6542 let index = sender.selectedSegmentIndex
6643
6744 switch index {
6845 case 0:
69− switch type {
70− case .BASIC_INFO:
71− vcProfile?.changeToPosts()
72− case .NEWS:
73− vc?.setUpNavigationItemsforPosts()
74− default:
75− break
76− }
46+ vc?.setUpNavigationItemsforPosts()
7747 case 1:
78− switch type {
79− case .PROFILE:
80− print("info")
81− vcProfile?.changeToBasicInfo()
82− case .NEWS:
83− print("channels")
84− vc?.setUpNavigationItemsForChannels()
85− default:
86− break
87− }
48+ vc?.setUpNavigationItemsForChannels()
8849 default:
8950 break
9051 }