Most of the time, I find ideas for a song when improvising within a scale, then I find the chord that are within that scale for rythm guitar. It can be good sometime to add non-diatonic chord to make things a bit more unpredictable and interesting.